Merge pull request #488 from LydiaCai1203/feat-calc-conversation-count

feat: 存储用户对话轮数到 redis

Author: yokowu

backend/biz/task/handler/v1/task.go

@@ -35,16 +35,16 @@ var errRoundEnded = errors.New("round ended")
 
 // TaskHandler 任务处理器
 type TaskHandler struct {
-	cfg          *config.Config
-	usecase      domain.TaskUsecase
-	userusecase  domain.UserUsecase
-	pubhost      domain.PublicHostUsecase
-	logger       *slog.Logger
-	taskflow     taskflow.Clienter
-	loki         *loki.Client
-	nls          *nls.NLS
-	taskConns    *ws.TaskConn
-	controlConns *ws.ControlConn
+	cfg           *config.Config
+	usecase       domain.TaskUsecase
+	userusecase   domain.UserUsecase
+	pubhost       domain.PublicHostUsecase
+	logger        *slog.Logger
+	taskflow      taskflow.Clienter
+	loki          *loki.Client
+	nls           *nls.NLS
+	taskConns     *ws.TaskConn
+	controlConns  *ws.ControlConn
 	taskSummary   *service.TaskSummaryService
 	idleRefresher vmidle.VMIdleRefresher
 }
@@ -77,17 +77,17 @@ func NewTaskHandler(i *do.Injector) (*TaskHandler, error) {
 	}
 
 	h := &TaskHandler{
-		cfg:          cfg,
-		usecase:      uc,
-		userusecase:  uuc,
-		pubhost:      pubhost,
-		logger:       logger.With("handler", "task.handler"),
-		taskflow:     tf,
-		loki:         lok,
-		nls:          nlsSvc,
-		taskConns:    tc,
-		controlConns: cc,
-		taskSummary:  ts,
+		cfg:           cfg,
+		usecase:       uc,
+		userusecase:   uuc,
+		pubhost:       pubhost,
+		logger:        logger.With("handler", "task.handler"),
+		taskflow:      tf,
+		loki:          lok,
+		nls:           nlsSvc,
+		taskConns:     tc,
+		controlConns:  cc,
+		taskSummary:   ts,
 		idleRefresher: ir,
 	}
 
@@ -566,6 +566,9 @@ func (h *TaskHandler) handleClientMessage(ctx context.Context, logger *slog.Logg
 		if err := h.usecase.Continue(ctx, user, task.ID, string(m.Data)); err != nil {
 			logger.With("error", err).WarnContext(ctx, "failed to push task content")
 		}
+		if err := h.usecase.IncrUserInputCount(ctx, user.ID, task.ID); err != nil {
+			logger.With("error", err).WarnContext(ctx, "failed to incr user input count")
+		}
 		h.enqueueSummary(ctx, logger, task.ID.String(), task.CreatedAt)
 
 	case consts.TaskStreamTypeUserStop:

backend/biz/task/usecase/task.go

@@ -258,6 +258,23 @@ func (a *TaskUsecase) Continue(ctx context.Context, user *domain.User, id uuid.U
 	return nil
 }
 
+// IncrUserInputCount 记录用户输入次数到 Redis Hash，并按天计数
+func (a *TaskUsecase) IncrUserInputCount(ctx context.Context, userID, taskID uuid.UUID) error {
+	// 按 task 维度计数（总量，不设过期）
+	key := fmt.Sprintf("mcai:user:%s:input_count", userID.String())
+	if err := a.redis.HIncrBy(ctx, key, taskID.String(), 1).Err(); err != nil {
+		return err
+	}
+
+	// 按天计数（用于时间范围统计，90 天过期）
+	dailyKey := fmt.Sprintf("mcai:user:%s:input_daily:%s", userID.String(), time.Now().Format("2006-01-02"))
+	pipe := a.redis.Pipeline()
+	pipe.Incr(ctx, dailyKey)
+	pipe.Expire(ctx, dailyKey, 90*24*time.Hour)
+	_, err := pipe.Exec(ctx)
+	return err
+}
+
 // Create implements domain.TaskUsecase.
 func (a *TaskUsecase) Create(ctx context.Context, user *domain.User, req domain.CreateTaskReq) (*domain.ProjectTask, error) {
 	r, err := a.taskflow.Host().IsOnline(ctx, &taskflow.IsOnlineReq[string]{
@@ -460,6 +477,10 @@ func (a *TaskUsecase) Create(ctx context.Context, user *domain.User, req domain.
 		}
 	}
 
+	if err := a.IncrUserInputCount(ctx, user.ID, pt.Edges.Task.ID); err != nil {
+		a.logger.WarnContext(ctx, "failed to incr user input count on create", "error", err)
+	}
+
 	result := cvt.From(pt, &domain.ProjectTask{})
 
 	// 通知 TaskHook（如内部项目的 git task 创建等）

backend/domain/task.go

@@ -27,6 +27,7 @@ type TaskUsecase interface {
 	AutoApprove(ctx context.Context, user *User, id uuid.UUID, approve bool) error
 	GitTask(ctx context.Context, id uuid.UUID) (*GitTask, error)
 	Delete(ctx context.Context, user *User, id uuid.UUID) error
+	IncrUserInputCount(ctx context.Context, userID, taskID uuid.UUID) error
 }
 
 // TaskRepo 任务数据访问接口