Add: set user acitve

This commit is contained in:
2021-12-13 14:20:36 +08:00
parent adee9bcb65
commit ab67575976
10 changed files with 199 additions and 11 deletions

View File

@@ -22,7 +22,7 @@ func (database *Database) LoginAsAnonymous() (*User, error) {
return user, nil
}
func (database *Database) Register(username string, password string, usertype int64) (error) {
func (database *Database) Register(username string, password string, usertype int64) error {
countAdmin, err := database.CountAdmin()
if err != nil {
return err
@@ -64,3 +64,31 @@ func (database *Database) CountAdmin() (int64, error) {
}
return count, nil
}
func (database *Database) GetUsers() ([]*User, error) {
users := make([]*User, 0)
rows, err := database.stmt.getUsers.Query()
if err != nil {
return nil, err
}
defer rows.Close()
for rows.Next() {
user := &User{}
err = rows.Scan(&user.ID, &user.Username, &user.Role, &user.Active, &user.AvatarId)
if err != nil {
return nil, err
}
users = append(users, user)
}
return users, nil
}
func (database *Database) UpdateUserActive(id int64, active bool) error {
_, err := database.stmt.updateUserActive.Exec(active, id)
if err != nil {
return err
}
return nil
}