cid/dao/cid_request_dao.go

package dao

import (
	"context"

	"cid/model/entity"

	"gitea.com/red-future/common/beans"
	"gitea.com/red-future/common/db/mongo"
	"go.mongodb.org/mongo-driver/v2/bson"
)

var CIDRequest = &cidRequestDao{}

type cidRequestDao struct {
}

// Create 创建CID请求记录
func (d *cidRequestDao) Create(ctx context.Context, request *entity.CidRequest) (id string, err error) {
	ids, err := mongo.DB().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}

	// 分页查询，使用common/mongo的Find方法，自动处理分页、租户等
	pageBean := &beans.Page{PageNum: int64(page), PageSize: int64(size)}
	total, err = mongo.DB().Find(ctx, filter, &list, entity.CidRequestCollection, pageBean, nil)
	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.DB().Count(ctx, bson.M{"userId": userId}, entity.CidRequestCollection)
	if err != nil {
		return nil, err
	}
	stats["total_requests"] = totalRequests

	// 成功请求数
	successfulRequests, err := mongo.DB().Count(ctx, 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
}