feat:Add token when login and get current user

Author: nk-akun

api/entrance.go

@@ -40,6 +40,7 @@ func AppRun() {
 		// user.Use()
 		user.POST("/register", guest.RegisterByEmail)
 		user.POST("/login", guest.Login)
+		user.GET("/currentUser", guest.GetCurrentUser)
 
 		user.POST("/topic", guest.PostArticle)
 		user.GET("/topics", guest.GetArticleList)

api/guest/users.go

@@ -4,9 +4,15 @@ import (
 	"fmt"
 
 	"github.com/gin-gonic/gin"
+	"github.com/nk-akun/NeighborBBS/model"
 	"github.com/nk-akun/NeighborBBS/service"
 )
 
+// GetCurrentUser ...
+func GetCurrentUser(c *gin.Context) {
+	service.UserService.GetCurrentUser(c)
+}
+
 // RegisterByEmail ...
 func RegisterByEmail(c *gin.Context) {
 	user, err := service.UserService.SignUp(c)
@@ -23,7 +29,9 @@ func Login(c *gin.Context) {
 	if err != nil {
 		setAPIResponse(c, nil, err.Error(), false)
 	} else {
-		setAPIResponse(c, user, "登录成功", true)
+		token := service.UserService.SetToken(user.ID)
+		value := model.NewResponseValue().Set("token", token).Set("user", user)
+		setAPIResponse(c, value, "登录成功", true)
 	}
 }
 

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/mitchellh/mapstructure v1.3.3 // indirect
 	github.com/natefinch/lumberjack v2.0.0+incompatible
 	github.com/pelletier/go-toml v1.8.1 // indirect
+	github.com/satori/go.uuid v1.2.0 // indirect
 	github.com/spf13/afero v1.4.0 // indirect
 	github.com/spf13/cast v1.3.1 // indirect
 	github.com/spf13/jwalterweatherman v1.1.0 // indirect

go.sum

@@ -199,6 +199,8 @@ github.com/prometheus/tsdb v0.7.1/go.mod h1:qhTCs0VvXwvX/y3TZrWD7rabWM+ijKTux40T
 github.com/rogpeppe/fastuuid v0.0.0-20150106093220-6724a57986af/go.mod h1:XWv6SoW27p1b0cqNHllgS5HIMJraePCO15w5zCzIWYg=
 github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4=
 github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb6uqfes/u+d4ooFouqFdy9/2g9QGwK3SQygK0Ts=
+github.com/satori/go.uuid v1.2.0 h1:0uYX9dsZ2yD7q2RtLRtPSdGDWzjeM3TbMJP9utgA0ww=
+github.com/satori/go.uuid v1.2.0/go.mod h1:dA0hQrYB0VpLJoorglMZABFdXlWrHn1NEOzdhQKdks0=
 github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=
 github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
 github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=

model/constant.go

@@ -8,4 +8,6 @@ const (
 	CTXAPIResponseValue   = "api_response_value"
 	CTXAPIResponseMessage = "api_response_message"
 	CTXAPIResponseSuccess = "api_response_success"
+
+	TokenExpireDays = 2
 )

model/model.go

@@ -33,6 +33,15 @@ type User struct {
 	DeleteTime            int64  `gorm:"column:delete_time;default:null" json:"delete_time"`
 }
 
+// UserToken stores user's token
+type UserToken struct {
+	Model
+	UserID     int64  `gorm:"column:user_id;type:int" json:"user_id"`
+	Token      string `gorm:"size:32;unique;not null" json:"token"`
+	ExpiredAt  int64  `gorm:"column:expired_at;type:int" json:"expired_at"`
+	CreateTime int64  `gorm:"column:create_time;default:null" json:"create_time"`
+}
+
 // Article stores article infomation
 type Article struct {
 	Model

model/response.go

@@ -69,3 +69,21 @@ type CommentInfo struct {
 	LikeCount  int    `json:"like_count"`
 	CreateTime int64  `json:"create_time"`
 }
+
+// ResponseValue ...
+type ResponseValue struct {
+	Value map[string]interface{}
+}
+
+// NewResponseValue ...
+func NewResponseValue() *ResponseValue {
+	return &ResponseValue{
+		Value: make(map[string]interface{}),
+	}
+}
+
+// Set set data into ResponseValue
+func (r *ResponseValue) Set(name string, data interface{}) *ResponseValue {
+	r.Value[name] = data
+	return r
+}

repository/token_repo.go

@@ -0,0 +1,36 @@
+package repository
+
+import (
+	"github.com/nk-akun/NeighborBBS/logs"
+	"github.com/nk-akun/NeighborBBS/model"
+	"gorm.io/gorm"
+)
+
+type userTokenRepository struct {
+}
+
+// UserTokenRepository is the entrance as a convenient interface
+var UserTokenRepository = newUserTokenRepository()
+
+func newUserTokenRepository() *userTokenRepository {
+	return new(userTokenRepository)
+}
+
+func (r *userTokenRepository) Create(db *gorm.DB, userToken *model.UserToken) error {
+	return db.Create(userToken).Error
+}
+
+func (r *userTokenRepository) GetUserIDByToken(db *gorm.DB, token string) (*model.UserToken, error) {
+	return r.take(db, "token = ?", token)
+}
+
+func (r *userTokenRepository) take(db *gorm.DB, column string, value interface{}) (*model.UserToken, error) {
+	result := &model.UserToken{}
+	// err := db.Where(column, value).Take(result).Error
+	err := db.Where(column, value).Find(&result).Error
+	if err != nil {
+		logs.Logger.Errorf("query db error:", err)
+		return nil, err
+	}
+	return result, nil
+}

service/users.go

@@ -2,12 +2,14 @@ package service
 
 import (
 	"errors"
+	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/nk-akun/NeighborBBS/logs"
 	"github.com/nk-akun/NeighborBBS/model"
 	"github.com/nk-akun/NeighborBBS/repository"
 	"github.com/nk-akun/NeighborBBS/util"
+	uuid "github.com/satori/go.uuid"
 	"golang.org/x/crypto/bcrypt"
 )
 
@@ -21,6 +23,30 @@ func newUserService() *userService {
 	return new(userService)
 }
 
+// GetCurrentUser ...
+func (s *userService) GetCurrentUser(c *gin.Context) *model.User {
+	token := s.GetToken(c)
+	userToken, err := repository.UserTokenRepository.GetUserIDByToken(util.DB(), token)
+	if err != nil {
+		logs.Logger.Errorf("数据库查询token出错")
+		return nil
+	}
+	if userToken == nil || userToken.ExpiredAt < util.NowTimestamp() { // 不存在或者过期了
+		return nil
+	}
+	user, err := repository.UserRepository.GetUserByUserID(util.DB(), userToken.UserID)
+	if err != nil {
+		logs.Logger.Errorf("数据库查询user出错")
+		return nil
+	}
+	return user
+}
+
+func (s *userService) GetToken(c *gin.Context) string {
+	token := c.GetHeader("X-User-Token")
+	return token
+}
+
 func (s *userService) SignUp(c *gin.Context) (*model.User, error) {
 	req := getReqFromContext(c).(*model.RegisterRequest)
 
@@ -80,7 +106,7 @@ func (s *userService) SignUp(c *gin.Context) (*model.User, error) {
 	// })
 }
 
-func (s *userService) Login(c *gin.Context) (*model.User, error) {
+func (s *userService) Login(c *gin.Context) (user *model.User, err error) {
 	req := getReqFromContext(c).(*model.LoginRequest)
 	if req.Email == "" && req.Username == "" || req.Email != "" && req.Username != "" {
 		return nil, errors.New("请使用用户名或邮箱二者之一登录")
@@ -89,7 +115,6 @@ func (s *userService) Login(c *gin.Context) (*model.User, error) {
 		return s.loginByEmail(req.Email, req.Password)
 	}
 	return s.loginByUsername(req.Username, req.Password)
-
 }
 
 func (s *userService) loginByEmail(email string, password string) (*model.User, error) {
@@ -113,3 +138,16 @@ func (s *userService) loginByUsername(username string, password string) (*model.
 	}
 	return user, nil
 }
+
+func (s *userService) SetToken(userID int64) string {
+	token := uuid.NewV4().String()
+	expireTime := time.Now().Add(time.Hour * 24 * time.Duration(model.TokenExpireDays))
+	userToken := &model.UserToken{
+		UserID:     userID,
+		Token:      token,
+		ExpiredAt:  util.Timestamp(expireTime),
+		CreateTime: util.NowTimestamp(),
+	}
+	repository.UserTokenRepository.Create(util.DB(), userToken)
+	return token
+}