aa

services/user.go

@@ -10,6 +10,7 @@ import (
 	"git.rosy.net.cn/jx-print/model"
 	putils "git.rosy.net.cn/jx-print/utils"
 	"github.com/gin-gonic/gin"
+	"github.com/jmoiron/sqlx"
 	"strings"
 	"time"
 )
@@ -20,7 +21,15 @@ const (
 	TokenTypeSep = "."
 )
 
-func GetUsers(c *gin.Context, name, mobile, userID string) (users []*model.User, err error) {
+type UserService struct {
+	C  *gin.Context
+	DB *sqlx.DB
+}
+
+type UserServiceInterface interface {
+}
+
+func (c *UserService) GetUsers(name, mobile, userID string) (users []*model.User, err error) {
 	return dao.GetUsers(globals.GetDB(), userID, name, mobile)
 }
 
@@ -48,9 +57,9 @@ type LoginResult struct {
 	Token string `json:"token"` //token
 }
 
-func Login(c *gin.Context, name, password string) (loginResult *LoginResult, err error) {
+func (c *UserService) Login(name, password string) (loginResult *LoginResult, err error) {
 	var (
-		db    = globals.GetDB()
+		db    = c.DB
 		now   = time.Now()
 		user  = &model.User{}
 		token string
@@ -70,7 +79,7 @@ func Login(c *gin.Context, name, password string) (loginResult *LoginResult, err
 	loginResult.Token = token
 	//更新登录时间和ip
 	user.LastLoginAt = &now
-	user.LastLoginIP = c.ClientIP()
+	user.LastLoginIP = c.C.ClientIP()
 	err = dao.Update(db, user, "last_login_at", "last_login_ip")
 	return loginResult, err
 }