fix: should always requeue

internal/bots/telegram/handlers/recap/toggle_recap.go

@@ -132,7 +132,7 @@ func (h *EnableRecapCommandHandler) Handle(c *tgbot.Context) (tgbot.Response, er
 		return nil, tgbot.NewExceptionError(err).WithMessage("聊天记录回顾功能开启失败，请稍后再试！").WithReply(c.Update.Message)
 	}
 
-	err = h.tgchats.QueueOneSendChatHistoriesRecapTaskForChatID(c.Update.Message.Chat.ID)
+	err = h.tgchats.QueueOneSendChatHistoriesRecapTaskForChatID(c.Update.Message.Chat.ID, telegram.ChatType(c.Update.Message.Chat.Type), c.Update.Message.Chat.Title)
 	if err != nil {
 		return nil, tgbot.NewExceptionError(err).WithMessage("聊天记录回顾功能开启失败，请稍后再试！").WithReply(c.Update.Message)
 	}

internal/models/tgchats/feature_flags.go

@@ -132,7 +132,7 @@ func (m *Model) HasChatHistoriesRecapEnabled(chatID int64, chatType telegram.Cha
 	return featureFlags.FeatureChatHistoriesRecap, nil
 }
 
-func (m *Model) ListChatHistoriesRecapEnabledChats() ([]int64, error) {
+func (m *Model) ListChatHistoriesRecapEnabledChats() ([]*ent.TelegramChatFeatureFlags, error) {
 	featureFlagsChats, err := m.ent.TelegramChatFeatureFlags.
 		Query().
 		Where(
@@ -144,28 +144,26 @@ func (m *Model) ListChatHistoriesRecapEnabledChats() ([]int64, error) {
 		return nil, err
 	}
 
-	return lo.Map(featureFlagsChats, func(featureFlags *ent.TelegramChatFeatureFlags, _ int) int64 {
-		return featureFlags.ChatID
-	}), nil
+	return featureFlagsChats, nil
 }
 
 func (m *Model) QueueSendChatHistoriesRecapTask() {
-	chatIDs, err := m.ListChatHistoriesRecapEnabledChats()
+	chats, err := m.ListChatHistoriesRecapEnabledChats()
 	if err != nil {
 		m.logger.Errorf("failed to list chat histories recap enabled chats: %v", err)
 		return
 	}
 
-	for _, chatID := range chatIDs {
-		err = m.QueueOneSendChatHistoriesRecapTaskForChatID(chatID)
+	for _, chat := range chats {
+		err = m.QueueOneSendChatHistoriesRecapTaskForChatID(chat.ChatID, telegram.ChatType(chat.ChatType), chat.ChatTitle)
 		if err != nil {
 			m.logger.Errorf("failed to queue send chat histories recap task: %v", err)
 			continue
 		}
 	}
 }
 
-func (m *Model) QueueOneSendChatHistoriesRecapTaskForChatID(chatID int64) error {
+func (m *Model) QueueOneSendChatHistoriesRecapTaskForChatID(chatID int64, chatType telegram.ChatType, chatTitle string) error {
 	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
 	defer cancel()
 
@@ -194,7 +192,12 @@ func (m *Model) QueueOneSendChatHistoriesRecapTaskForChatID(chatID int64) error
 
 	for _, schedule := range scheduleSets {
 		m.logger.Infof("scheduled one send chat histories recap task for %d at %s", chatID, schedule)
-		return m.digger.BuryUtil(ctx, timecapsules.AutoRecapCapsule{ChatID: chatID}, schedule.UnixMilli())
+
+		return m.digger.BuryUtil(ctx, timecapsules.AutoRecapCapsule{
+			ChatID:    chatID,
+			ChatType:  chatType,
+			ChatTitle: chatTitle,
+		}, schedule.UnixMilli())
 	}
 
 	return nil

internal/models/tgchats/feature_flags_test.go

@@ -4,9 +4,11 @@ import (
 	"context"
 	"testing"
 
+	"github.com/nekomeowww/insights-bot/ent"
 	"github.com/nekomeowww/insights-bot/ent/telegramchatfeatureflags"
 	"github.com/nekomeowww/insights-bot/pkg/types/telegram"
 	"github.com/nekomeowww/insights-bot/pkg/utils"
+	"github.com/samber/lo"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -85,8 +87,8 @@ func TestListChatHistoriesRecapEnabledChats(t *testing.T) {
 		assert.NoError(err)
 	}()
 
-	chatIDs, err := model.ListChatHistoriesRecapEnabledChats()
+	chats, err := model.ListChatHistoriesRecapEnabledChats()
 	require.NoError(err)
-	require.Len(chatIDs, 3)
-	assert.ElementsMatch([]int64{chatID1, chatID2, chatID3}, chatIDs)
+	require.Len(chats, 3)
+	assert.ElementsMatch([]int64{chatID1, chatID2, chatID3}, lo.Map(chats, func(item *ent.TelegramChatFeatureFlags, _ int) int64 { return item.ChatID }))
 }

internal/services/autorecap/autorecap.go

@@ -71,6 +71,39 @@ func (m *AutoRecapService) sendChatHistoriesRecap(
 	digger *timecapsule.TimeCapsuleDigger[timecapsules.AutoRecapCapsule],
 	capsule *timecapsule.TimeCapsule[timecapsules.AutoRecapCapsule],
 ) {
+	var enabled bool
+
+	_, err := lo.Attempt(10, func(index int) error {
+		var err error
+		enabled, err = m.tgchats.HasChatHistoriesRecapEnabled(capsule.Payload.ChatID, capsule.Payload.ChatType, capsule.Payload.ChatTitle)
+		if err != nil {
+			m.logger.Errorf("failed to check chat histories recap enabled: %v", err)
+			return err
+		}
+
+		return nil
+	})
+	if err != nil {
+		// requeue if failed
+		err = m.tgchats.QueueOneSendChatHistoriesRecapTaskForChatID(capsule.Payload.ChatID, capsule.Payload.ChatType, capsule.Payload.ChatTitle)
+		if err != nil {
+			m.logger.Errorf("failed to queue one send chat histories recap task for chat %d: %v", capsule.Payload.ChatID, err)
+		}
+
+		m.logger.Errorf("failed to check chat histories recap enabled: %v", err)
+
+		return
+	}
+	if !enabled {
+		return
+	}
+
+	// always requeue
+	err = m.tgchats.QueueOneSendChatHistoriesRecapTaskForChatID(capsule.Payload.ChatID, capsule.Payload.ChatType, capsule.Payload.ChatTitle)
+	if err != nil {
+		m.logger.Errorf("failed to queue one send chat histories recap task for chat %d: %v", capsule.Payload.ChatID, err)
+	}
+
 	m.logger.Infof("generating chat histories recap for chat %d", capsule.Payload.ChatID)
 
 	histories, err := m.chathistories.FindLastSixHourChatHistories(capsule.Payload.ChatID)
@@ -118,9 +151,4 @@ func (m *AutoRecapService) sendChatHistoriesRecap(
 			m.logger.Errorf("failed to send chat histories recap: %v", err)
 		}
 	}
-
-	err = m.tgchats.QueueOneSendChatHistoriesRecapTaskForChatID(capsule.Payload.ChatID)
-	if err != nil {
-		m.logger.Errorf("failed to queue one send chat histories recap task for chat %d: %v", capsule.Payload.ChatID, err)
-	}
 }

pkg/types/timecapsules/auto_recap.go

@@ -1,5 +1,9 @@
 package timecapsules
 
+import "github.com/nekomeowww/insights-bot/pkg/types/telegram"
+
 type AutoRecapCapsule struct {
-	ChatID int64 `json:"chat_id"`
+	ChatID    int64             `json:"chat_id"`
+	ChatType  telegram.ChatType `json:"chat_type"`
+	ChatTitle string            `json:"chat_title"`
 }