pet-house/business/api/common.go

package api

import (
	"crypto/md5"
	"encoding/hex"
	"github.com/bwmarrin/snowflake"
	"github.com/kataras/iris/v12/context"
	"go.uber.org/zap"
	"pet-house.com/business/models"
	"pet-house.com/core/server/cache"
	"pet-house.com/core/server/cron_server"
	"pet-house.com/core/server/database"
	"pet-house.com/core/server/zap_server"
	"strconv"
	"time"
)

type Response struct {
	Code int    `json:"code"`           //code
	Msg  string `json:"msg"`            //msg
	Data any    `json:"data,omitempty"` //数据
}

type ResponseData struct {
	Data any `json:"data"` //数据
}

type Error struct {
	Code int    `json:"code"`
	Msg  string `json:"msg"`
}

type DefParty struct {
	Prefix string
}

var (
	ParamError                   = Error{Code: 201, Msg: "参数错误"}
	IllegalError                 = Error{Code: 202, Msg: "非法请求"}
	UserError                    = Error{Code: 203, Msg: "用户错误"}
	TokenError                   = Error{Code: 204, Msg: "Token失效，请重新登录"}
	UserNotExistError            = Error{Code: 205, Msg: "用户不存在"}
	PetNotExistError             = Error{Code: 206, Msg: "用户宠物不存在"}
	PetBaseNotExistError         = Error{Code: 207, Msg: "宠物基础信息不存在"}
	UserServiceAddrNotExistError = Error{Code: 207, Msg: "用户服务地址不存在"}
	ServiceAddrNotExistError     = Error{Code: 208, Msg: "服务地址区域，请重新配置地址"}
	NotInServiceExistError       = Error{Code: 209, Msg: "不在服务范围"}
	OrderCreateError             = Error{Code: 210, Msg: "订单创建失败"}
	LoginError                   = Error{Code: 211, Msg: "授权登录失败"}
)

func Success(ctx *context.Context, request any, data any) {
	response := Response{200, "success", data}
	reqTime, _ := ctx.Values().GetTime("reqTime")
	zap_server.ZAPLOG.Info(ctx.Path(), zap.Any("request", request), zap.Any("response", response), zap.Any("costTime", time.Now().Sub(reqTime).Milliseconds()))
	write(ctx, response)
}

func (error *Error) Fail(ctx *context.Context, request any) {
	zap_server.ZAPLOG.Info(ctx.Path(), zap.Any("request", request), zap.Any("error", error), zap.Any("err", error.Msg))
	write(ctx, Error{error.Code, error.Msg})
}

func (error *Error) DefFail(ctx *context.Context, request any, err any) {
	zap_server.ZAPLOG.Info(ctx.Path(), zap.Any("request", request), zap.Any("error", error), zap.Any("err", err))
	write(ctx, Error{error.Code, err.(string)})
}

func write(ctx *context.Context, resp any) {
	ctx.JSON(resp)
}

var NextId, _ = snowflake.NewNode(1)

func genToken(uid int64) string {
	uToken, err := cache.GetCacheString("u_uid_token:" + strconv.FormatInt(uid, 10))
	if err == nil && len(uToken) > 0 {
		return uToken
	}
	sId := NextId.Generate()
	hash := md5.Sum(sId.Bytes())
	token := hex.EncodeToString(hash[:])
	cache.SetCache("u_token:"+token, uid, time.Hour*24*30)
	cache.SetCache("u_uid_token:"+strconv.FormatInt(uid, 10), token, time.Hour*24*30)
	return token
}

func GetTokenInfo(token string) string {
	uId, _ := cache.GetCacheString("u_token:" + token)
	return uId
}

type HeaderBaseInfo struct {
	Token string
	Uid   int64
}

func GetHeaderBaseInfo(ctx *context.Context) HeaderBaseInfo {
	token := ctx.GetHeader("X-Token")
	uid := ctx.GetHeader("X-U-Id")
	_uid, _ := strconv.ParseInt(uid, 10, 64)
	return HeaderBaseInfo{
		token, _uid,
	}
}

type FrontExclude struct {
	path string
}

var PetBaseInfoMap map[int]models.PetBaseInfo
var ServiceAddrMap map[int64]models.ServiceAddr
var PetGoodsMap map[string][]models.PetGoods
var GoodsMap map[int64]models.Goods

func DataInit() {
	//--------------------------------------------------宠物基础信息数据---------------------------------------------------------
	var petBastInfoList []models.PetBaseInfo
	database.Instance().Model(&models.PetBaseInfo{}).Find(&petBastInfoList)
	PetBaseInfoMap = make(map[int]models.PetBaseInfo)
	for _, value := range petBastInfoList {
		PetBaseInfoMap[value.Id] = value
	}
	zap_server.ZAPLOG.Info("dataInit  petBastInfoMap : ", zap.Any("petBastInfoMap", PetBaseInfoMap))
	//--------------------------------------------------服务地址数据---------------------------------------------------------
	var serviceAddrList []models.ServiceAddr
	database.Instance().Model(&models.ServiceAddr{}).Find(&serviceAddrList)
	ServiceAddrMap = make(map[int64]models.ServiceAddr)
	for _, value := range serviceAddrList {
		ServiceAddrMap[value.Id] = value
	}
	zap_server.ZAPLOG.Info("dataInit  ServiceAddrMap : ", zap.Any("ServiceAddrMap", ServiceAddrMap))
	//--------------------------------------------------宠物商品关联数据---------------------------------------------------------
	var petGoodsList []models.PetGoods
	database.Instance().Model(&models.PetGoods{}).Find(&petGoodsList)
	PetGoodsMap = make(map[string][]models.PetGoods)
	for _, value := range petGoodsList {
		key := strconv.Itoa(value.Assortment) + strconv.Itoa(value.PetType) + strconv.Itoa(value.Size)
		if _, ok := PetGoodsMap[key]; ok {
			PetGoodsMap[key] = append(PetGoodsMap[key], value)
		} else {
			PetGoodsMap[key] = []models.PetGoods{value}
		}
	}
	zap_server.ZAPLOG.Info("dataInit  petGoodsMap : ", zap.Any("petGoodsMap", PetGoodsMap))
	var goodsList []models.Goods
	database.Instance().Model(&models.Goods{}).Find(&goodsList)
	GoodsMap = make(map[int64]models.Goods)
	for _, value := range goodsList {
		GoodsMap[value.Id] = value
	}
	zap_server.ZAPLOG.Info("dataInit  GoodsMap : ", zap.Any("GoodsMap", GoodsMap))
}

func DataCacheJob() {
	c, err := cron_server.CronInstance().AddFunc("@every 1m", DataInit)
	if err != nil {
		zap_server.ZAPLOG.Info("DataCacheJob  err : ", zap.Any("err", err))
	}
	zap_server.ZAPLOG.Info("DataCacheJob  c : ", zap.Any("c", c), zap.Any("err", err))
	cron_server.CronInstance().Start()
}