lc_veterans/app/user/controller/user.go

package controller

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"iot_manager_service/app/user/model"
	"iot_manager_service/app/user/service"
	"iot_manager_service/util/cache"
	"iot_manager_service/util/common"
	"iot_manager_service/util/token"
	"iot_manager_service/util/websocket"
	"net/http"
	"strconv"
)

var UserController = new(user)

type user struct{}

func (u *user) Login(c *gin.Context) {
	loginUser := model.LoginUser{}
	if err := c.ShouldBindJSON(&loginUser); err != nil {
		c.JSON(http.StatusOK, common.ParamsInvalidResponse(err.Error(), nil))
		return
	}
	info := service.UserService.Login(loginUser.Account, loginUser.PassWord)
	c.JSON(http.StatusOK, info)
}

func (u *user) GetPublicKey(c *gin.Context) {
	key := service.UserService.GetPublicKey()
	c.JSON(http.StatusOK, key)
}

func (u *user) GetDeptUsers(c *gin.Context) {
	depts := model.ReqDepts{}
	err := c.ShouldBindJSON(&depts)
	if err != nil {
		c.JSON(http.StatusOK, common.ParamsInvalidResponse(err.Error(), nil))
		return
	}
	users := service.UserService.GetDeptUsers(depts.Depts)
	c.JSON(http.StatusOK, users)
}

func (u *user) GetNewTokens(c *gin.Context) {
	header := c.GetHeader("Authorization")
	claims, _ := token.JwtClaims.ParseJwtToken(header)
	nowUser, _ := cache.GetNowUser(claims.ID)

	UUID := uuid.New().String()
	accessToken, _ := token.JwtClaims.CreateJwtToken(strconv.Itoa(nowUser.User.ID), nowUser.User.UserName, UUID, 1)
	refreshToken, _ := token.JwtClaims.CreateJwtToken(strconv.Itoa(nowUser.User.ID), nowUser.User.UserName, UUID, 24*7)
	tokens := make(map[string]interface{})

	tokens["accessToken"] = accessToken
	tokens["refreshToken"] = refreshToken

	c.JSON(http.StatusOK, common.SuccessResponse("获取成功", tokens))
}

func (u *user) Logout(c *gin.Context) {
	header := c.GetHeader("Authorization")
	claims, _ := token.JwtClaims.ParseJwtToken(header)
	err := cache.DeleteToken(claims.ID)
	if err != nil {
		c.JSON(http.StatusOK, common.ParamsInvalidResponse(err.Error(), nil))
		return
	}
	//关闭websocket连接
	websocket.CloseConn(claims.ID)
	c.JSON(http.StatusOK, common.SuccessResponse("退出成功", nil))
}

func (u *user) GetUser(c *gin.Context) {
	fmt.Println("demo。。。")
}