feat: Implement core backend services for team owner management, SQLite persistence, and logging, alongside frontend monitoring and record views.

2026-01-30 18:15:50 +08:00
parent e61430b60d
commit 165c6d69b9
7 changed files with 844 additions and 125 deletions
--- a/backend/internal/api/team_process.go
+++ b/backend/internal/api/team_process.go
@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
-	"strings"
 	"sync"
 	"sync/atomic"
 	"time"
@@ -187,7 +186,17 @@ func runTeamProcess(req TeamProcessRequest) {
 		workerCount = 2 // 默认 2 个并发
 	}

-	logger.Info(fmt.Sprintf("Starting Team process: %d owners, %d concurrent workers", totalOwners, workerCount), "", "team")
+	// 创建批次记录
+	var batchID int64
+	if database.Instance != nil {
+		var err error
+		batchID, err = database.Instance.CreateBatchRun(totalOwners)
+		if err != nil {
+			logger.Error(fmt.Sprintf("创建批次记录失败: %v", err), "", "team")
+		}
+	}
+
+	logger.Info(fmt.Sprintf("开始批量处理: 共 %d 个 Team, 并发数: %d", totalOwners, workerCount), "", "team")

 	// 任务队列
 	taskChan := make(chan int, totalOwners)
@@ -228,13 +237,42 @@ func runTeamProcess(req TeamProcessRequest) {
 		close(resultChan)
 	}()

+	// 统计总数
+	var totalRegistered, totalAddedToS2A int
+	var allErrors []string
+
 	for result := range resultChan {
 		teamProcessState.mu.Lock()
 		teamProcessState.Results = append(teamProcessState.Results, result)
 		teamProcessState.mu.Unlock()
+
+		totalRegistered += result.Registered
+		totalAddedToS2A += result.AddedToS2A
+		allErrors = append(allErrors, result.Errors...)
 	}

-	logger.Success(fmt.Sprintf("Team process complete: %d/%d teams processed", teamProcessState.Completed, totalOwners), "", "team")
+	// 更新批次记录
+	if database.Instance != nil && batchID > 0 {
+		errorsStr := ""
+		if len(allErrors) > 0 {
+			// 只保留前10条错误
+			if len(allErrors) > 10 {
+				allErrors = allErrors[:10]
+			}
+			errorsStr = fmt.Sprintf("%v", allErrors)
+		}
+		database.Instance.UpdateBatchRun(batchID, totalRegistered, totalAddedToS2A, errorsStr)
+	}
+
+	// 计算成功率
+	expectedTotal := totalOwners * req.MembersPerTeam
+	successRate := float64(0)
+	if expectedTotal > 0 {
+		successRate = float64(totalAddedToS2A) / float64(expectedTotal) * 100
+	}
+
+	logger.Success(fmt.Sprintf("批量处理完成: %d/%d 个 Team | 注册: %d, 入库: %d, 成功率: %.1f%%",
+		teamProcessState.Completed, totalOwners, totalRegistered, totalAddedToS2A, successRate), "", "team")
 }

 // processSingleTeam 处理单个 Team
@@ -249,7 +287,24 @@ func processSingleTeam(idx int, req TeamProcessRequest) TeamProcessResult {
 	}

 	logPrefix := fmt.Sprintf("[Team %d]", idx+1)
-	logger.Info(fmt.Sprintf("%s Starting with owner: %s", logPrefix, owner.Email), owner.Email, "team")
+	logger.Info(fmt.Sprintf("%s 开始处理 | 母号: %s", logPrefix, owner.Email), owner.Email, "team")
+
+	// 标记 owner 为处理中
+	if database.Instance != nil {
+		database.Instance.MarkOwnerAsProcessing(owner.Email)
+	}
+
+	// 处理失败时的清理函数
+	markOwnerResult := func(success bool) {
+		if database.Instance != nil {
+			if success {
+				database.Instance.MarkOwnerAsUsed(owner.Email)
+			} else {
+				// 失败时恢复为 valid，允许重试
+				database.Instance.MarkOwnerAsFailed(owner.Email)
+			}
+		}
+	}

 	// Step 1: 获取 Team ID（优先使用已存储的 account_id）
 	var teamID string
@@ -268,92 +323,102 @@ func processSingleTeam(idx int, req TeamProcessRequest) TeamProcessResult {
 			result.Errors = append(result.Errors, fmt.Sprintf("获取 Team ID 失败: %v", err))
 			result.DurationMs = time.Since(startTime).Milliseconds()
 			logger.Error(fmt.Sprintf("%s Failed to get Team ID: %v", logPrefix, err), owner.Email, "team")
+			markOwnerResult(false)
 			return result
 		}
 		logger.Success(fmt.Sprintf("%s 获取到 Team ID: %s", logPrefix, teamID), owner.Email, "team")
 	}
 	result.TeamID = teamID

-	// Step 2: 生成成员邮箱并发送邀请
+	// Step 2: 并发注册成员
+	// 每个成员：邀请 → 注册，失败重试1次
+	// Team 有4次额外补救机会
 	type MemberAccount struct {
 		Email    string
 		Password string
 		Success  bool
 	}
 	children := make([]MemberAccount, req.MembersPerTeam)
-	for i := 0; i < req.MembersPerTeam; i++ {
-		children[i].Email = mail.GenerateEmail()
-		children[i].Password = register.GeneratePassword()
-		logger.Info(fmt.Sprintf("%s [Member %d] Email: %s", logPrefix, i+1, children[i].Email), children[i].Email, "team")
-	}
-
-	// 发送邀请
-	inviteEmails := make([]string, req.MembersPerTeam)
-	for i, c := range children {
-		inviteEmails[i] = c.Email
-	}
-	if err := inviter.SendInvites(inviteEmails); err != nil {
-		result.Errors = append(result.Errors, fmt.Sprintf("发送邀请失败: %v", err))
-		result.DurationMs = time.Since(startTime).Milliseconds()
-		return result
-	}
-	logger.Success(fmt.Sprintf("%s Sent %d invite(s)", logPrefix, len(inviteEmails)), owner.Email, "team")
-
-	// Step 3: 并发注册成员
+	var memberMu sync.Mutex
 	var memberWg sync.WaitGroup
-	memberMutex := sync.Mutex{}

-	for i := range children {
-		memberWg.Add(1)
-		go func(memberIdx int) {
-			defer memberWg.Done()
+	// 注册单个成员的函数（带1次重试）
+	registerMember := func(memberIdx int, email, password string) bool {
+		name := register.GenerateName()
+		birthdate := register.GenerateBirthdate()

-			memberMutex.Lock()
-			email := children[memberIdx].Email
-			password := children[memberIdx].Password
-			memberMutex.Unlock()
-
-			name := register.GenerateName()
-			birthdate := register.GenerateBirthdate()
-
-			// 重试逻辑
-			for attempt := 0; attempt < 3; attempt++ {
-				if !teamProcessState.Running {
-					return
-				}
-
-				if attempt > 0 {
-					email = mail.GenerateEmail()
-					password = register.GeneratePassword()
-					logger.Info(fmt.Sprintf("%s [Member %d] Retry %d: %s", logPrefix, memberIdx+1, attempt, email), email, "team")
-
-					if err := inviter.SendInvites([]string{email}); err != nil {
-						continue
-					}
-				}
-
-				_, err := registerWithTimeout(email, password, name, birthdate, req.Proxy)
-				if err != nil {
-					if strings.Contains(err.Error(), "验证码") {
-						continue
-					}
-					result.Errors = append(result.Errors, fmt.Sprintf("Member %d: %v", memberIdx+1, err))
-					return
-				}
-
-				memberMutex.Lock()
-				children[memberIdx].Email = email
-				children[memberIdx].Password = password
-				children[memberIdx].Success = true
-				memberMutex.Unlock()
-
-				logger.Success(fmt.Sprintf("%s [Member %d] Registered", logPrefix, memberIdx+1), email, "team")
-				return
+		for attempt := 0; attempt < 2; attempt++ { // 最多尝试2次（首次+1次重试）
+			if !teamProcessState.Running {
+				return false
 			}
+
+			currentEmail := email
+			currentPassword := password
+			if attempt > 0 {
+				// 重试时使用新邮箱
+				currentEmail = mail.GenerateEmail()
+				currentPassword = register.GeneratePassword()
+				logger.Warning(fmt.Sprintf("%s [成员 %d] 重试，新邮箱: %s", logPrefix, memberIdx+1, currentEmail), currentEmail, "team")
+			}
+
+			// 发送邀请
+			if err := inviter.SendInvites([]string{currentEmail}); err != nil {
+				logger.Error(fmt.Sprintf("%s [成员 %d] 邀请失败: %v", logPrefix, memberIdx+1, err), currentEmail, "team")
+				continue
+			}
+
+			// 注册
+			_, err := registerWithTimeout(currentEmail, currentPassword, name, birthdate, req.Proxy)
+			if err != nil {
+				logger.Error(fmt.Sprintf("%s [成员 %d] 注册失败: %v", logPrefix, memberIdx+1, err), currentEmail, "team")
+				continue
+			}
+
+			// 成功
+			memberMu.Lock()
+			children[memberIdx] = MemberAccount{Email: currentEmail, Password: currentPassword, Success: true}
+			memberMu.Unlock()
+			logger.Success(fmt.Sprintf("%s [成员 %d] ✓ 注册成功", logPrefix, memberIdx+1), currentEmail, "team")
+			return true
+		}
+		return false
+	}
+
+	// 第一轮：并发注册4个成员
+	logger.Info(fmt.Sprintf("%s 开始并发注册 %d 个成员", logPrefix, req.MembersPerTeam), owner.Email, "team")
+	for i := 0; i < req.MembersPerTeam; i++ {
+		memberWg.Add(1)
+		go func(idx int) {
+			defer memberWg.Done()
+			email := mail.GenerateEmail()
+			password := register.GeneratePassword()
+			logger.Info(fmt.Sprintf("%s [成员 %d] 邮箱: %s", logPrefix, idx+1, email), email, "team")
+			registerMember(idx, email, password)
 		}(i)
 	}
 	memberWg.Wait()

+	// 统计失败的成员
+	failedSlots := make([]int, 0)
+	for i, c := range children {
+		if !c.Success {
+			failedSlots = append(failedSlots, i)
+		}
+	}
+
+	// 第二轮：Team 有 4 次额外补救机会
+	teamRetries := 4
+	for retry := 0; retry < teamRetries && len(failedSlots) > 0 && teamProcessState.Running; retry++ {
+		slotIdx := failedSlots[0]
+		logger.Warning(fmt.Sprintf("%s [补救 %d/%d] 尝试补充成员 %d", logPrefix, retry+1, teamRetries, slotIdx+1), owner.Email, "team")
+
+		email := mail.GenerateEmail()
+		password := register.GeneratePassword()
+		if registerMember(slotIdx, email, password) {
+			failedSlots = failedSlots[1:] // 成功，移除这个槽位
+		}
+	}
+
 	// 统计注册成功数
 	registeredChildren := make([]MemberAccount, 0)
 	for _, c := range children {
@@ -363,7 +428,11 @@ func processSingleTeam(idx int, req TeamProcessRequest) TeamProcessResult {
 			result.Registered++
 		}
 	}
-	logger.Info(fmt.Sprintf("%s Registered: %d/%d", logPrefix, result.Registered, req.MembersPerTeam), owner.Email, "team")
+
+	if len(failedSlots) > 0 {
+		result.Errors = append(result.Errors, fmt.Sprintf("%d 个成员注册失败", len(failedSlots)))
+	}
+	logger.Info(fmt.Sprintf("%s 注册完成: %d/%d 成功", logPrefix, result.Registered, req.MembersPerTeam), owner.Email, "team")

 	// Step 4: S2A 授权入库（成员）
 	for i, child := range registeredChildren {
@@ -407,7 +476,7 @@ func processSingleTeam(idx int, req TeamProcessRequest) TeamProcessResult {
 		}

 		result.AddedToS2A++
-		logger.Success(fmt.Sprintf("%s [Member %d] Added to S2A", logPrefix, i+1), child.Email, "team")
+		logger.Success(fmt.Sprintf("%s [成员 %d] ✓ 入库成功", logPrefix, i+1), child.Email, "team")
 	}

 	// Step 5: 母号也入库（如果开启）
@@ -442,14 +511,18 @@ func processSingleTeam(idx int, req TeamProcessRequest) TeamProcessResult {
 					result.Errors = append(result.Errors, fmt.Sprintf("Owner S2A: %v", err))
 				} else {
 					result.AddedToS2A++
-					logger.Success(fmt.Sprintf("%s [Owner] Added to S2A", logPrefix), owner.Email, "team")
+					logger.Success(fmt.Sprintf("%s [母号] ✓ 入库成功", logPrefix), owner.Email, "team")
 				}
 			}
 		}
 	}

 	result.DurationMs = time.Since(startTime).Milliseconds()
-	logger.Success(fmt.Sprintf("%s Complete: %d registered, %d in S2A", logPrefix, result.Registered, result.AddedToS2A), owner.Email, "team")
+	logger.Success(fmt.Sprintf("%s 完成 | 注册: %d, 入库: %d, 耗时: %.1fs", logPrefix, result.Registered, result.AddedToS2A, float64(result.DurationMs)/1000), owner.Email, "team")
+
+	// 根据入库结果标记 owner 状态
+	// 只要有任何一个账号入库成功，就标记为已使用
+	markOwnerResult(result.AddedToS2A > 0)

 	return result
 }