lejianwen месяцев назад: 11
Родитель
Сommit
0a78c206de
1 измененных файлов с 10 добавлено и 1 удалено
  1. 10 1
      cmd/apimain.go

+ 10 - 1
cmd/apimain.go

@@ -51,6 +51,10 @@ var resetPwdCmd = &cobra.Command{
51
 	Run: func(cmd *cobra.Command, args []string) {
51
 	Run: func(cmd *cobra.Command, args []string) {
52
 		pwd := args[0]
52
 		pwd := args[0]
53
 		admin := service.AllService.UserService.InfoById(1)
53
 		admin := service.AllService.UserService.InfoById(1)
54
+		if admin.Id == 0 {
55
+			global.Logger.Warn("user not found! ")
56
+			return
57
+		}
54
 		err := service.AllService.UserService.UpdatePassword(admin, pwd)
58
 		err := service.AllService.UserService.UpdatePassword(admin, pwd)
55
 		if err != nil {
59
 		if err != nil {
56
 			global.Logger.Error("reset password fail! ", err)
60
 			global.Logger.Error("reset password fail! ", err)
@@ -77,6 +81,10 @@ var resetUserPwdCmd = &cobra.Command{
77
 			return
81
 			return
78
 		}
82
 		}
79
 		u := service.AllService.UserService.InfoById(uint(uid))
83
 		u := service.AllService.UserService.InfoById(uint(uid))
84
+		if u.Id == 0 {
85
+			global.Logger.Warn("user not found! ")
86
+			return
87
+		}
80
 		err = service.AllService.UserService.UpdatePassword(u, pwd)
88
 		err = service.AllService.UserService.UpdatePassword(u, pwd)
81
 		if err != nil {
89
 		if err != nil {
82
 			global.Logger.Warn("reset password fail! ", err)
90
 			global.Logger.Warn("reset password fail! ", err)
@@ -144,7 +152,6 @@ func InitGlobal() {
144
 			MaxOpenConns: global.Config.Gorm.MaxOpenConns,
152
 			MaxOpenConns: global.Config.Gorm.MaxOpenConns,
145
 		})
153
 		})
146
 	}
154
 	}
147
-	DatabaseAutoUpdate()
148
 
155
 
149
 	//validator
156
 	//validator
150
 	global.ApiInitValidator()
157
 	global.ApiInitValidator()
@@ -167,6 +174,8 @@ func InitGlobal() {
167
 
174
 
168
 	//service
175
 	//service
169
 	service.New(&global.Config, global.DB, global.Logger, global.Jwt, global.Lock)
176
 	service.New(&global.Config, global.DB, global.Logger, global.Jwt, global.Lock)
177
+
178
+	DatabaseAutoUpdate()
170
 }
179
 }
171
 func DatabaseAutoUpdate() {
180
 func DatabaseAutoUpdate() {
172
 	version := 262
181
 	version := 262