307 lines
8.3 KiB
Go
307 lines
8.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
|
|||
|
|
"nl-pms-api/internal/commonservice"
|
|||
|
|
"nl-pms-api/internal/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AdminService 运营后台：统计、用户、团队。
|
|||
|
|
type AdminService struct {
|
|||
|
|
DB *gorm.DB
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type StatsPoint struct {
|
|||
|
|
Date string `json:"date"`
|
|||
|
|
Count int64 `json:"count"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type TokenPoint struct {
|
|||
|
|
Date string `json:"date"`
|
|||
|
|
PromptTokens int64 `json:"promptTokens"`
|
|||
|
|
CompletionTokens int64 `json:"completionTokens"`
|
|||
|
|
Calls int64 `json:"calls"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type OverviewStats struct {
|
|||
|
|
UserCount int64 `json:"userCount"`
|
|||
|
|
TeamCount int64 `json:"teamCount"`
|
|||
|
|
DAUToday int64 `json:"dauToday"`
|
|||
|
|
DAUSeries []StatsPoint `json:"dauSeries"`
|
|||
|
|
TokenToday TokenPoint `json:"tokenToday"`
|
|||
|
|
TokenSeries []TokenPoint `json:"tokenSeries"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Overview 概览统计。
|
|||
|
|
func (s *AdminService) Overview(days int) (*OverviewStats, error) {
|
|||
|
|
if days <= 0 || days > 90 {
|
|||
|
|
days = 14
|
|||
|
|
}
|
|||
|
|
out := &OverviewStats{
|
|||
|
|
DAUSeries: []StatsPoint{},
|
|||
|
|
TokenSeries: []TokenPoint{},
|
|||
|
|
}
|
|||
|
|
s.DB.Model(&model.User{}).Count(&out.UserCount)
|
|||
|
|
s.DB.Model(&model.Team{}).Count(&out.TeamCount)
|
|||
|
|
|
|||
|
|
today := time.Now().In(time.Local).Format("2006-01-02")
|
|||
|
|
s.DB.Model(&model.UserDailyActive{}).Where("active_date = ?", today).Count(&out.DAUToday)
|
|||
|
|
|
|||
|
|
start := time.Now().In(time.Local).AddDate(0, 0, -(days - 1)).Format("2006-01-02")
|
|||
|
|
|
|||
|
|
type dauRow struct {
|
|||
|
|
ActiveDate string
|
|||
|
|
Cnt int64
|
|||
|
|
}
|
|||
|
|
var dauRows []dauRow
|
|||
|
|
s.DB.Model(&model.UserDailyActive{}).
|
|||
|
|
Select("active_date, COUNT(*) AS cnt").
|
|||
|
|
Where("active_date >= ?", start).
|
|||
|
|
Group("active_date").
|
|||
|
|
Order("active_date").
|
|||
|
|
Scan(&dauRows)
|
|||
|
|
dauMap := map[string]int64{}
|
|||
|
|
for _, r := range dauRows {
|
|||
|
|
dauMap[r.ActiveDate] = r.Cnt
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type tokRow struct {
|
|||
|
|
UsageDate string
|
|||
|
|
PromptTokens int64
|
|||
|
|
CompletionTokens int64
|
|||
|
|
Calls int64
|
|||
|
|
}
|
|||
|
|
var tokRows []tokRow
|
|||
|
|
s.DB.Model(&model.AIUsageDaily{}).
|
|||
|
|
Select("usage_date, SUM(prompt_tokens) AS prompt_tokens, SUM(completion_tokens) AS completion_tokens, SUM(calls) AS calls").
|
|||
|
|
Where("usage_date >= ?", start).
|
|||
|
|
Group("usage_date").
|
|||
|
|
Order("usage_date").
|
|||
|
|
Scan(&tokRows)
|
|||
|
|
tokMap := map[string]tokRow{}
|
|||
|
|
for _, r := range tokRows {
|
|||
|
|
tokMap[r.UsageDate] = r
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for i := 0; i < days; i++ {
|
|||
|
|
d := time.Now().In(time.Local).AddDate(0, 0, -(days - 1 - i)).Format("2006-01-02")
|
|||
|
|
out.DAUSeries = append(out.DAUSeries, StatsPoint{Date: d, Count: dauMap[d]})
|
|||
|
|
t := tokMap[d]
|
|||
|
|
pt := TokenPoint{
|
|||
|
|
Date: d,
|
|||
|
|
PromptTokens: t.PromptTokens,
|
|||
|
|
CompletionTokens: t.CompletionTokens,
|
|||
|
|
Calls: t.Calls,
|
|||
|
|
}
|
|||
|
|
out.TokenSeries = append(out.TokenSeries, pt)
|
|||
|
|
if d == today {
|
|||
|
|
out.TokenToday = pt
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return out, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type AdminUserItem struct {
|
|||
|
|
ID int64 `json:"id"`
|
|||
|
|
Username string `json:"username"`
|
|||
|
|
Nickname string `json:"nickname"`
|
|||
|
|
AIBanned int `json:"aiBanned"`
|
|||
|
|
Disabled int `json:"disabled"`
|
|||
|
|
LastLoginIP string `json:"lastLoginIp"`
|
|||
|
|
LastSeenAt string `json:"lastSeenAt"`
|
|||
|
|
CreatedAt string `json:"createdAt"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ListUsers 用户列表。
|
|||
|
|
func (s *AdminService) ListUsers() ([]AdminUserItem, error) {
|
|||
|
|
var users []model.User
|
|||
|
|
if err := s.DB.Order("id ASC").Find(&users).Error; err != nil {
|
|||
|
|
return nil, commonservice.Internal("QUERY_FAILED")
|
|||
|
|
}
|
|||
|
|
ids := make([]int64, 0, len(users))
|
|||
|
|
for _, u := range users {
|
|||
|
|
ids = append(ids, u.ID)
|
|||
|
|
}
|
|||
|
|
nick := map[int64]string{}
|
|||
|
|
if len(ids) > 0 {
|
|||
|
|
var profiles []model.UserProfile
|
|||
|
|
s.DB.Where("user_id IN ?", ids).Find(&profiles)
|
|||
|
|
for _, p := range profiles {
|
|||
|
|
nick[p.UserID] = p.Nickname
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
out := make([]AdminUserItem, 0, len(users))
|
|||
|
|
for _, u := range users {
|
|||
|
|
out = append(out, AdminUserItem{
|
|||
|
|
ID: u.ID,
|
|||
|
|
Username: u.Username,
|
|||
|
|
Nickname: nick[u.ID],
|
|||
|
|
AIBanned: u.AIBanned,
|
|||
|
|
Disabled: u.Disabled,
|
|||
|
|
LastLoginIP: u.LastLoginIP,
|
|||
|
|
LastSeenAt: u.LastSeenAt,
|
|||
|
|
CreatedAt: u.CreatedAt,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
return out, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PatchUser 更新用户 ai_banned / disabled（不可禁用自己）。
|
|||
|
|
func (s *AdminService) PatchUser(actorID, targetID int64, aiBanned, disabled *int) error {
|
|||
|
|
if actorID != commonservice.AdminUserID {
|
|||
|
|
return commonservice.Forbidden("FORBIDDEN")
|
|||
|
|
}
|
|||
|
|
if targetID <= 0 {
|
|||
|
|
return commonservice.BadRequest("BAD_REQUEST")
|
|||
|
|
}
|
|||
|
|
if targetID == commonservice.AdminUserID && disabled != nil && *disabled != 0 {
|
|||
|
|
return commonservice.BadRequest("CANNOT_DISABLE_ADMIN")
|
|||
|
|
}
|
|||
|
|
updates := map[string]any{}
|
|||
|
|
if aiBanned != nil {
|
|||
|
|
if *aiBanned != 0 {
|
|||
|
|
updates["ai_banned"] = 1
|
|||
|
|
} else {
|
|||
|
|
updates["ai_banned"] = 0
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if disabled != nil {
|
|||
|
|
if *disabled != 0 {
|
|||
|
|
updates["disabled"] = 1
|
|||
|
|
} else {
|
|||
|
|
updates["disabled"] = 0
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if len(updates) == 0 {
|
|||
|
|
return commonservice.BadRequest("BAD_REQUEST")
|
|||
|
|
}
|
|||
|
|
res := s.DB.Model(&model.User{}).Where("id = ?", targetID).Updates(updates)
|
|||
|
|
if res.Error != nil {
|
|||
|
|
return commonservice.Internal("SAVE_FAILED")
|
|||
|
|
}
|
|||
|
|
if res.RowsAffected == 0 {
|
|||
|
|
return commonservice.NotFound("NOT_FOUND")
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type AdminTeamItem struct {
|
|||
|
|
ID int64 `json:"id"`
|
|||
|
|
Name string `json:"name"`
|
|||
|
|
OwnerID int64 `json:"ownerId"`
|
|||
|
|
OwnerName string `json:"ownerName"`
|
|||
|
|
AIBanned int `json:"aiBanned"`
|
|||
|
|
Members int64 `json:"members"`
|
|||
|
|
CreatedAt string `json:"createdAt"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ListTeams 团队列表。
|
|||
|
|
func (s *AdminService) ListTeams() ([]AdminTeamItem, error) {
|
|||
|
|
var teams []model.Team
|
|||
|
|
if err := s.DB.Order("id ASC").Find(&teams).Error; err != nil {
|
|||
|
|
return nil, commonservice.Internal("QUERY_FAILED")
|
|||
|
|
}
|
|||
|
|
out := make([]AdminTeamItem, 0, len(teams))
|
|||
|
|
for _, t := range teams {
|
|||
|
|
item := AdminTeamItem{
|
|||
|
|
ID: t.ID, Name: t.Name, OwnerID: t.OwnerID,
|
|||
|
|
AIBanned: t.AIBanned, CreatedAt: t.CreatedAt,
|
|||
|
|
}
|
|||
|
|
var u model.User
|
|||
|
|
if s.DB.Select("username").First(&u, t.OwnerID).Error == nil {
|
|||
|
|
item.OwnerName = u.Username
|
|||
|
|
}
|
|||
|
|
s.DB.Model(&model.TeamMember{}).Where("team_id = ?", t.ID).Count(&item.Members)
|
|||
|
|
out = append(out, item)
|
|||
|
|
}
|
|||
|
|
return out, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// PatchTeam 更新团队 ai_banned。
|
|||
|
|
func (s *AdminService) PatchTeam(actorID, teamID int64, aiBanned *int) error {
|
|||
|
|
if actorID != commonservice.AdminUserID {
|
|||
|
|
return commonservice.Forbidden("FORBIDDEN")
|
|||
|
|
}
|
|||
|
|
if teamID <= 0 || aiBanned == nil {
|
|||
|
|
return commonservice.BadRequest("BAD_REQUEST")
|
|||
|
|
}
|
|||
|
|
v := 0
|
|||
|
|
if *aiBanned != 0 {
|
|||
|
|
v = 1
|
|||
|
|
}
|
|||
|
|
res := s.DB.Model(&model.Team{}).Where("id = ?", teamID).Update("ai_banned", v)
|
|||
|
|
if res.Error != nil {
|
|||
|
|
return commonservice.Internal("SAVE_FAILED")
|
|||
|
|
}
|
|||
|
|
if res.RowsAffected == 0 {
|
|||
|
|
return commonservice.NotFound("NOT_FOUND")
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AIPolicy 当前用户是否允许使用 AI。
|
|||
|
|
type AIPolicy struct {
|
|||
|
|
Allowed bool `json:"allowed"`
|
|||
|
|
Reason string `json:"reason,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetAIPolicy 综合用户禁用/禁 AI 与所属团队禁 AI。
|
|||
|
|
func (s *AdminService) GetAIPolicy(userID int64) (*AIPolicy, error) {
|
|||
|
|
var u model.User
|
|||
|
|
if err := s.DB.Select("id", "disabled", "ai_banned").First(&u, userID).Error; err != nil {
|
|||
|
|
return nil, commonservice.Unauthorized("UNAUTHORIZED")
|
|||
|
|
}
|
|||
|
|
if u.Disabled != 0 {
|
|||
|
|
return &AIPolicy{Allowed: false, Reason: "ACCOUNT_DISABLED"}, nil
|
|||
|
|
}
|
|||
|
|
if u.AIBanned != 0 {
|
|||
|
|
return &AIPolicy{Allowed: false, Reason: "USER_AI_BANNED"}, nil
|
|||
|
|
}
|
|||
|
|
var n int64
|
|||
|
|
s.DB.Table("team_members").
|
|||
|
|
Joins("JOIN teams ON teams.id = team_members.team_id").
|
|||
|
|
Where("team_members.user_id = ? AND teams.ai_banned = 1", userID).
|
|||
|
|
Count(&n)
|
|||
|
|
if n > 0 {
|
|||
|
|
return &AIPolicy{Allowed: false, Reason: "TEAM_AI_BANNED"}, nil
|
|||
|
|
}
|
|||
|
|
return &AIPolicy{Allowed: true}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ReportAIUsage 累加当日 token 用量。
|
|||
|
|
func (s *AdminService) ReportAIUsage(userID, teamID int64, provider string, prompt, completion int64, estimated bool) error {
|
|||
|
|
if userID <= 0 {
|
|||
|
|
return commonservice.Unauthorized("UNAUTHORIZED")
|
|||
|
|
}
|
|||
|
|
provider = strings.TrimSpace(provider)
|
|||
|
|
if provider == "" {
|
|||
|
|
provider = "unknown"
|
|||
|
|
}
|
|||
|
|
if prompt < 0 {
|
|||
|
|
prompt = 0
|
|||
|
|
}
|
|||
|
|
if completion < 0 {
|
|||
|
|
completion = 0
|
|||
|
|
}
|
|||
|
|
today := time.Now().In(time.Local).Format("2006-01-02")
|
|||
|
|
est := 0
|
|||
|
|
if estimated {
|
|||
|
|
est = 1
|
|||
|
|
}
|
|||
|
|
q := `INSERT INTO ai_usage_daily(user_id,team_id,usage_date,provider,prompt_tokens,completion_tokens,calls,estimated)
|
|||
|
|
VALUES(?,?,?,?,?,?,1,?)
|
|||
|
|
ON DUPLICATE KEY UPDATE
|
|||
|
|
prompt_tokens=prompt_tokens+VALUES(prompt_tokens),
|
|||
|
|
completion_tokens=completion_tokens+VALUES(completion_tokens),
|
|||
|
|
calls=calls+1,
|
|||
|
|
estimated=IF(VALUES(estimated)=1,1,estimated)`
|
|||
|
|
if err := s.DB.Exec(q, userID, teamID, today, provider, prompt, completion, est).Error; err != nil {
|
|||
|
|
return commonservice.Internal("SAVE_FAILED")
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|