feat: dynamically update step of ai summary

Author: Crnaneo

apis/hole/apis.go

@@ -992,12 +992,12 @@ func GenerateSummary(c *fiber.Ctx) error {
 		})
 	}
 	if forceRefresh {
-		err := DeleteCache(fmt.Sprintf("AISummary_%d_v%d", id, hole.Reply-hole.Reply%config.Config.SummarySteps))
+		err := DeleteCache(hole.SummaryCacheName())
 		if err != nil {
 			log.Err(err).Msg("AISummary: delete cache err")
 		}
 	}
-	if GetCache(fmt.Sprintf("AISummary_%d_v%d", id, hole.Reply-hole.Reply%config.Config.SummarySteps), &cachedData) {
+	if GetCache(hole.SummaryCacheName(), &cachedData) {
 
 		switch cachedData.Code {
 		case 1000:
@@ -1053,22 +1053,22 @@ func GenerateSummary(c *fiber.Ctx) error {
 							Msg("AISummary: hole id error")
 						cachedData.Data.HoleID = id
 					}
-					err = SetCache(fmt.Sprintf("AISummary_%d_v%d", id, hole.Reply-hole.Reply%config.Config.SummarySteps), cachedData, 24*time.Hour)
+					err = SetCache(hole.SummaryCacheName(), cachedData, 24*time.Hour)
 					if err != nil {
 						log.Err(err).Msg("AISummary: set cache err")
 					}
 				}
 				return c.Status(200).JSON(cachedData)
 
 			} else {
-				err := DeleteCache(fmt.Sprintf("AISummary_%d_v%d", id, hole.Reply-hole.Reply%config.Config.SummarySteps))
+				err := DeleteCache(hole.SummaryCacheName())
 				if err != nil {
 					log.Err(err).Msg("AISummary: delete cache err")
 				}
 			}
 
 		default:
-			err := DeleteCache("AISummary" + strconv.Itoa(id))
+			err := DeleteCache(hole.SummaryCacheName())
 			if err != nil {
 				log.Err(err).Msg("AISummary: delete cache err")
 			}
@@ -1185,7 +1185,7 @@ func GenerateSummary(c *fiber.Ctx) error {
 	case 1000, 1001:
 		var cache Summary
 		cache.Code = 1001
-		err := SetCache(fmt.Sprintf("AISummary_%d_v%d", id, hole.Reply-hole.Reply%config.Config.SummarySteps), cache, 24*time.Hour)
+		err := SetCache(hole.SummaryCacheName(), cache, 24*time.Hour)
 		if err != nil {
 			log.Err(err).Msg("AISummary: set cache err")
 		}

config/config.go

@@ -59,6 +59,9 @@ var Config struct {
 	SummaryFloorLimit            int      `env:"SUMMARY_FLOOR_LIMIT" envDefault:"15"`
 	SummaryContentLimit          int64    `env:"SUMMARY_CONTENT_LIMIT" envDefault:"500"`
 	SummarySteps                 int      `env:"SUMMARY_STEPS" envDefault:"5"`
+	SummaryReplyBoundary1        int      `env:"SUMMARY_REPLY_BOUNDARY_1" envDefault:"50"`
+	SummaryReplyBoundary2        int      `env:"SUMMARY_REPLY_BOUNDARY_1" envDefault:"200"`
+	SummaryReplyBoundary3        int      `env:"SUMMARY_REPLY_BOUNDARY_1" envDefault:"1000"`
 	SummaryLogLimit              int      `env:"SUMMARY_LOG_LIMIT" envDefault:"1000"`
 	WhiteListUserIds             []int    `env:"WHITE_LIST_USER_IDS"`
 }

models/hole.go

@@ -98,6 +98,20 @@ func (hole *Hole) CacheName() string {
 	return fmt.Sprintf("hole_%d", hole.ID)
 }
 
+func (hole *Hole) SummaryCacheName() string {
+	var step int
+	if hole.Reply <= config.Config.SummaryReplyBoundary1 {
+		step = config.Config.SummarySteps
+	} else if hole.Reply > config.Config.SummaryReplyBoundary1 && hole.Reply <= config.Config.SummaryReplyBoundary2 {
+		step = config.Config.SummarySteps * 2
+	} else if hole.Reply > config.Config.SummaryReplyBoundary2 && hole.Reply <= config.Config.SummaryReplyBoundary3 {
+		step = config.Config.SummarySteps * 4
+	} else {
+		step = config.Config.SummarySteps * 10
+	}
+	return fmt.Sprintf("AISummary_%d_v%d", hole.ID, hole.Reply-hole.Reply%step)
+}
+
 type Holes []*Hole
 
 func IsHolesExist(tx *gorm.DB, holeID []int) bool {