ZiHangQin/electricity_bill_calc_service
1
0
Fork 0
forked from free-lancers/electricity_bill_calc_service
Code Pull Requests Activity

fix(user):增加对于用户是否被禁用的判断。

Browse Source
This commit is contained in:
徐涛
2022-08-12 17:18:11 +08:00
parent fd9de54262
commit a9f281d4df
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
service/user.go
View File

@@ -31,6 +31,9 @@ func (_UserService) ProcessEnterpriseUserLogin(username, password string) (*mode
if user.Type != 0 { if user.Type != 0 {
return nil, exceptions.NewAuthenticationError(401, "用户类型不正确。") return nil, exceptions.NewAuthenticationError(401, "用户类型不正确。")
} }
if !user.Enabled {
return nil, exceptions.NewAuthenticationError(401, "用户已被禁用。")
}
hash := sha512.New512_256() hash := sha512.New512_256()
hash.Write([]byte(password)) hash.Write([]byte(password))
hashedPassword := fmt.Sprintf("%x", hash.Sum(nil)) hashedPassword := fmt.Sprintf("%x", hash.Sum(nil))
@@ -69,6 +72,9 @@ func (_UserService) ProcessManagementUserLogin(username, password string) (*mode
if user.Type != 1 && user.Type != 2 { if user.Type != 1 && user.Type != 2 {
return nil, exceptions.NewAuthenticationError(401, "用户类型不正确。") return nil, exceptions.NewAuthenticationError(401, "用户类型不正确。")
} }
if !user.Enabled {
return nil, exceptions.NewAuthenticationError(401, "用户已被禁用。")
}
hash := sha512.New512_256() hash := sha512.New512_256()
hash.Write([]byte(password)) hash.Write([]byte(password))
hashedPassword := fmt.Sprintf("%x", hash.Sum(nil)) hashedPassword := fmt.Sprintf("%x", hash.Sum(nil))