epic-go/internal/logic/cron/third_party_sync.go

package cron

import (
	"context"
	"epic/internal/consts"
	"epic/internal/model/dto"
	"epic/utility"
	"fmt"
	"github.com/gogf/gf/v2/encoding/gjson"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/gclient"
	"github.com/gogf/gf/v2/os/gtime"
	"time"
)

// ThirdPartyDataSync 第三方数据同步器
type ThirdPartyDataSync struct {
	client *gclient.Client
}

// NewThirdPartyDataSync 创建第三方数据同步器
func NewThirdPartyDataSync() *ThirdPartyDataSync {
	return &ThirdPartyDataSync{
		client: gclient.New().Timeout(30 * time.Second),
	}
}

// SyncHeroData 同步英雄数据
func (t *ThirdPartyDataSync) SyncHeroData(ctx context.Context) error {
	g.Log().Info(ctx, "开始同步英雄数据...")

	// 示例：从第三方API获取英雄数据
	heroData, err := t.fetchHeroDataFromAPI(ctx)
	if err != nil || heroData == nil {
		g.Log().Error(ctx, "获取英雄数据失败:", err)
		return err
	}

	// 处理并保存数据
	if err := t.processAndSaveHeroData(ctx, heroData); err != nil {
		g.Log().Error(ctx, "处理英雄数据失败:", err)
		return err
	}

	g.Log().Info(ctx, "英雄数据同步完成")
	return nil
}

// SyncArtifactData 同步神器数据
func (t *ThirdPartyDataSync) SyncArtifactData(ctx context.Context) error {
	g.Log().Info(ctx, "开始同步神器数据...")

	utility.RedisCache.Set(ctx, "artifacts_all11", "asd", 0)

	// 示例：从第三方API获取神器数据
	//artifactData, err := t.fetchArtifactDataFromAPI(ctx)
	//if err != nil {
	//	g.Log().Error(ctx, "获取神器数据失败:", err)
	//	return err
	//}
	//
	//// 处理并保存数据
	//if err := t.processAndSaveArtifactData(ctx, artifactData); err != nil {
	//	g.Log().Error(ctx, "处理神器数据失败:", err)
	//	return err
	//}

	g.Log().Info(ctx, "神器数据同步完成")
	return nil
}

// fetchHeroDataFromAPI 从API获取英雄数据
func (t *ThirdPartyDataSync) fetchHeroDataFromAPI(ctx context.Context) ([]byte, error) {
	// 示例API地址，实际使用时需要替换为真实的API
	apiURL := consts.HeroListURL

	// 添加请求头
	headers := map[string]string{
		"User-Agent": "EpicGameBot/1.0",
		"Accept":     "application/json",
	}

	// 发送GET请求
	resp, err := t.client.Header(headers).Get(ctx, apiURL)
	if err != nil {
		return nil, fmt.Errorf("API请求失败: %v", err)
	}
	defer resp.Close()

	// 检查响应状态
	if resp.StatusCode != 200 {
		return nil, fmt.Errorf("API响应错误，状态码: %d", resp.StatusCode)
	}

	// 读取响应内容
	content := resp.ReadAll()
	g.Log().Debug(ctx, "API响应内容长度:", len(content))

	return content, nil
}

// 从API获取神器数据
func (t *ThirdPartyDataSync) fetchArtifactDataFromAPI(ctx context.Context) (string, error) {
	// 示例API地址
	apiURL := "https://static.smilegatemegaport.com/gameRecord/epic7/epic7_artifact.json?_=1729322698936"

	headers := map[string]string{
		//"User-Agent": "EpicGameBot/1.0",
		"Accept": "application/json",
	}

	resp, err := t.client.Header(headers).Get(ctx, apiURL)
	if err != nil {
		return "", fmt.Errorf("API请求失败: %v", err)
	}
	defer resp.Close()

	if resp.StatusCode != 200 {
		return "", fmt.Errorf("API响应错误，状态码: %d", resp.StatusCode)
	}

	content := resp.ReadAll()
	g.Log().Debug(ctx, "神器API响应内容长度:", len(content))

	return string(content), nil
}

// processAndSaveHeroData 处理并保存英雄数据
func (t *ThirdPartyDataSync) processAndSaveHeroData(ctx context.Context, data []byte) error {
	// 使用 gjson 解析
	j := gjson.New(data)

	if j == nil || j.IsNil() {
		return fmt.Errorf("英雄数据格式错误，期望是一个JSON对象")
	}

	// 先解析为 map[string]*ThirdPartyHeroDTO
	var heroes map[string]*dto.ThirdPartyHeroDTO
	if err := j.Scan(&heroes); err != nil {
		return fmt.Errorf("解析英雄数据到DTO失败: %v", err)
	}

	g.Log().Info(ctx, "解析到", len(heroes), "个英雄数据")

	// 遍历 map，设置 Name 字段，并保存
	for name, hero := range heroes {
		hero.Name = name // 将 map 的 key 作为 Name 字段
		if err := t.saveHeroData(ctx, hero); err != nil {
			g.Log().Error(ctx, "保存英雄数据失败:", err)
			continue
		}
	}

	return nil
}

// processAndSaveArtifactData 处理并保存神器数据
func (t *ThirdPartyDataSync) processAndSaveArtifactData(ctx context.Context, data string) error {
	// 使用 gjson 解析
	j := gjson.New(data)
	zhcn := j.Get("zh-CN")
	// 检查json对象本身和其内部值，并使用 .Var().IsSlice() 这种更可靠的方式判断是否为数组
	if !zhcn.IsSlice() {
		return fmt.Errorf("神器数据格式错误，期望是一个JSON数组")
	}

	var artifacts []*dto.ThirdPartyArtifactDTO
	if err := zhcn.Scan(&artifacts); err != nil {
		return fmt.Errorf("解析神器数据到DTO失败: %v", err)
	}

	g.Log().Info(ctx, "解析到", len(artifacts), "个神器数据")

	// 批量处理数据
	for _, artifact := range artifacts {
		if err := t.saveArtifactData(ctx, artifact); err != nil {
			g.Log().Error(ctx, "保存神器数据失败:", err)
			continue
		}
	}

	return nil
}

// saveHeroData 保存单个英雄数据
func (t *ThirdPartyDataSync) saveHeroData(ctx context.Context, hero *dto.ThirdPartyHeroDTO) error {
	// TODO: 实现具体的数据库保存逻辑
	// 现在 hero 是一个强类型对象，可以直接使用 hero.Code, hero.Name 等

	// 示例：记录同步日志
	syncLog := map[string]interface{}{
		"type":      "hero_sync",
		"hero_code": hero.Code,
		"sync_time": gtime.Now(),
		"status":    "success",
	}

	g.Log().Debug(ctx, "保存英雄数据:", syncLog)

	return nil
}

// 保存单个神器数据
func (t *ThirdPartyDataSync) saveArtifactData(ctx context.Context, artifact *dto.ThirdPartyArtifactDTO) error {
	// TODO: 实现具体的数据库保存逻辑
	// 现在 artifact 是一个强类型对象, 可以直接使用 artifact.Code, artifact.Name 等
	// 示例：记录同步日志
	syncLog := map[string]interface{}{
		"type":          "artifact_sync",
		"artifact_code": artifact.Code,
		"sync_time":     gtime.Now(),
		"status":        "success",
	}

	g.Log().Debug(ctx, "保存神器数据:", syncLog)

	return nil
}

// SyncAllData 同步所有数据
func (t *ThirdPartyDataSync) SyncAllData(ctx context.Context) error {
	g.Log().Info(ctx, "开始同步所有第三方数据...")

	// 同步英雄数据
	if err := t.SyncHeroData(ctx); err != nil {
		g.Log().Error(ctx, "英雄数据同步失败:", err)
		// 继续同步其他数据
	}

	// 同步神器数据
	if err := t.SyncArtifactData(ctx); err != nil {
		g.Log().Error(ctx, "神器数据同步失败:", err)
		// 继续同步其他数据
	}

	// 可以继续添加其他数据类型的同步

	g.Log().Info(ctx, "所有第三方数据同步完成")
	return nil
}