37 lines
733 B
Go
37 lines
733 B
Go
package util
|
|
|
|
type FieldUtil struct{}
|
|
|
|
func (f FieldUtil) AllowSelect() []string {
|
|
return []string{
|
|
"download_folder",
|
|
}
|
|
}
|
|
|
|
func (f FieldUtil) AllowUpdate() []string {
|
|
return []string{
|
|
"download_folder",
|
|
}
|
|
}
|
|
|
|
func (f FieldUtil) IsAllow(allFields []string, names ...string) bool {
|
|
allowedFields := make(map[string]struct{})
|
|
for _, field := range allFields {
|
|
allowedFields[field] = struct{}{}
|
|
}
|
|
for _, name := range names {
|
|
if _, exists := allowedFields[name]; !exists {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (f FieldUtil) IsAllowSelect(names ...string) bool {
|
|
return f.IsAllow(f.AllowSelect(), names...)
|
|
}
|
|
|
|
func (f FieldUtil) IsAllowUpdate(names ...string) bool {
|
|
return f.IsAllow(f.AllowUpdate(), names...)
|
|
}
|