@@ -34,12 +34,15 @@ func (User) FindAll(db *gorm.DB, admin_or_user ...string) ([]User, error) {
3434 users := []User {}
3535
3636 var err error
37- if len (admin_or_user ) == 0 || (len (admin_or_user ) == 1 && admin_or_user [0 ] == "all" ) {
37+
38+ // same,
39+ // if len(*admin_or_user) == 0 || len(*admin_or_user) == 1 && (*admin_or_user)[0] == "all" {...}
40+ if isAll (& admin_or_user ) {
3841 // Limit: 25 ?
3942 err = db .Limit (25 ).Find (& users ).Error
40- } else if len ( admin_or_user ) == 1 && admin_or_user [ 0 ] == "admin" {
43+ } else if isAdmin ( & admin_or_user ) {
4144 err = db .Limit (25 ).Where ("is_admin = 1" ).Find (& users ).Error
42- } else if len ( admin_or_user ) == 1 && admin_or_user [ 0 ] == "user" {
45+ } else if isUser ( & admin_or_user ) {
4346 err = db .Limit (25 ).Where ("is_admin = 0" ).Find (& users ).Error
4447 } else { // admin_or_user agrs [2,..]=string
4548 return nil , errors .New (`models.User{}.FirstAll: admin_or_user agrs [2]{"admin", "user"}=string` )
@@ -175,7 +178,15 @@ func (user User) Delete(db *gorm.DB, id int) error {
175178 return nil
176179}
177180
178- // TODO: is?
179- func (User ) isAdmin (admin_or_user ... string ) bool {
180- return len (admin_or_user ) == 0 || (len (admin_or_user ) == 1 && admin_or_user [0 ] == "all" )
181+ // is? all, admin or user?
182+ func isAll (admin_or_user * []string ) bool {
183+ return len (* admin_or_user ) == 0 || len (* admin_or_user ) == 1 && (* admin_or_user )[0 ] == "all"
184+ }
185+
186+ func isAdmin (admin_or_user * []string ) bool {
187+ return len (* admin_or_user ) == 1 && (* admin_or_user )[0 ] == "admin"
188+ }
189+
190+ func isUser (admin_or_user * []string ) bool {
191+ return len (* admin_or_user ) == 1 && (* admin_or_user )[0 ] == "user"
181192}
0 commit comments