116 lines
3.0 KiB
Go
116 lines
3.0 KiB
Go
package syncdata
|
||
|
||
import (
|
||
"context"
|
||
dto "dataengine/model/dto/copydata"
|
||
"dataengine/service/copydata"
|
||
"encoding/json"
|
||
"fmt"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
)
|
||
|
||
type AccountReportSync struct {
|
||
*BaseReportSync
|
||
converter *DataConverter
|
||
mockGen *MockDataGenerator
|
||
}
|
||
|
||
func NewAccountReportSync() *AccountReportSync {
|
||
return &AccountReportSync{
|
||
BaseReportSync: NewBaseReportSync(),
|
||
converter: NewDataConverter(),
|
||
mockGen: NewMockDataGenerator(),
|
||
}
|
||
}
|
||
|
||
func (c *AccountReportSync) FetchReport(ctx context.Context, params interface{}) (interface{}, error) {
|
||
req, ok := params.(*AccountReportRequest)
|
||
if !ok {
|
||
return nil, fmt.Errorf("参数类型错误，期望 AccountReportRequest 类型")
|
||
}
|
||
|
||
useMock := false
|
||
|
||
if useMock {
|
||
logrus.Info("使用 Mock 数据")
|
||
return c.mockGen.GenerateAccountReportResponse(), nil
|
||
}
|
||
|
||
respBytes, err := NewHttpClient("https://ad.e.kuaishou.com", 0).Post(ctx, "/rest/openapi/gw/esp/report/accountReport", req)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("调用 API 失败：%w", err)
|
||
}
|
||
|
||
var response AccountReportResponse
|
||
if err := json.Unmarshal(respBytes, &response); err != nil {
|
||
return nil, fmt.Errorf("解析响应失败：%w", err)
|
||
}
|
||
|
||
if response.Code != 0 {
|
||
return nil, fmt.Errorf("API 返回错误：code=%d, message=%s", response.Code, response.Message)
|
||
}
|
||
|
||
return &response, nil
|
||
}
|
||
|
||
func (c *AccountReportSync) ConvertToSum(apiData interface{}, dataType string) interface{} {
|
||
response, ok := apiData.(*AccountReportResponse)
|
||
if !ok || response.Data == nil || response.Data.Sum == nil {
|
||
return nil
|
||
}
|
||
|
||
return c.converter.ConvertToSumItem(response.Data.Sum, dataType, 0)
|
||
}
|
||
|
||
func (c *AccountReportSync) ConvertToDetails(apiData interface{}, dataType string) []interface{} {
|
||
response, ok := apiData.(*AccountReportResponse)
|
||
if !ok || response.Data == nil || len(response.Data.Detail) == 0 {
|
||
return nil
|
||
}
|
||
|
||
detailItems := c.converter.ConvertToDetailItems(response.Data.Detail, dataType, 0)
|
||
|
||
result := make([]interface{}, len(detailItems))
|
||
for i, item := range detailItems {
|
||
result[i] = item
|
||
}
|
||
return result
|
||
}
|
||
|
||
func (c *AccountReportSync) SaveSum(ctx context.Context, data interface{}) (int64, error) {
|
||
sumItem, ok := data.(*dto.CidAccountReportSumItem)
|
||
if !ok {
|
||
return 0, fmt.Errorf("数据类型错误，期望 CidAccountReportSumItem 类型")
|
||
}
|
||
|
||
res, err := copydata.CidAccountReportDetail.CreateSum(ctx, sumItem)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
return res.Id, nil
|
||
}
|
||
|
||
func (c *AccountReportSync) SaveDetails(ctx context.Context, data []interface{}) (int64, int64, error) {
|
||
detailItems := make([]*dto.CidAccountReportDetailItem, len(data))
|
||
for i, item := range data {
|
||
detailItem, ok := item.(*dto.CidAccountReportDetailItem)
|
||
if !ok {
|
||
return 0, 0, fmt.Errorf("第 %d 条数据类型错误", i)
|
||
}
|
||
detailItems[i] = detailItem
|
||
}
|
||
|
||
req := &dto.BatchCreateCidAccountReportDetailReq{
|
||
Items: detailItems,
|
||
}
|
||
|
||
res, err := copydata.CidAccountReportDetail.BatchCreate(ctx, req)
|
||
if err != nil {
|
||
return 0, 0, err
|
||
}
|
||
|
||
return res.SuccessCount, res.FailCount, nil
|
||
}
|