cid/dao/cid_request_dao.go

package dao

import (
	"context"

	"cid/model/entity"
	"gitee.com/red-future---jilin-g/common/mongo"
	"go.mongodb.org/mongo-driver/v2/bson"
	"go.mongodb.org/mongo-driver/v2/mongo/options"
)

var CIDRequest = &cidRequestDao{
	NoCache: true,
}

type cidRequestDao struct {
	NoCache bool
}

func (d *cidRequestDao) SetNoCache() {
	CIDRequest.NoCache = true
}

// Create 创建CID请求记录
func (d *cidRequestDao) Create(ctx context.Context, request *entity.CidRequest) (id string, err error) {
	ids, err := mongo.Insert(ctx, []interface{}{request}, entity.CidRequestCollection)
	if err != nil {
		return "", err
	}
	if len(ids) > 0 {
		id = ids[0].(string)
	}
	return
}

// GetHistory 获取CID请求历史
func (d *cidRequestDao) GetHistory(ctx context.Context, userId string, page, size int) (list []*entity.CidRequest, total int64, err error) {
	filter := bson.M{"userId": userId}

	// 获取总数
	total, err = mongo.Count(ctx, d.NoCache, filter, entity.CidRequestCollection)
	if err != nil {
		return
	}

	// 分页查询
	offset := (page - 1) * size
	err = mongo.Find(ctx, d.NoCache, filter, &list, entity.CidRequestCollection,
		options.Find().SetSort(bson.M{"createdAt": -1}).
			SetSkip(int64(offset)).
			SetLimit(int64(size)))

	return
}

// GetStatistics 获取统计信息
func (d *cidRequestDao) GetStatistics(ctx context.Context, userId string) (stats map[string]interface{}, err error) {
	stats = make(map[string]interface{})

	// 总请求数
	totalRequests, err := mongo.Count(ctx, d.NoCache, bson.M{"userId": userId}, entity.CidRequestCollection)
	if err != nil {
		return nil, err
	}
	stats["total_requests"] = totalRequests

	// 成功请求数
	successfulRequests, err := mongo.Count(ctx, d.NoCache, bson.M{"userId": userId, "status": "completed"}, entity.CidRequestCollection)
	if err != nil {
		return nil, err
	}
	stats["successful_requests"] = successfulRequests

	// 平均处理时间需要单独计算，MongoDB聚合查询
	// 这里简化处理，返回0
	stats["average_process_time"] = 0

	return
}