opms_backend/opms_admin/app/handler/auth.go

package handler

import (
	"context"
	"dashoo.cn/common_definition/admin/user_def"
	"dashoo.cn/common_definition/comm_def"
	"dashoo.cn/micro/app/common/gtoken"
	"dashoo.cn/opms_libary/micro_srv"
	"github.com/gogf/gf/encoding/gjson"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/glog"
	"github.com/gogf/gf/util/gconv"

	"dashoo.cn/common_definition/auth"
)

type Auth struct{}

var gfToken *gtoken.GfToken

func init() {
	// 启动gtoken
	gfToken = &gtoken.GfToken{
		Timeout:    24 * 60 * 60 * 1000, // 24小时
		EncryptKey: []byte(g.Config().GetString("gtoken.encrypt-key")),
		CacheMode:  g.Config().GetInt8("gtoken.cache-mode"),
		MultiLogin: g.Config().GetBool("gtoken.multi-login"),
	}
	gfToken.Start()
}

// Login 用户登录验证
func (e *Auth) Login(ctx context.Context, req *user_def.LoginReq, rsp *comm_def.CommonMsg) error {
	// 参数验证
	if req.UserName == "" || req.Password == "" {
		return NoParamsErr
	}
	//userService, err := service.NewUserService(ctx)
	//if err != nil {
	//	g.Log().Error(err)
	//	return gerror.New("系统异常，请重新尝试")
	//}
	//userInfo, err := userService.Login(ctx, tenant, req.UserName, pridecrypt)
	//_, err, code, msg := myerrors.CheckError(err)
	//if err != nil {
	//	glog.Error(err)
	//	return err
	//}
	//rsp.Data = userInfo
	//rsp.Code = code
	//rsp.Msg = msg
	return nil
}

// LogOut 退出登录
func (e *Auth) LogOut(ctx context.Context, req interface{}, rsp *comm_def.CommonMsg) error {
	if token, err := micro_srv.GetToken(ctx); err != nil {
		return err
	} else if token == "" {
		return NoParamsErr
	}
	//userService, err := service.NewUserService(ctx)
	//if err != nil {
	//	g.Log().Error(err)
	//	return gerror.New("系统异常，请重新尝试")
	//}
	//err = userService.LogOut(ctx, token)
	//_, err, code, msg := myerrors.CheckError(err)
	//if err != nil {
	//	glog.Error(err)
	//	return err
	//}
	//rsp.Code = code
	//rsp.Msg = msg
	return nil
}

func (e *Auth) GetToken(ctx context.Context, req *auth.Request, rsp *auth.Response) (err error) {
	glog.Info("Received Auth.GetToken request, UserKey:", req.UserKey, " @ ", req.Tenant)
	// 获取Data，换成对象
	var json interface{}
	if req.Data != "" {
		json, err = gjson.DecodeToJson(req.Data)
		if err != nil {
			return err
		}
	}
	tokenResp := gfToken.GetOrGenToken(req.Tenant, req.UserKey, req.Uuid, json)
	if err := gconv.Struct(tokenResp, rsp); err != nil {
		return err
	}
	glog.Info(tokenResp.DataString())
	return nil
}

func (e *Auth) RemoveToken(ctx context.Context, reqToken string, rsp *auth.Response) error {
	glog.Info("Received Auth.RemoveToken request, token:", reqToken)
	tokenResp := gfToken.RemoveToken(reqToken)
	if err := gconv.Struct(tokenResp, rsp); err != nil {
		return err
	}
	return nil
}

func (e *Auth) ValidToken(ctx context.Context, reqToken string, rsp *auth.Response) error {
	glog.Info("Received Auth.ValidToken request, token:", reqToken)
	tokenResp := gfToken.ValidToken(reqToken)
	if err := gconv.Struct(tokenResp, rsp); err != nil {
		return err
	}

	return nil
}