new

2025-09-12 01:15:09 +08:00
parent c563b2266b
commit e05ad9e223
103 changed files with 20034 additions and 1041 deletions
--- a/internal/infrastructure/database/repositories/user/gorm_user_repository.go
+++ b/internal/infrastructure/database/repositories/user/gorm_user_repository.go
@@ -6,6 +6,7 @@ package repositories
 import (
 	"context"
 	"errors"
+	"fmt"
 	"time"

 	"go.uber.org/zap"
@@ -71,6 +72,20 @@ func (r *GormUserRepository) GetByIDWithEnterpriseInfo(ctx context.Context, id s
 	return user, nil
 }

+func (r *GormUserRepository) BatchGetByIDsWithEnterpriseInfo(ctx context.Context, ids []string) ([]*entities.User, error) {
+	if len(ids) == 0 {
+		return []*entities.User{}, nil
+	}
+
+	var users []*entities.User
+	if err := r.GetDB(ctx).Preload("EnterpriseInfo").Where("id IN ?", ids).Find(&users).Error; err != nil {
+		r.GetLogger().Error("批量查询用户失败", zap.Error(err), zap.Strings("ids", ids))
+		return nil, err
+	}
+
+	return users, nil
+}
+
 func (r *GormUserRepository) ExistsByUnifiedSocialCode(ctx context.Context, unifiedSocialCode string, excludeUserID string) (bool, error) {
 	var count int64
 	query := r.GetDB(ctx).Model(&entities.User{}).
@@ -337,3 +352,315 @@ func (r *GormUserRepository) GetStatsByDateRange(ctx context.Context, startDate,

 	return &stats, nil
 }
+
+// GetSystemUserStats 获取系统用户统计信息
+func (r *GormUserRepository) GetSystemUserStats(ctx context.Context) (*repositories.UserStats, error) {
+	var stats repositories.UserStats
+
+	db := r.GetDB(ctx)
+
+	// 总用户数
+	if err := db.Model(&entities.User{}).Count(&stats.TotalUsers).Error; err != nil {
+		return nil, err
+	}
+
+	// 活跃用户数（最近30天有登录）
+	thirtyDaysAgo := time.Now().AddDate(0, 0, -30)
+	if err := db.Model(&entities.User{}).Where("last_login_at >= ?", thirtyDaysAgo).Count(&stats.ActiveUsers).Error; err != nil {
+		return nil, err
+	}
+
+	// 已认证用户数
+	if err := db.Model(&entities.User{}).Where("is_certified = ?", true).Count(&stats.CertifiedUsers).Error; err != nil {
+		return nil, err
+	}
+
+	// 今日注册数
+	today := time.Now().Truncate(24 * time.Hour)
+	if err := db.Model(&entities.User{}).Where("created_at >= ?", today).Count(&stats.TodayRegistrations).Error; err != nil {
+		return nil, err
+	}
+
+	// 今日登录数
+	if err := db.Model(&entities.User{}).Where("last_login_at >= ?", today).Count(&stats.TodayLogins).Error; err != nil {
+		return nil, err
+	}
+
+	return &stats, nil
+}
+
+// GetSystemUserStatsByDateRange 获取系统指定时间范围内的用户统计信息
+func (r *GormUserRepository) GetSystemUserStatsByDateRange(ctx context.Context, startDate, endDate time.Time) (*repositories.UserStats, error) {
+	var stats repositories.UserStats
+
+	db := r.GetDB(ctx)
+
+	// 指定时间范围内的注册数
+	if err := db.Model(&entities.User{}).
+		Where("created_at >= ? AND created_at <= ?", startDate, endDate).
+		Count(&stats.TodayRegistrations).Error; err != nil {
+		return nil, err
+	}
+
+	// 指定时间范围内的登录数
+	if err := db.Model(&entities.User{}).
+		Where("last_login_at >= ? AND last_login_at <= ?", startDate, endDate).
+		Count(&stats.TodayLogins).Error; err != nil {
+		return nil, err
+	}
+
+	return &stats, nil
+}
+
+// GetSystemDailyUserStats 获取系统每日用户统计
+func (r *GormUserRepository) GetSystemDailyUserStats(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 count
+		FROM users 
+		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
+}
+
+// GetSystemMonthlyUserStats 获取系统每月用户统计
+func (r *GormUserRepository) GetSystemMonthlyUserStats(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 count
+		FROM users 
+		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
+}
+
+// GetUserCallRankingByCalls 按调用次数获取用户排行
+func (r *GormUserRepository) GetUserCallRankingByCalls(ctx context.Context, period string, limit int) ([]map[string]interface{}, error) {
+	var sql string
+	var args []interface{}
+	
+	switch period {
+	case "today":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COUNT(ac.id) as calls
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN api_calls ac ON u.id = ac.user_id 
+				AND DATE(ac.created_at) = CURRENT_DATE
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COUNT(ac.id) > 0
+			ORDER BY calls DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "month":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COUNT(ac.id) as calls
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN api_calls ac ON u.id = ac.user_id 
+				AND DATE_TRUNC('month', ac.created_at) = DATE_TRUNC('month', CURRENT_DATE)
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COUNT(ac.id) > 0
+			ORDER BY calls DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "total":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COUNT(ac.id) as calls
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN api_calls ac ON u.id = ac.user_id
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COUNT(ac.id) > 0
+			ORDER BY calls 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
+}
+
+// GetUserCallRankingByConsumption 按消费金额获取用户排行
+func (r *GormUserRepository) GetUserCallRankingByConsumption(ctx context.Context, period string, limit int) ([]map[string]interface{}, error) {
+	var sql string
+	var args []interface{}
+	
+	switch period {
+	case "today":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(wt.amount), 0) as consumption
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN wallet_transactions wt ON u.id = wt.user_id 
+				AND DATE(wt.created_at) = CURRENT_DATE
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(wt.amount), 0) > 0
+			ORDER BY consumption DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "month":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(wt.amount), 0) as consumption
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN wallet_transactions wt ON u.id = wt.user_id 
+				AND DATE_TRUNC('month', wt.created_at) = DATE_TRUNC('month', CURRENT_DATE)
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(wt.amount), 0) > 0
+			ORDER BY consumption DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "total":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(wt.amount), 0) as consumption
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN wallet_transactions wt ON u.id = wt.user_id
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(wt.amount), 0) > 0
+			ORDER BY consumption 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
+}
+
+// GetRechargeRanking 获取充值排行
+func (r *GormUserRepository) GetRechargeRanking(ctx context.Context, period string, limit int) ([]map[string]interface{}, error) {
+	var sql string
+	var args []interface{}
+	
+	switch period {
+	case "today":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(rr.amount), 0) as amount
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN recharge_records rr ON u.id = rr.user_id 
+				AND DATE(rr.created_at) = CURRENT_DATE
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(rr.amount), 0) > 0
+			ORDER BY amount DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "month":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(rr.amount), 0) as amount
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN recharge_records rr ON u.id = rr.user_id 
+				AND DATE_TRUNC('month', rr.created_at) = DATE_TRUNC('month', CURRENT_DATE)
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(rr.amount), 0) > 0
+			ORDER BY amount DESC
+			LIMIT $1
+		`
+		args = []interface{}{limit}
+	case "total":
+		sql = `
+			SELECT 
+				u.id as user_id,
+				COALESCE(ei.company_name, u.username, u.phone) as username,
+				COALESCE(SUM(rr.amount), 0) as amount
+			FROM users u
+			LEFT JOIN enterprise_infos ei ON u.id = ei.user_id
+			LEFT JOIN recharge_records rr ON u.id = rr.user_id
+			WHERE u.deleted_at IS NULL
+			GROUP BY u.id, ei.company_name, u.username, u.phone
+			HAVING COALESCE(SUM(rr.amount), 0) > 0
+			ORDER BY amount 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
+}