87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package copydata
|
||
|
||
import (
|
||
consts "cid/consts/public"
|
||
dto "cid/model/dto/copydata"
|
||
entity "cid/model/entity/copydata"
|
||
"context"
|
||
"errors"
|
||
|
||
"gitea.com/red-future/common/db/gfdb"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
)
|
||
|
||
var CidAccountReportSum = new(CidAccountReportSumDao)
|
||
|
||
type CidAccountReportSumDao struct{}
|
||
|
||
// Insert 插入广告数据报表详情
|
||
func (d *CidAccountReportSumDao) Insert(ctx context.Context, req *dto.CidAccountReportSumItem) (id int64, err error) {
|
||
var entityData *entity.CidAccountReportSum
|
||
if err = gconv.Struct(req, &entityData); err != nil {
|
||
return
|
||
}
|
||
|
||
r, err := gfdb.DB(ctx).Model(ctx, consts.CidAccountReportSumTable).Data(&entityData).Insert()
|
||
if err != nil {
|
||
return
|
||
}
|
||
return r.LastInsertId()
|
||
}
|
||
|
||
// BatchInsert 批量插入广告数据报表详情
|
||
func (d *CidAccountReportSumDao) BatchInsert(ctx context.Context, reqs []*dto.CidAccountReportSumItem) (successCount int64, failCount int64, failedIndexes []int64, err error) {
|
||
if len(reqs) == 0 {
|
||
return 0, 0, nil, errors.New("批量插入数据不能为空")
|
||
}
|
||
|
||
// 分批处理，每批100条
|
||
batchSize := 100
|
||
successCount = 0
|
||
failCount = 0
|
||
failedIndexes = make([]int64, 0)
|
||
|
||
for i := 0; i < len(reqs); i += batchSize {
|
||
end := i + batchSize
|
||
if end > len(reqs) {
|
||
end = len(reqs)
|
||
}
|
||
|
||
batch := reqs[i:end]
|
||
entityList := make([]*entity.CidAccountReportSum, len(batch))
|
||
|
||
for j, req := range batch {
|
||
var entityData entity.CidAccountReportSum
|
||
if err = gconv.Struct(req, &entityData); err != nil {
|
||
failCount++
|
||
failedIndexes = append(failedIndexes, int64(i+j))
|
||
continue
|
||
}
|
||
entityList[j] = &entityData
|
||
}
|
||
|
||
if len(entityList) == 0 {
|
||
continue
|
||
}
|
||
|
||
// 执行批量插入
|
||
_, err = gfdb.DB(ctx).Model(ctx, consts.CidAccountReportSumTable).Data(entityList).Insert()
|
||
if err != nil {
|
||
// 批量插入失败，尝试逐条插入
|
||
for k := range batch {
|
||
_, singleErr := d.Insert(ctx, batch[k])
|
||
if singleErr != nil {
|
||
failCount++
|
||
failedIndexes = append(failedIndexes, int64(i+k))
|
||
} else {
|
||
successCount++
|
||
}
|
||
}
|
||
} else {
|
||
successCount += int64(len(entityList))
|
||
}
|
||
}
|
||
|
||
return successCount, failCount, failedIndexes, nil
|
||
}
|