init commit

water/service/auth_service.go

@@ -0,0 +1,94 @@
+package service
+
+import (
+	"errors"
+	providerApiEntity "flswld.com/annie-user-api/entity"
+	"flswld.com/common/utils/endec"
+	uuid "github.com/satori/go.uuid"
+	"strconv"
+	"time"
+	"water/entity"
+)
+
+func (s *Service) generateAccessToken(user *providerApiEntity.User) (*entity.AccessToken, error) {
+	accessToken, err := s.dao.InsertAccessToken(&entity.AccessToken{
+		Uid:         user.Uid,
+		Username:    user.Username,
+		AccessToken: uuid.NewV4().String(),
+		CreateTime:  strconv.FormatInt(time.Now().Unix(), 10),
+	})
+	return accessToken, err
+}
+
+func (s *Service) generateRefreshToken(user *providerApiEntity.User) (*entity.RefreshToken, error) {
+	refreshToken, err := s.dao.InsertRefreshToken(&entity.RefreshToken{
+		Uid:          user.Uid,
+		Username:     user.Username,
+		RefreshToken: uuid.NewV4().String(),
+		CreateTime:   strconv.FormatInt(time.Now().Unix(), 10),
+	})
+	return refreshToken, err
+}
+
+func (s *Service) LoginAuth(user *providerApiEntity.User) (string, string, error) {
+	userList := make([]providerApiEntity.User, 0)
+	// 用户服务
+	ok := s.rpcUserConsumer.CallFunction("RpcService", "RpcQueryUser", &providerApiEntity.User{Username: user.Username}, &userList)
+	if !ok || len(userList) != 1 {
+		return "", "", errors.New("query username error")
+	}
+	if userList[0].Password != endec.Md5Str(user.Password) {
+		return "", "", errors.New("password error")
+	}
+	accessToken, err := s.generateAccessToken(&providerApiEntity.User{Uid: userList[0].Uid, Username: userList[0].Username})
+	if err != nil {
+		return "", "", errors.New("generate access token error")
+	}
+	refreshToken, err := s.generateRefreshToken(&providerApiEntity.User{Uid: userList[0].Uid, Username: userList[0].Username})
+	if err != nil {
+		return "", "", errors.New("generate refresh token error")
+	}
+	return accessToken.AccessToken, refreshToken.RefreshToken, nil
+}
+
+func (s *Service) VerifyAccessToken(accessToken string) bool {
+	tokenList, err := s.dao.QueryAccessToken(&entity.AccessToken{AccessToken: accessToken})
+	if err != nil || len(tokenList) == 0 {
+		return false
+	}
+	createTime, err := strconv.ParseInt(tokenList[0].CreateTime, 10, 64)
+	if err != nil || (time.Now().Unix()-createTime) > 1800 {
+		return false
+	}
+	return true
+}
+
+func (s *Service) QueryUserByAccessToken(accessToken string) (*providerApiEntity.User, error) {
+	tokenList, err := s.dao.QueryAccessToken(&entity.AccessToken{AccessToken: accessToken})
+	if err != nil || len(tokenList) != 1 {
+		return nil, errors.New("query access token error")
+	}
+	userList := make([]providerApiEntity.User, 0)
+	// 用户服务
+	ok := s.rpcUserConsumer.CallFunction("RpcService", "RpcQueryUser", &providerApiEntity.User{Uid: tokenList[0].Uid}, &userList)
+	if !ok || len(userList) != 1 {
+		return nil, errors.New("query user error")
+	}
+	return &(userList[0]), nil
+}
+
+func (s *Service) RefreshToken(refreshToken string) (string, error) {
+	tokenList, err := s.dao.QueryRefreshToken(&entity.RefreshToken{RefreshToken: refreshToken})
+	if err != nil || len(tokenList) == 0 {
+		return "", errors.New("query refresh token error")
+	}
+	createTime, err := strconv.ParseInt(tokenList[0].CreateTime, 10, 64)
+	if err != nil || (time.Now().Unix()-createTime) > 24*3600 {
+		return "", errors.New("refresh token overtime")
+	}
+	accessToken, err := s.generateAccessToken(&providerApiEntity.User{Uid: tokenList[0].Uid, Username: tokenList[0].Username})
+	if err != nil {
+		return "", errors.New("generate access token error")
+	}
+	return accessToken.AccessToken, nil
+}