Initial commit

models/channelpackage/helper.go

@@ -0,0 +1,83 @@
+package channelpackage
+
+import (
+	"context"
+	"fmt"
+	"go.mongodb.org/mongo-driver/bson"
+	"math"
+	"math/rand"
+	"net/url"
+	"stat/common/stdLog"
+	"time"
+)
+
+func (c *ChannelPackage) IncDownCount(ctx context.Context) *stdLog.SrvErr {
+	update := bson.M{
+		"$inc": bson.M{
+			FN.ChannelPackage.DownCount: 1,
+		},
+		"$set": bson.M{
+			FN.ChannelPackage.DownLatest: time.Now(),
+		},
+	}
+	return UpdateOneChannelPackage(ctx, bson.M{FN.ChannelPackage.Id: c.Id}, update)
+}
+
+func (c *ChannelPackage) GetRandomLink(domains []string) string {
+	if len(domains) > 0 {
+		base := domains[rand.Intn(len(domains))]
+		link, _ := url.JoinPath(base, c.UrlPath)
+		return link
+	}
+	return ""
+}
+
+func (c *ChannelPackage) GetAllLink(domains []string) []string {
+	if len(domains) > 0 {
+		list := make([]string, 0)
+		for _, domain := range domains {
+			link, _ := url.JoinPath(domain, c.UrlPath)
+			list = append(list, link)
+		}
+		return list
+	}
+	return []string{}
+}
+
+func (c *ChannelPackage) CanMake(force bool) (bool, time.Duration, string) {
+	// 先判断锁定时间(强制生成不可破锁)
+	duration := ApkLockTimeDuration - time.Since(c.LockTime)
+	if duration > 0 {
+		return false, duration, fmt.Sprintf("记录锁定")
+	}
+	if force {
+		return true, 0, "强制生成"
+	}
+	if c.Status == ChanApkStatusSuccess {
+		// 已成功的记录
+		return false, duration, "已经生成成功过，无需重新生成"
+	} else if c.Status == ChanApkStatusMaking {
+		// 判断超时的记录
+		current := time.Now().Sub(c.GenTime)
+		diff := ApkNoResponseDuration - current
+		if diff <= 0 {
+			duration := time.Duration(math.Pow(3, float64(c.GenCount))) * time.Minute
+			if current > duration {
+				return true, -1, "可以生成"
+			} else {
+				return false, duration - current, "未达到可以再次生成的时间"
+			}
+		} else {
+			return false, diff, "正在生成中"
+		}
+	} else {
+		// 判断失败的记录
+		duration := time.Duration(math.Pow(3, float64(c.GenCount))) * time.Minute
+		current := time.Now().Sub(c.GenTime)
+		if current > duration {
+			return true, -1, "上次生成失败已超时，现在可以重新生成"
+		} else {
+			return false, duration - current, "上次生成失败，重新生成的时间未到"
+		}
+	}
+}