增加套餐相关代码

2025-11-13 07:10:25 +08:00 · 2021-10-29 14:02:40 +08:00
parent a0ff24adb6
commit d36f7d01df
14 changed files with 504 additions and 2 deletions
--- a/internal/db/models/user_plan_dao.go
+++ b/internal/db/models/user_plan_dao.go
@@ -0,0 +1,138 @@
+package models
+
+import (
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"time"
+)
+
+const (
+	UserPlanStateEnabled  = 1 // 已启用
+	UserPlanStateDisabled = 0 // 已禁用
+)
+
+type UserPlanDAO dbs.DAO
+
+func NewUserPlanDAO() *UserPlanDAO {
+	return dbs.NewDAO(&UserPlanDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeUserPlans",
+			Model:  new(UserPlan),
+			PkName: "id",
+		},
+	}).(*UserPlanDAO)
+}
+
+var SharedUserPlanDAO *UserPlanDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedUserPlanDAO = NewUserPlanDAO()
+	})
+}
+
+// EnableUserPlan 启用条目
+func (this *UserPlanDAO) EnableUserPlan(tx *dbs.Tx, id uint64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", UserPlanStateEnabled).
+		Update()
+	return err
+}
+
+// DisableUserPlan 禁用条目
+func (this *UserPlanDAO) DisableUserPlan(tx *dbs.Tx, id int64) error {
+	_, err := this.Query(tx).
+		Pk(id).
+		Set("state", UserPlanStateDisabled).
+		Update()
+	return err
+}
+
+// FindEnabledUserPlan 查找启用中的条目
+func (this *UserPlanDAO) FindEnabledUserPlan(tx *dbs.Tx, id int64) (*UserPlan, error) {
+	result, err := this.Query(tx).
+		Pk(id).
+		Attr("state", UserPlanStateEnabled).
+		Find()
+	if result == nil {
+		return nil, err
+	}
+	return result.(*UserPlan), err
+}
+
+// CountAllEnabledUserPlans 计算套餐数量
+func (this *UserPlanDAO) CountAllEnabledUserPlans(tx *dbs.Tx, isAvailable bool, isExpired bool, expiringDays int32) (int64, error) {
+	var query = this.Query(tx).
+		State(UserPlanStateEnabled).
+		Where("userId IN (SELECT id FROM " + SharedUserDAO.Table + " WHERE state=1)").
+		Where("planId IN (SELECT id FROM " + SharedPlanDAO.Table + " WHERE state=1)")
+	var today = timeutil.Format("Y-m-d")
+	if isAvailable {
+		query.Gte("dayTo", today)
+	}
+	if isExpired {
+		query.Lt("dayTo", today)
+	}
+	if expiringDays > 0 {
+		var expiringDay = timeutil.Format("Y-m-d", time.Now().AddDate(0, 0, int(expiringDays)))
+		query.Gte("dayTo", today)
+		query.Lte("dayTo", expiringDay)
+	}
+	return query.Count()
+}
+
+// ListEnabledUserPlans 列出单页套餐
+func (this *UserPlanDAO) ListEnabledUserPlans(tx *dbs.Tx, isAvailable bool, isExpired bool, expiringDays int32, offset int64, size int64) (result []*UserPlan, err error) {
+	var query = this.Query(tx).
+		State(UserPlanStateEnabled).
+		Where("userId IN (SELECT id FROM " + SharedUserDAO.Table + " WHERE state=1)").
+		Where("planId IN (SELECT id FROM " + SharedPlanDAO.Table + " WHERE state=1)")
+	var today = timeutil.Format("Y-m-d")
+	if isAvailable {
+		query.Gte("dayTo", today)
+	}
+	if isExpired {
+		query.Lt("dayTo", today)
+	}
+	if expiringDays > 0 {
+		var expiringDay = timeutil.Format("Y-m-d", time.Now().AddDate(0, 0, int(expiringDays)))
+		query.Gte("dayTo", today)
+		query.Lte("dayTo", expiringDay)
+	}
+	_, err = query.
+		DescPk().
+		Offset(offset).
+		Limit(size).
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// CreateUserPlan 创建套餐
+func (this *UserPlanDAO) CreateUserPlan(tx *dbs.Tx, userId int64, planId int64, dayTo string) (int64, error) {
+	var op = NewUserPlanOperator()
+	op.UserId = userId
+	op.PlanId = planId
+	op.DayTo = dayTo
+	op.IsOn = true
+	op.State = UserStateEnabled
+	return this.SaveInt64(tx, op)
+}
+
+// UpdateUserPlan 修改套餐
+func (this *UserPlanDAO) UpdateUserPlan(tx *dbs.Tx, userPlanId int64, planId int64, dayTo string, isOn bool) error {
+	if userPlanId <= 0 {
+		return errors.New("invalid userPlanId")
+	}
+	var op = NewUserPlanOperator()
+	op.Id = userPlanId
+	op.PlanId = planId
+	op.DayTo = dayTo
+	op.IsOn = isOn
+	return this.Save(tx, op)
+}