125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"time"
|
||
|
||
"github.com/user-management-system/internal/domain"
|
||
"github.com/user-management-system/internal/repository"
|
||
)
|
||
|
||
// StatsService 统计服务
|
||
type StatsService struct {
|
||
userRepo *repository.UserRepository
|
||
loginLogRepo *repository.LoginLogRepository
|
||
}
|
||
|
||
// NewStatsService 创建统计服务
|
||
func NewStatsService(
|
||
userRepo *repository.UserRepository,
|
||
loginLogRepo *repository.LoginLogRepository,
|
||
) *StatsService {
|
||
return &StatsService{
|
||
userRepo: userRepo,
|
||
loginLogRepo: loginLogRepo,
|
||
}
|
||
}
|
||
|
||
// UserStats 用户统计数据
|
||
type UserStats struct {
|
||
TotalUsers int64 `json:"total_users"`
|
||
ActiveUsers int64 `json:"active_users"`
|
||
InactiveUsers int64 `json:"inactive_users"`
|
||
LockedUsers int64 `json:"locked_users"`
|
||
DisabledUsers int64 `json:"disabled_users"`
|
||
NewUsersToday int64 `json:"new_users_today"`
|
||
NewUsersWeek int64 `json:"new_users_week"`
|
||
NewUsersMonth int64 `json:"new_users_month"`
|
||
}
|
||
|
||
// LoginStats 登录统计数据
|
||
type LoginStats struct {
|
||
LoginsTodaySuccess int64 `json:"logins_today_success"`
|
||
LoginsTodayFailed int64 `json:"logins_today_failed"`
|
||
LoginsWeek int64 `json:"logins_week"`
|
||
}
|
||
|
||
// DashboardStats 仪表盘综合统计
|
||
type DashboardStats struct {
|
||
Users UserStats `json:"users"`
|
||
Logins LoginStats `json:"logins"`
|
||
}
|
||
|
||
// GetUserStats 获取用户统计
|
||
func (s *StatsService) GetUserStats(ctx context.Context) (*UserStats, error) {
|
||
stats := &UserStats{}
|
||
|
||
// 统计总用户数
|
||
_, total, err := s.userRepo.List(ctx, 0, 1)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
stats.TotalUsers = total
|
||
|
||
// 按状态统计
|
||
statusCounts := map[domain.UserStatus]*int64{
|
||
domain.UserStatusActive: &stats.ActiveUsers,
|
||
domain.UserStatusInactive: &stats.InactiveUsers,
|
||
domain.UserStatusLocked: &stats.LockedUsers,
|
||
domain.UserStatusDisabled: &stats.DisabledUsers,
|
||
}
|
||
for status, countPtr := range statusCounts {
|
||
_, cnt, err := s.userRepo.ListByStatus(ctx, status, 0, 1)
|
||
if err == nil {
|
||
*countPtr = cnt
|
||
}
|
||
}
|
||
|
||
// 今日新增
|
||
stats.NewUsersToday = s.countNewUsers(ctx, daysAgo(0))
|
||
// 本周新增
|
||
stats.NewUsersWeek = s.countNewUsers(ctx, daysAgo(7))
|
||
// 本月新增
|
||
stats.NewUsersMonth = s.countNewUsers(ctx, daysAgo(30))
|
||
|
||
return stats, nil
|
||
}
|
||
|
||
// countNewUsers 统计指定时间之后的新增用户数
|
||
func (s *StatsService) countNewUsers(ctx context.Context, since time.Time) int64 {
|
||
_, count, err := s.userRepo.ListCreatedAfter(ctx, since, 0, 0)
|
||
if err != nil {
|
||
return 0
|
||
}
|
||
return count
|
||
}
|
||
|
||
// GetDashboardStats 获取仪表盘综合统计
|
||
func (s *StatsService) GetDashboardStats(ctx context.Context) (*DashboardStats, error) {
|
||
userStats, err := s.GetUserStats(ctx)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
loginStats := &LoginStats{}
|
||
// 今日登录成功/失败
|
||
today := daysAgo(0)
|
||
if s.loginLogRepo != nil {
|
||
loginStats.LoginsTodaySuccess = s.loginLogRepo.CountByResultSince(ctx, true, today)
|
||
loginStats.LoginsTodayFailed = s.loginLogRepo.CountByResultSince(ctx, false, today)
|
||
loginStats.LoginsWeek = s.loginLogRepo.CountByResultSince(ctx, true, daysAgo(7))
|
||
}
|
||
|
||
return &DashboardStats{
|
||
Users: *userStats,
|
||
Logins: *loginStats,
|
||
}, nil
|
||
}
|
||
|
||
// daysAgo 返回N天前的时间（当天0点）
|
||
func daysAgo(n int) time.Time {
|
||
now := time.Now()
|
||
start := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
||
return start.AddDate(0, 0, -n)
|
||
}
|