f

internal/infrastructure/database/repositories/api/gorm_api_call_repository.go

@@ -0,0 +1,556 @@
+package api
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+	"hyapi-server/internal/domains/api/entities"
+	"hyapi-server/internal/domains/api/repositories"
+	"hyapi-server/internal/shared/database"
+	"hyapi-server/internal/shared/interfaces"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+const (
+	ApiCallsTable   = "api_calls"
+	ApiCallCacheTTL = 10 * time.Minute
+)
+
+// ApiCallWithProduct 包含产品名称的API调用记录
+type ApiCallWithProduct struct {
+	entities.ApiCall
+	ProductName string `json:"product_name" gorm:"column:product_name"`
+}
+
+type GormApiCallRepository struct {
+	*database.CachedBaseRepositoryImpl
+}
+
+var _ repositories.ApiCallRepository = (*GormApiCallRepository)(nil)
+
+func NewGormApiCallRepository(db *gorm.DB, logger *zap.Logger) repositories.ApiCallRepository {
+	return &GormApiCallRepository{
+		CachedBaseRepositoryImpl: database.NewCachedBaseRepositoryImpl(db, logger, ApiCallsTable),
+	}
+}
+
+func (r *GormApiCallRepository) Create(ctx context.Context, call *entities.ApiCall) error {
+	return r.CreateEntity(ctx, call)
+}
+
+func (r *GormApiCallRepository) Update(ctx context.Context, call *entities.ApiCall) error {
+	return r.UpdateEntity(ctx, call)
+}
+
+func (r *GormApiCallRepository) FindById(ctx context.Context, id string) (*entities.ApiCall, error) {
+	var call entities.ApiCall
+	err := r.SmartGetByID(ctx, id, &call)
+	if err != nil {
+		return nil, err
+	}
+	return &call, nil
+}
+
+func (r *GormApiCallRepository) FindByUserId(ctx context.Context, userId string, limit, offset int) ([]*entities.ApiCall, error) {
+	var calls []*entities.ApiCall
+	options := database.CacheListOptions{
+		Where:  "user_id = ?",
+		Args:   []interface{}{userId},
+		Order:  "created_at DESC",
+		Limit:  limit,
+		Offset: offset,
+	}
+	err := r.ListWithCache(ctx, &calls, ApiCallCacheTTL, options)
+	return calls, err
+}
+
+func (r *GormApiCallRepository) ListByUserId(ctx context.Context, userId string, options interfaces.ListOptions) ([]*entities.ApiCall, int64, error) {
+	var calls []*entities.ApiCall
+	var total int64
+
+	// 构建查询条件
+	whereCondition := "user_id = ?"
+	whereArgs := []interface{}{userId}
+
+	// 获取总数
+	count, err := r.CountWhere(ctx, &entities.ApiCall{}, whereCondition, whereArgs...)
+	if err != nil {
+		return nil, 0, err
+	}
+	total = count
+
+	// 使用基础仓储的分页查询方法
+	err = r.ListWithOptions(ctx, &entities.ApiCall{}, &calls, options)
+	return calls, total, err
+}
+
+func (r *GormApiCallRepository) ListByUserIdWithFilters(ctx context.Context, userId string, filters map[string]interface{}, options interfaces.ListOptions) ([]*entities.ApiCall, int64, error) {
+	var calls []*entities.ApiCall
+	var total int64
+
+	// 构建基础查询条件
+	whereCondition := "user_id = ?"
+	whereArgs := []interface{}{userId}
+
+	// 应用筛选条件
+	if filters != nil {
+		// 时间范围筛选
+		if startTime, ok := filters["start_time"].(time.Time); ok {
+			whereCondition += " AND created_at >= ?"
+			whereArgs = append(whereArgs, startTime)
+		}
+		if endTime, ok := filters["end_time"].(time.Time); ok {
+			whereCondition += " AND created_at <= ?"
+			whereArgs = append(whereArgs, endTime)
+		}
+
+		// TransactionID筛选
+		if transactionId, ok := filters["transaction_id"].(string); ok && transactionId != "" {
+			whereCondition += " AND transaction_id LIKE ?"
+			whereArgs = append(whereArgs, "%"+transactionId+"%")
+		}
+
+		// 产品ID筛选
+		if productId, ok := filters["product_id"].(string); ok && productId != "" {
+			whereCondition += " AND product_id = ?"
+			whereArgs = append(whereArgs, productId)
+		}
+
+		// 状态筛选
+		if status, ok := filters["status"].(string); ok && status != "" {
+			whereCondition += " AND status = ?"
+			whereArgs = append(whereArgs, status)
+		}
+	}
+
+	// 获取总数
+	count, err := r.CountWhere(ctx, &entities.ApiCall{}, whereCondition, whereArgs...)
+	if err != nil {
+		return nil, 0, err
+	}
+	total = count
+
+	// 使用基础仓储的分页查询方法
+	err = r.ListWithOptions(ctx, &entities.ApiCall{}, &calls, options)
+	return calls, total, err
+}
+
+// ListByUserIdWithFiltersAndProductName 根据用户ID和筛选条件获取API调用记录（包含产品名称）
+func (r *GormApiCallRepository) ListByUserIdWithFiltersAndProductName(ctx context.Context, userId string, filters map[string]interface{}, options interfaces.ListOptions) (map[string]string, []*entities.ApiCall, int64, error) {
+	var callsWithProduct []*ApiCallWithProduct
+	var total int64
+
+	// 构建基础查询条件
+	whereCondition := "ac.user_id = ?"
+	whereArgs := []interface{}{userId}
+
+	// 应用筛选条件
+	if filters != nil {
+		// 时间范围筛选
+		if startTime, ok := filters["start_time"].(time.Time); ok {
+			whereCondition += " AND ac.created_at >= ?"
+			whereArgs = append(whereArgs, startTime)
+		}
+		if endTime, ok := filters["end_time"].(time.Time); ok {
+			whereCondition += " AND ac.created_at <= ?"
+			whereArgs = append(whereArgs, endTime)
+		}
+
+		// TransactionID筛选
+		if transactionId, ok := filters["transaction_id"].(string); ok && transactionId != "" {
+			whereCondition += " AND ac.transaction_id LIKE ?"
+			whereArgs = append(whereArgs, "%"+transactionId+"%")
+		}
+
+		// 产品名称筛选
+		if productName, ok := filters["product_name"].(string); ok && productName != "" {
+			whereCondition += " AND p.name LIKE ?"
+			whereArgs = append(whereArgs, "%"+productName+"%")
+		}
+
+		// 状态筛选
+		if status, ok := filters["status"].(string); ok && status != "" {
+			whereCondition += " AND ac.status = ?"
+			whereArgs = append(whereArgs, status)
+		}
+	}
+
+	// 构建JOIN查询
+	query := r.GetDB(ctx).Table("api_calls ac").
+		Select("ac.*, p.name as product_name").
+		Joins("LEFT JOIN product p ON ac.product_id = p.id").
+		Where(whereCondition, whereArgs...)
+
+	// 获取总数
+	var count int64
+	err := query.Count(&count).Error
+	if err != nil {
+		return nil, nil, 0, err
+	}
+	total = count
+
+	// 应用排序和分页
+	if options.Sort != "" {
+		query = query.Order("ac." + options.Sort + " " + options.Order)
+	} else {
+		query = query.Order("ac.created_at DESC")
+	}
+
+	if options.Page > 0 && options.PageSize > 0 {
+		offset := (options.Page - 1) * options.PageSize
+		query = query.Offset(offset).Limit(options.PageSize)
+	}
+
+	// 执行查询
+	err = query.Find(&callsWithProduct).Error
+	if err != nil {
+		return nil, nil, 0, err
+	}
+
+	// 转换为entities.ApiCall并构建产品名称映射
+	var calls []*entities.ApiCall
+	productNameMap := make(map[string]string)
+
+	for _, c := range callsWithProduct {
+		call := c.ApiCall
+		calls = append(calls, &call)
+		// 构建产品ID到产品名称的映射
+		if c.ProductName != "" {
+			productNameMap[call.ID] = c.ProductName
+		}
+	}
+
+	return productNameMap, calls, total, nil
+}
+
+func (r *GormApiCallRepository) CountByUserId(ctx context.Context, userId string) (int64, error) {
+	return r.CountWhere(ctx, &entities.ApiCall{}, "user_id = ?", userId)
+}
+
+// CountByUserIdAndProductId 按用户ID和产品ID统计API调用次数
+func (r *GormApiCallRepository) CountByUserIdAndProductId(ctx context.Context, userId string, productId string) (int64, error) {
+	return r.CountWhere(ctx, &entities.ApiCall{}, "user_id = ? AND product_id = ?", userId, productId)
+}
+
+// CountByUserIdAndDateRange 按用户ID和日期范围统计API调用次数
+func (r *GormApiCallRepository) CountByUserIdAndDateRange(ctx context.Context, userId string, startDate, endDate time.Time) (int64, error) {
+	return r.CountWhere(ctx, &entities.ApiCall{}, "user_id = ? AND created_at >= ? AND created_at < ?", userId, startDate, endDate)
+}
+
+// GetDailyStatsByUserId 获取用户每日API调用统计
+func (r *GormApiCallRepository) GetDailyStatsByUserId(ctx context.Context, userId string, startDate, endDate time.Time) ([]map[string]interface{}, error) {
+	var results []map[string]interface{}
+
+	// 构建SQL查询 - 使用PostgreSQL语法，使用具体的日期范围
+	sql := `
+		SELECT
+			DATE(created_at) as date,
+			COUNT(*) as calls
+		FROM api_calls
+		WHERE user_id = $1
+			AND DATE(created_at) >= $2
+			AND DATE(created_at) <= $3
+		GROUP BY DATE(created_at)
+		ORDER BY date ASC
+	`
+
+	err := r.GetDB(ctx).Raw(sql, userId, startDate.Format("2006-01-02"), endDate.Format("2006-01-02")).Scan(&results).Error
+	if err != nil {
+		return nil, err
+	}
+
+	return results, nil
+}
+
+// GetMonthlyStatsByUserId 获取用户每月API调用统计
+func (r *GormApiCallRepository) GetMonthlyStatsByUserId(ctx context.Context, userId string, startDate, endDate time.Time) ([]map[string]interface{}, error) {
+	var results []map[string]interface{}
+
+	// 构建SQL查询 - 使用PostgreSQL语法，使用具体的日期范围
+	sql := `
+		SELECT
+			TO_CHAR(created_at, 'YYYY-MM') as month,
+			COUNT(*) as calls
+		FROM api_calls
+		WHERE user_id = $1
+			AND created_at >= $2
+			AND created_at <= $3
+		GROUP BY TO_CHAR(created_at, 'YYYY-MM')
+		ORDER BY month ASC
+	`
+
+	err := r.GetDB(ctx).Raw(sql, userId, startDate, endDate).Scan(&results).Error
+	if err != nil {
+		return nil, err
+	}
+
+	return results, nil
+}
+
+func (r *GormApiCallRepository) FindByTransactionId(ctx context.Context, transactionId string) (*entities.ApiCall, error) {
+	var call entities.ApiCall
+	err := r.FindOne(ctx, &call, "transaction_id = ?", transactionId)
+	if err != nil {
+		return nil, err
+	}
+	return &call, nil
+}
+
+// ListWithFiltersAndProductName 管理端：根据条件筛选所有API调用记录（包含产品名称）
+func (r *GormApiCallRepository) ListWithFiltersAndProductName(ctx context.Context, filters map[string]interface{}, options interfaces.ListOptions) (map[string]string, []*entities.ApiCall, int64, error) {
+	var callsWithProduct []*ApiCallWithProduct
+	var total int64
+
+	// 构建基础查询条件
+	whereCondition := "1=1"
+	whereArgs := []interface{}{}
+
+	// 应用筛选条件
+	if filters != nil {
+		// 用户ID筛选（支持单个user_id和多个user_ids）
+		// 如果同时存在，优先使用user_ids（批量查询）
+		if userIds, ok := filters["user_ids"].(string); ok && userIds != "" {
+			// 解析逗号分隔的用户ID列表
+			userIdsList := strings.Split(userIds, ",")
+			// 去除空白字符
+			var cleanUserIds []string
+			for _, id := range userIdsList {
+				id = strings.TrimSpace(id)
+				if id != "" {
+					cleanUserIds = append(cleanUserIds, id)
+				}
+			}
+			if len(cleanUserIds) > 0 {
+				placeholders := strings.Repeat("?,", len(cleanUserIds))
+				placeholders = placeholders[:len(placeholders)-1] // 移除最后一个逗号
+				whereCondition += " AND ac.user_id IN (" + placeholders + ")"
+				for _, id := range cleanUserIds {
+					whereArgs = append(whereArgs, id)
+				}
+			}
+		} else if userId, ok := filters["user_id"].(string); ok && userId != "" {
+			// 单个用户ID筛选
+			whereCondition += " AND ac.user_id = ?"
+			whereArgs = append(whereArgs, userId)
+		}
+
+		// 时间范围筛选
+		if startTime, ok := filters["start_time"].(time.Time); ok {
+			whereCondition += " AND ac.created_at >= ?"
+			whereArgs = append(whereArgs, startTime)
+		}
+		if endTime, ok := filters["end_time"].(time.Time); ok {
+			whereCondition += " AND ac.created_at <= ?"
+			whereArgs = append(whereArgs, endTime)
+		}
+
+		// TransactionID筛选
+		if transactionId, ok := filters["transaction_id"].(string); ok && transactionId != "" {
+			whereCondition += " AND ac.transaction_id LIKE ?"
+			whereArgs = append(whereArgs, "%"+transactionId+"%")
+		}
+
+		// 产品名称筛选
+		if productName, ok := filters["product_name"].(string); ok && productName != "" {
+			whereCondition += " AND p.name LIKE ?"
+			whereArgs = append(whereArgs, "%"+productName+"%")
+		}
+
+		// 企业名称筛选
+		if companyName, ok := filters["company_name"].(string); ok && companyName != "" {
+			whereCondition += " AND ei.company_name LIKE ?"
+			whereArgs = append(whereArgs, "%"+companyName+"%")
+		}
+
+		// 状态筛选
+		if status, ok := filters["status"].(string); ok && status != "" {
+			whereCondition += " AND ac.status = ?"
+			whereArgs = append(whereArgs, status)
+		}
+	}
+
+	// 构建JOIN查询
+	// 需要JOIN product表获取产品名称，JOIN users和enterprise_infos表获取企业名称
+	query := r.GetDB(ctx).Table("api_calls ac").
+		Select("ac.*, p.name as product_name").
+		Joins("LEFT JOIN product p ON ac.product_id = p.id").
+		Joins("LEFT JOIN users u ON ac.user_id = u.id").
+		Joins("LEFT JOIN enterprise_infos ei ON u.id = ei.user_id").
+		Where(whereCondition, whereArgs...)
+
+	// 获取总数
+	var count int64
+	err := query.Count(&count).Error
+	if err != nil {
+		return nil, nil, 0, err
+	}
+	total = count
+
+	// 应用排序和分页
+	if options.Sort != "" {
+		query = query.Order("ac." + options.Sort + " " + options.Order)
+	} else {
+		query = query.Order("ac.created_at DESC")
+	}
+
+	if options.Page > 0 && options.PageSize > 0 {
+		offset := (options.Page - 1) * options.PageSize
+		query = query.Offset(offset).Limit(options.PageSize)
+	}
+
+	// 执行查询
+	err = query.Find(&callsWithProduct).Error
+	if err != nil {
+		return nil, nil, 0, err
+	}
+
+	// 转换为entities.ApiCall并构建产品名称映射
+	var calls []*entities.ApiCall
+	productNameMap := make(map[string]string)
+
+	for _, c := range callsWithProduct {
+		call := c.ApiCall
+		calls = append(calls, &call)
+		// 构建产品ID到产品名称的映射
+		if c.ProductName != "" {
+			productNameMap[call.ID] = c.ProductName
+		}
+	}
+
+	return productNameMap, calls, total, nil
+}
+
+// GetSystemTotalCalls 获取系统总API调用次数
+func (r *GormApiCallRepository) GetSystemTotalCalls(ctx context.Context) (int64, error) {
+	var count int64
+	err := r.GetDB(ctx).Model(&entities.ApiCall{}).Count(&count).Error
+	return count, err
+}
+
+// GetSystemCallsByDateRange 获取系统指定时间范围内的API调用次数
+// endDate 应该是结束日期当天的次日00:00:00（日统计）或下个月1号00:00:00（月统计），使用 < 而不是 <=
+func (r *GormApiCallRepository) GetSystemCallsByDateRange(ctx context.Context, startDate, endDate time.Time) (int64, error) {
+	var count int64
+	err := r.GetDB(ctx).Model(&entities.ApiCall{}).
+		Where("created_at >= ? AND created_at < ?", startDate, endDate).
+		Count(&count).Error
+	return count, err
+}
+
+// GetSystemDailyStats 获取系统每日API调用统计
+func (r *GormApiCallRepository) GetSystemDailyStats(ctx context.Context, startDate, endDate time.Time) ([]map[string]interface{}, error) {
+	var results []map[string]interface{}
+
+	sql := `
+		SELECT
+			DATE(created_at) as date,
+			COUNT(*) as calls
+		FROM api_calls
+		WHERE DATE(created_at) >= $1
+			AND DATE(created_at) <= $2
+		GROUP BY DATE(created_at)
+		ORDER BY date ASC
+	`
+
+	err := r.GetDB(ctx).Raw(sql, startDate.Format("2006-01-02"), endDate.Format("2006-01-02")).Scan(&results).Error
+	if err != nil {
+		return nil, err
+	}
+
+	return results, nil
+}
+
+// GetSystemMonthlyStats 获取系统每月API调用统计
+func (r *GormApiCallRepository) GetSystemMonthlyStats(ctx context.Context, startDate, endDate time.Time) ([]map[string]interface{}, error) {
+	var results []map[string]interface{}
+
+	sql := `
+		SELECT
+			TO_CHAR(created_at, 'YYYY-MM') as month,
+			COUNT(*) as calls
+		FROM api_calls
+		WHERE created_at >= $1
+			AND created_at < $2
+		GROUP BY TO_CHAR(created_at, 'YYYY-MM')
+		ORDER BY month ASC
+	`
+
+	err := r.GetDB(ctx).Raw(sql, startDate, endDate).Scan(&results).Error
+	if err != nil {
+		return nil, err
+	}
+
+	return results, nil
+}
+
+// GetApiPopularityRanking 获取API受欢迎程度排行榜
+func (r *GormApiCallRepository) GetApiPopularityRanking(ctx context.Context, period string, limit int) ([]map[string]interface{}, error) {
+	var sql string
+	var args []interface{}
+
+	switch period {
+	case "today":
+		sql = `
+			SELECT
+				p.id as product_id,
+				p.name as api_name,
+				p.description as api_description,
+				COUNT(ac.id) as call_count
+			FROM product p
+			LEFT JOIN api_calls ac ON p.id = ac.product_id
+				AND DATE(ac.created_at) = CURRENT_DATE
+			WHERE p.deleted_at IS NULL
+			GROUP BY p.id, p.name, p.description
+			HAVING COUNT(ac.id) > 0
+			ORDER BY call_count DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "month":
+		sql = `
+			SELECT
+				p.id as product_id,
+				p.name as api_name,
+				p.description as api_description,
+				COUNT(ac.id) as call_count
+			FROM product p
+			LEFT JOIN api_calls ac ON p.id = ac.product_id
+				AND DATE_TRUNC('month', ac.created_at) = DATE_TRUNC('month', CURRENT_DATE)
+			WHERE p.deleted_at IS NULL
+			GROUP BY p.id, p.name, p.description
+			HAVING COUNT(ac.id) > 0
+			ORDER BY call_count DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "total":
+		sql = `
+			SELECT
+				p.id as product_id,
+				p.name as api_name,
+				p.description as api_description,
+				COUNT(ac.id) as call_count
+			FROM product p
+			LEFT JOIN api_calls ac ON p.id = ac.product_id
+			WHERE p.deleted_at IS NULL
+			GROUP BY p.id, p.name, p.description
+			HAVING COUNT(ac.id) > 0
+			ORDER BY call_count DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	default:
+		return nil, fmt.Errorf("不支持的时间周期: %s", period)
+	}
+
+	var results []map[string]interface{}
+	err := r.GetDB(ctx).Raw(sql, args...).Scan(&results).Error
+	if err != nil {
+		return nil, err
+	}
+
+	return results, nil
+}

internal/infrastructure/database/repositories/api/gorm_api_user_repository.go

@@ -0,0 +1,56 @@
+package api
+
+import (
+	"context"
+	"hyapi-server/internal/domains/api/entities"
+	"hyapi-server/internal/domains/api/repositories"
+	"hyapi-server/internal/shared/database"
+
+	"time"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+const (
+	ApiUsersTable = "api_users"
+	ApiUserCacheTTL = 30 * time.Minute
+)
+
+type GormApiUserRepository struct {
+	*database.CachedBaseRepositoryImpl
+}
+
+var _ repositories.ApiUserRepository = (*GormApiUserRepository)(nil)
+
+func NewGormApiUserRepository(db *gorm.DB, logger *zap.Logger) repositories.ApiUserRepository {
+	return &GormApiUserRepository{
+		CachedBaseRepositoryImpl: database.NewCachedBaseRepositoryImpl(db, logger, ApiUsersTable),
+	}
+}
+
+func (r *GormApiUserRepository) Create(ctx context.Context, user *entities.ApiUser) error {
+	return r.CreateEntity(ctx, user)
+}
+
+func (r *GormApiUserRepository) Update(ctx context.Context, user *entities.ApiUser) error {
+	return r.UpdateEntity(ctx, user)
+}
+
+func (r *GormApiUserRepository) FindByAccessId(ctx context.Context, accessId string) (*entities.ApiUser, error) {
+	var user entities.ApiUser
+	err := r.SmartGetByField(ctx, &user, "access_id", accessId, ApiUserCacheTTL)
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+func (r *GormApiUserRepository) FindByUserId(ctx context.Context, userId string) (*entities.ApiUser, error) {
+	var user entities.ApiUser
+	err := r.SmartGetByField(ctx, &user, "user_id", userId, ApiUserCacheTTL)
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}

internal/infrastructure/database/repositories/api/gorm_enterprise_report_repository.go

@@ -0,0 +1,44 @@
+package api
+
+import (
+	"context"
+
+	"hyapi-server/internal/domains/api/entities"
+	"hyapi-server/internal/domains/api/repositories"
+	"hyapi-server/internal/shared/database"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+const (
+	ReportsTable = "reports"
+)
+
+// GormReportRepository 报告记录 GORM 仓储实现
+type GormReportRepository struct {
+	*database.BaseRepositoryImpl
+}
+
+var _ repositories.ReportRepository = (*GormReportRepository)(nil)
+
+// NewGormReportRepository 创建报告记录仓储实现
+func NewGormReportRepository(db *gorm.DB, logger *zap.Logger) repositories.ReportRepository {
+	return &GormReportRepository{
+		BaseRepositoryImpl: database.NewBaseRepositoryImpl(db, logger),
+	}
+}
+
+// Create 创建报告记录
+func (r *GormReportRepository) Create(ctx context.Context, report *entities.Report) error {
+	return r.CreateEntity(ctx, report)
+}
+
+// FindByReportID 根据报告编号查询记录
+func (r *GormReportRepository) FindByReportID(ctx context.Context, reportID string) (*entities.Report, error) {
+	var report entities.Report
+	if err := r.FindOneByField(ctx, &report, "report_id", reportID); err != nil {
+		return nil, err
+	}
+	return &report, nil
+}