EdgeAPI/internal/db/models/user_plan_dao.go

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)
}