Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

skd/service/promotionCode.go

@@ -0,0 +1,58 @@
+package service
+
+import (
+	"91porn-server/common"
+	"91porn-server/common/constant/redisconst"
+	"91porn-server/common/log"
+	"91porn-server/models/v/usermod"
+	"91porn-server/skd/skdg"
+)
+
+// 填充推广码
+func MakePromotionCode() {
+	redisKey := redisconst.PromotionCodeKey()
+	var low int64 = 5000
+	total, err := skdg.Redis.SCard(redisKey)
+	if err != nil {
+		log.Warn("service MakePromotionCode SCard error", log.E(err), log.Any("total", total))
+		return
+	}
+	if total < low {
+		savePromotionCode(total)
+	}
+}
+
+func savePromotionCode(total int64) {
+	var max int64 = 9999
+	size := 100
+	redisKey := redisconst.PromotionCodeKey()
+	if total < max {
+		promotionCodeCollection := make([]string, size)
+		for i := 0; i < size; i++ {
+			p, err := getPromotionCode()
+			if err != nil {
+				break
+			}
+			promotionCodeCollection[i] = p
+		}
+		count, err := skdg.Redis.SAdd(redisKey, promotionCodeCollection)
+		if err != nil {
+			log.Warn("service savePromotionCode SAdd error", log.E(err), log.Any("total", total))
+			return
+		}
+		savePromotionCode(total + count)
+	}
+}
+
+// 获取不重复的推广码
+func getPromotionCode() (string, error) {
+	promotionCode := common.InvitePromotionCodeGenera()
+	u, err := usermod.FindUserPromotionCode(promotionCode)
+	if err != nil {
+		return "", err
+	}
+	if u == nil {
+		return promotionCode, nil
+	}
+	return getPromotionCode()
+}