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

+ 10 - 1
cmd/apimain.go

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