token/handler/user-handler.go

package handler

import (
	"encoding/json"
	"io/ioutil"
	"strconv"
	"time"

	"github.com/dgrijalva/jwt-go"
	"github.com/gin-gonic/gin"
	log "github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"myschools.me/wyh/token.git/config"
	"myschools.me/wyh/token.git/model"
	"myschools.me/wyh/token.git/mysql"
)

// 获取当前用户
func userCurrent(c *gin.Context) *model.UserClaims {
	obj, ok := c.Get("user")
	if !ok {
		return nil
	}
	return obj.(*model.UserClaims)
}

func UserToken(c *gin.Context) {

	reqbody, _ := ioutil.ReadAll(c.Request.Body)
	log.WithFields(log.Fields{
		"func": "UserToken",
	}).Infof("%s", string(reqbody))

	var up = &model.User{}
	if err := json.Unmarshal(reqbody, up); err != nil {
		log.WithFields(log.Fields{
			"func": "UserToken",
		}).Warnf("%s", err)
	}
	db := mysql.New()
	var result = &model.User{}
	if err := db.Where("uname=? and password=?", up.Uname, up.Password).First(result).Error; err != nil {
		if err != gorm.ErrRecordNotFound {
			log.WithFields(log.Fields{
				"func": "UserToken",
			}).Warnf("%s", err)
		}
	}
	if result.Uid == 0 {
		c.AbortWithStatusJSON(400, gin.H{
			"code": 1,
			"msg":  "用户账号密码错误",
			"data": "",
		})
		return
	}
	var claims = &model.UserClaims{
		StandardClaims: jwt.StandardClaims{
			ExpiresAt: time.Now().Add(time.Minute * 5).Unix(),
			Id:        strconv.Itoa(result.Uid),
		},
		User: *up,
	}
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	tokenString, err := token.SignedString([]byte(config.JwtKey))
	if err != nil {
		log.WithFields(log.Fields{
			"func": "UserToken",
		}).Warnf("%s", err)
	}
	c.JSON(200, gin.H{
		"good": tokenString,
	})
}

func UserDetail(c *gin.Context) {
	usr := userCurrent(c)
	c.JSON(200, usr)
}