@@ -8,29 +8,27 @@ import (
88 "labelproject-back/model"
99 "labelproject-back/util"
1010 "log"
11- "net/http"
1211 "strconv"
1312 "strings"
1413 "time"
1514
1615 "github.com/gin-gonic/gin"
1716)
1817
19- func Login (ctx * gin.Context ) {
18+ type userDto struct {
19+ Username string `json:"username"`
20+ UserID int64 `json:"userId"`
21+ Authorities string `json:"authorities"`
22+ }
2023
21- type userDto struct {
22- Username string `json:"username"`
23- UserID int64 `json:"userId"`
24- Authorities string `json:"authorities"`
25- }
24+ // Login check the username & password. If everything is right, writing successful message to context.
25+ func Login (ctx * gin.Context ) {
2626
27- db := common .GetDB ()
28- cache := common .GetCache ()
29- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
27+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
28+ redisUtilInstance := util .RedisUtilInstance (common .GetCache ())
3029
31- //使用结构体
3230 var requestMap = model.User {}
33- json .NewDecoder (ctx .Request .Body ).Decode (& requestMap ) //其中一种
31+ json .NewDecoder (ctx .Request .Body ).Decode (& requestMap )
3432
3533 user , err := adminUserReposityInstance .FindByUserName (requestMap .Username )
3634 if err != nil {
@@ -42,38 +40,35 @@ func Login(ctx *gin.Context) {
4240 util .ManagerInstance .FailWithoutData (ctx , "密码错误" )
4341 return
4442 }
45- //TOKEN
46- token , err := middleware .ReleaseToken (ctx , user ) //发放token
43+
44+ token , err := middleware .ReleaseToken (ctx , user )
4745 if err != nil {
4846 util .ManagerInstance .FailWithoutData (ctx , "系统异常" )
4947 return
5048 }
5149
52- cookie := http.Cookie {Name : "request_token" , Value : "6MIhycayVQizGoweGhRvUFVARhAARiTyJ1NS6YNfiuQJ1ZHU" , Expires : time .Now ().AddDate (0 , 0 , 1 )}
53- http .SetCookie (ctx .Writer , & cookie )
50+ // cookie := http.Cookie{Name: "request_token", Value: "6MIhycayVQizGoweGhRvUFVARhAARiTyJ1NS6YNfiuQJ1ZHU", Expires: time.Now().AddDate(0, 0, 1)}
51+ // http.SetCookie(ctx.Writer, &cookie)
5452
55- redisUtilInstance := util .RedisUtilInstance (cache )
56- log .Println (ctx .Request .RemoteAddr )
5753 err = redisUtilInstance .AddTokenTORedis (token , requestMap .Username , ctx .Request .RemoteAddr )
58- //返回结果
54+ if err != nil {
55+ util .ManagerInstance .FailWithoutData (ctx , "add token to redis error" )
56+ return
57+ }
5958
60- // util.Success(ctx, gin.H{"user": model.ToUserDto(user), "token": token}, "SUCCESS")
6159 util .Success (ctx , gin.H {"user" : userDto {
6260 Username : user .Username ,
6361 UserID : user .UserID ,
6462 Authorities : user .Authorities ,
6563 }, "token" : token }, "SUCCESS" )
66- // util.Success(ctx, gin.H{"token": token}, "SUCCESS")
6764
6865 log .Println ("登录成功!" )
69-
7066}
7167
7268func Logout (ctx * gin.Context ) {
73- Authorization := ctx .GetHeader ("Authorization" )
7469
75- cache := common . GetCache ( )
76- redisUtilInstance := util .RedisUtilInstance (cache )
70+ Authorization := ctx . GetHeader ( "Authorization" )
71+ redisUtilInstance := util .RedisUtilInstance (common . GetCache () )
7772
7873 if Authorization != "" && strings .HasPrefix (Authorization , "Bearer " ) {
7974 authToken := strings .TrimLeft (Authorization , "Bearer " )
@@ -84,8 +79,7 @@ func Logout(ctx *gin.Context) {
8479}
8580
8681func GetCount (ctx * gin.Context ) {
87- db := common .GetDB ()
88- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
82+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
8983 taskCout , _ := adminUserReposityInstance .GetTaskCount ()
9084 reviewerCount , _ := adminUserReposityInstance .GetReviewerCount ()
9185 userCount , _ := adminUserReposityInstance .GetUserCount ()
@@ -95,8 +89,7 @@ func GetCount(ctx *gin.Context) {
9589
9690// GetUserList
9791func GetUserList (ctx * gin.Context ) {
98- db := common .GetDB ()
99- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
92+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
10093 users , _ := adminUserReposityInstance .GetUserList ()
10194 util .Success (ctx , gin.H {"userList" : users }, "SUCCESS" )
10295}
@@ -111,8 +104,7 @@ func EditUser(ctx *gin.Context) {
111104 }
112105 log .Println ("User Information: " , user .Username )
113106
114- db := common .GetDB ()
115- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
107+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
116108 err := adminUserReposityInstance .EditUser (user )
117109 if err != nil {
118110 util .ManagerInstance .FailWithoutData (ctx , "Edit User Error!!!" )
@@ -134,8 +126,7 @@ func AddUser(ctx *gin.Context) {
134126 }
135127 log .Println ("User Information: " , user .Username )
136128
137- db := common .GetDB ()
138- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
129+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
139130 err := adminUserReposityInstance .AddUser (user )
140131 if err != nil {
141132 util .ManagerInstance .FailWithoutData (ctx , "Add User Error!!!" )
@@ -158,8 +149,7 @@ func DeleteUser(ctx *gin.Context) {
158149 util .ManagerInstance .FailWithoutData (ctx , "Parameter Error : Bind User Wrong!!" )
159150 return
160151 }
161- db := common .GetDB ()
162- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
152+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
163153 user , err := adminUserReposityInstance .GetUserByID (tempuser .UserID )
164154 if err != nil {
165155 util .ManagerInstance .FailWithoutData (ctx , "Parameter Error : Can't Find the User By ID!!" )
@@ -192,8 +182,7 @@ func GetPendingUserList(ctx *gin.Context) {
192182 return
193183 }
194184
195- db := common .GetDB ()
196- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
185+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
197186 users , err := adminUserReposityInstance .GetPendingUserList (imageID )
198187 if err != nil {
199188 util .ManagerInstance .FailWithoutData (ctx , "Get PendingUser List Error!!!" )
@@ -218,8 +207,7 @@ func GetVideoPendingUserList(ctx *gin.Context) {
218207 return
219208 }
220209
221- db := common .GetDB ()
222- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
210+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
223211 users , err := adminUserReposityInstance .GetVideoPendingUserList (imageID )
224212 if err != nil {
225213 util .ManagerInstance .FailWithoutData (ctx , "Get VideoPendingUser List Error!!!" )
@@ -233,8 +221,7 @@ func GetVideoPendingUserList(ctx *gin.Context) {
233221func GetListUser (ctx * gin.Context ) {
234222 log .Println ("Get List User" )
235223
236- db := common .GetDB ()
237- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
224+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
238225 users , err := adminUserReposityInstance .GetListUser ()
239226 if err != nil {
240227 util .ManagerInstance .FailWithoutData (ctx , "Get List User Error!!!" )
@@ -248,8 +235,7 @@ func GetListUser(ctx *gin.Context) {
248235func GetListReviewer (ctx * gin.Context ) {
249236 log .Println ("Get List User" )
250237
251- db := common .GetDB ()
252- adminUserReposityInstance := repository .AdminUserReposityInstance (db )
238+ adminUserReposityInstance := repository .AdminUserReposityInstance (common .GetDB ())
253239 users , err := adminUserReposityInstance .GetListReviewer ()
254240 if err != nil {
255241 util .ManagerInstance .FailWithoutData (ctx , "Get List User Error!!!" )
0 commit comments