go-gin-api/internal/api/service/authorized_service/service_detailbykey.go

package authorized_service

import (
	"encoding/json"
	"time"

	"github.com/xinliangnote/go-gin-api/configs"
	"github.com/xinliangnote/go-gin-api/internal/api/repository/db_repo"
	"github.com/xinliangnote/go-gin-api/internal/api/repository/db_repo/authorized_api_repo"
	"github.com/xinliangnote/go-gin-api/internal/api/repository/db_repo/authorized_repo"
	"github.com/xinliangnote/go-gin-api/internal/pkg/cache"
	"github.com/xinliangnote/go-gin-api/internal/pkg/core"
)

// 定义缓存结构
type CacheAuthorizedData struct {
	Key    string         `json:"key"`     // 调用方 key
	Secret string         `json:"secret"`  // 调用方 secret
	IsUsed int32          `json:"is_used"` // 调用方启用状态 1=启用 -1=禁用
	Apis   []cacheApiData `json:"apis"`    // 调用方授权的 Apis
}

type cacheApiData struct {
	Method string `json:"method"` // 请求方式
	Api    string `json:"api"`    // 请求地址
}

func (s *service) DetailByKey(ctx core.Context, key string) (cacheData *CacheAuthorizedData, err error) {
	// 查询缓存
	cacheKey := configs.RedisKeyPrefixSignature + key
	value, err := s.cache.Get(cacheKey, cache.WithTrace(ctx.RequestContext().Trace))

	cacheData = new(CacheAuthorizedData)
	if err == nil && json.Unmarshal([]byte(value), cacheData) == nil {
		return
	}

	// 查询调用方信息
	authorizedInfo, err := authorized_repo.NewQueryBuilder().
		WhereIsDeleted(db_repo.EqualPredicate, -1).
		WhereBusinessKey(db_repo.EqualPredicate, key).
		First(s.db.GetDbR().WithContext(ctx.RequestContext()))

	if err != nil {
		return nil, err
	}

	// 查询调用方授权 API 信息
	authorizedApiInfo, err := authorized_api_repo.NewQueryBuilder().
		WhereIsDeleted(db_repo.EqualPredicate, -1).
		WhereBusinessKey(db_repo.EqualPredicate, key).
		OrderById(false).
		QueryAll(s.db.GetDbR().WithContext(ctx.RequestContext()))

	if err != nil {
		return nil, err
	}

	// 设置缓存 data
	cacheData = new(CacheAuthorizedData)
	cacheData.Key = key
	cacheData.Secret = authorizedInfo.BusinessSecret
	cacheData.IsUsed = authorizedInfo.IsUsed
	cacheData.Apis = make([]cacheApiData, len(authorizedApiInfo))

	for k, v := range authorizedApiInfo {
		data := cacheApiData{
			Method: v.Method,
			Api:    v.Api,
		}
		cacheData.Apis[k] = data
	}

	cacheDataByte, _ := json.Marshal(cacheData)

	err = s.cache.Set(cacheKey, string(cacheDataByte), time.Hour*24, cache.WithTrace(ctx.Trace()))
	if err != nil {
		return nil, err
	}

	return
}