64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
|
||
"cid/model/entity"
|
||
|
||
"gitee.com/red-future---jilin-g/common/beans"
|
||
"gitee.com/red-future---jilin-g/common/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
|
||
}
|