init: 1.0.0

internal/logic/public/user/queryUserSubscribeLogic.go

@@ -0,0 +1,83 @@
+package user
+
+import (
+	"context"
+	"time"
+
+	"github.com/perfect-panel/ppanel-server/pkg/constant"
+
+	"github.com/perfect-panel/ppanel-server/internal/model/user"
+	"github.com/perfect-panel/ppanel-server/internal/svc"
+	"github.com/perfect-panel/ppanel-server/internal/types"
+	"github.com/perfect-panel/ppanel-server/pkg/logger"
+	"github.com/perfect-panel/ppanel-server/pkg/tool"
+	"github.com/perfect-panel/ppanel-server/pkg/xerr"
+	"github.com/pkg/errors"
+)
+
+type QueryUserSubscribeLogic struct {
+	logger.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+// Query User Subscribe
+func NewQueryUserSubscribeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryUserSubscribeLogic {
+	return &QueryUserSubscribeLogic{
+		Logger: logger.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *QueryUserSubscribeLogic) QueryUserSubscribe() (resp *types.QueryUserSubscribeListResponse, err error) {
+	u, ok := l.ctx.Value(constant.CtxKeyUser).(*user.User)
+	if !ok {
+		logger.Error("current user is not found in context")
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidAccess), "Invalid Access")
+	}
+	data, err := l.svcCtx.UserModel.QueryUserSubscribe(l.ctx, u.Id, 1, 0)
+	if err != nil {
+		l.Errorw("[QueryUserSubscribeLogic] Query User Subscribe Error:", logger.Field("err", err.Error()))
+		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "Query User Subscribe Error")
+	}
+
+	resp = &types.QueryUserSubscribeListResponse{
+		List:  make([]types.UserSubscribe, 0),
+		Total: int64(len(data)),
+	}
+
+	for _, item := range data {
+		var sub types.UserSubscribe
+		tool.DeepCopy(&sub, item)
+		sub.ResetTime = calculateNextResetTime(&sub)
+		resp.List = append(resp.List, sub)
+	}
+	return
+}
+
+// 计算下次重置时间
+func calculateNextResetTime(sub *types.UserSubscribe) int64 {
+	startTime := time.UnixMilli(sub.StartTime)
+	now := time.Now()
+	switch sub.Subscribe.ResetCycle {
+	case 0:
+		return 0
+	case 1:
+		return time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, now.Location()).UnixMilli()
+	case 2:
+		if startTime.Day() > now.Day() {
+			return time.Date(now.Year(), now.Month(), startTime.Day(), 0, 0, 0, 0, now.Location()).UnixMilli()
+		} else {
+			return time.Date(now.Year(), now.Month()+1, startTime.Day(), 0, 0, 0, 0, now.Location()).UnixMilli()
+		}
+	case 3:
+		targetTime := time.Date(now.Year(), startTime.Month(), startTime.Day(), 0, 0, 0, 0, now.Location())
+		if targetTime.Before(now) {
+			targetTime = time.Date(now.Year()+1, startTime.Month(), startTime.Day(), 0, 0, 0, 0, now.Location())
+		}
+		return targetTime.UnixMilli()
+	default:
+		return 0
+	}
+}