aa

services/user.go

@@ -27,6 +27,16 @@ type UserService struct {
 }
 
 type UserServiceInterface interface {
+	Login(name, password string) (loginResult *LoginResult, err error)
+	RegisterUser(c *gin.Context, name, password string) (err error)
+}
+
+var (
+	UserServiceHandler UserServiceInterface
+)
+
+func InitHandler(handler UserServiceInterface) {
+	UserServiceHandler = handler
 }
 
 func (c *UserService) GetUsers(name, mobile, userID string) (users []*model.User, err error) {
@@ -57,9 +67,9 @@ type LoginResult struct {
 	Token string `json:"token"` //token
 }
 
-func (c *UserService) Login(name, password string) (loginResult *LoginResult, err error) {
+func Login(c *gin.Context, name, password string) (loginResult *LoginResult, err error) {
 	var (
-		db    = c.DB
+		db    = globals.GetDB()
 		now   = time.Now()
 		user  = &model.User{}
 		token string
@@ -81,7 +91,7 @@ func (c *UserService) Login(name, password string) (loginResult *LoginResult, er
 	loginResult.Token = token
 	//更新登录时间和ip
 	user.LastLoginAt = &now
-	user.LastLoginIP = c.C.ClientIP()
+	user.LastLoginIP = c.ClientIP()
 	err = dao.Update(db, user, "last_login_at", "last_login_ip")
 	return loginResult, err
 }