hotgo/server/utility/simple/simple.go

// Package simple
// @Link  https://github.com/bufanyun/hotgo
// @Copyright  Copyright (c) 2023 HotGo CLI
// @Author  Ms <133814250@qq.com>
// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
package simple

import (
	"context"
	"github.com/gogf/gf/v2/crypto/gmd5"
	"github.com/gogf/gf/v2/encoding/gbase64"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/glog"
	"github.com/gogf/gf/v2/os/grpool"
	"github.com/gogf/gf/v2/util/gconv"
	"hotgo/internal/consts"
	"hotgo/utility/encrypt"
)

// DecryptText 解密文本
func DecryptText(text string) (string, error) {
	str, err := gbase64.Decode([]byte(text))
	if err != nil {
		return "", err
	}

	str, err = encrypt.AesECBDecrypt(str, consts.RequestEncryptKey)
	if err != nil {
		return "", err
	}

	return string(str), nil
}

// CheckPassword 检查密码
func CheckPassword(input, salt, hash string) (err error) {
	// 解密密码
	password, err := DecryptText(input)
	if err != nil {
		return err
	}

	if hash != gmd5.MustEncryptString(password+salt) {
		err = gerror.New("用户密码不正确")
		return
	}
	return
}

// SafeGo 安全的调用协程，遇到错误时输出错误日志而不是抛出panic
func SafeGo(ctx context.Context, f func(ctx context.Context), level ...interface{}) {
	var newLevel = glog.LEVEL_ERRO
	if len(level) > 0 {
		newLevel = gconv.Int(level[0])
	}

	err := grpool.AddWithRecover(ctx, func(ctx context.Context) {
		f(ctx)
	}, func(ctx context.Context, err error) {
		Logf(newLevel, ctx, "SafeGo exec failed:%+v", err)
	})

	if err != nil {
		Logf(newLevel, ctx, "SafeGo AddWithRecover err:%+v", err)
		return
	}
}

func Logf(level int, ctx context.Context, format string, v ...interface{}) {
	switch level {
	case glog.LEVEL_DEBU:
		g.Log().Debugf(ctx, format, v)
	case glog.LEVEL_INFO:
		g.Log().Infof(ctx, format, v)
	case glog.LEVEL_NOTI:
		g.Log().Noticef(ctx, format, v)
	case glog.LEVEL_WARN:
		g.Log().Warningf(ctx, format, v)
	case glog.LEVEL_ERRO:
		g.Log().Errorf(ctx, format, v)
	case glog.LEVEL_CRIT:
		g.Log().Critical(ctx, format, v)
	case glog.LEVEL_PANI:
		g.Log().Panicf(ctx, format, v)
	case glog.LEVEL_FATA:
		g.Log().Fatalf(ctx, format, v)
	default:
		g.Log().Error(ctx, "Logf level not find")
	}
}

func Log(level int, ctx context.Context, v ...interface{}) {
	switch level {
	case glog.LEVEL_DEBU:
		g.Log().Debug(ctx, v)
	case glog.LEVEL_INFO:
		g.Log().Info(ctx, v)
	case glog.LEVEL_NOTI:
		g.Log().Notice(ctx, v)
	case glog.LEVEL_WARN:
		g.Log().Warning(ctx, v)
	case glog.LEVEL_ERRO:
		g.Log().Error(ctx, v)
	case glog.LEVEL_CRIT:
		g.Log().Critical(ctx, v)
	case glog.LEVEL_PANI:
		g.Log().Panic(ctx, v)
	case glog.LEVEL_FATA:
		g.Log().Fatal(ctx, v)
	default:
		g.Log().Error(ctx, "Logf level not find")
	}
}