1.4.5.2

EdgeAPI/internal/db/models/ad_package_dao_plus.go

@@ -0,0 +1,131 @@
+// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package models
+
+import (
+	"errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/iwind/TeaGo/dbs"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+)
+
+// CreatePackage 创建高防产品
+func (this *ADPackageDAO) CreatePackage(tx *dbs.Tx, networkId int64, protectionBandwidthSize int32, protectionBandwidthUnit userconfigs.ADPackageSizeUnit, serverBandwidthSize int32, serverBandwidthUnit userconfigs.ADPackageSizeUnit) (int64, error) {
+	var op = NewADPackageOperator()
+	op.NetworkId = networkId
+	op.ProtectionBandwidthSize = protectionBandwidthSize
+	op.ProtectionBandwidthUnit = protectionBandwidthUnit
+	op.ProtectionBandwidthBits = userconfigs.ADPackageSizeBits(protectionBandwidthSize, protectionBandwidthUnit)
+	op.ServerBandwidthSize = serverBandwidthSize
+	op.ServerBandwidthUnit = serverBandwidthUnit
+	op.ServerBandwidthBits = userconfigs.ADPackageSizeBits(serverBandwidthSize, serverBandwidthUnit)
+
+	op.IsOn = true
+	op.State = ADPackageStateEnabled
+	return this.SaveInt64(tx, op)
+}
+
+// UpdatePackage 修改高防产品
+func (this *ADPackageDAO) UpdatePackage(tx *dbs.Tx, packageId int64, isOn bool, networkId int64, protectionBandwidthSize int32, protectionBandwidthUnit userconfigs.ADPackageSizeUnit, serverBandwidthSize int32, serverBandwidthUnit userconfigs.ADPackageSizeUnit) error {
+	if packageId <= 0 {
+		return errors.New("invalid packageId")
+	}
+
+	var op = NewADPackageOperator()
+	op.Id = packageId
+	op.NetworkId = networkId
+	op.ProtectionBandwidthSize = protectionBandwidthSize
+	op.ProtectionBandwidthUnit = protectionBandwidthUnit
+	op.ProtectionBandwidthBits = userconfigs.ADPackageSizeBits(protectionBandwidthSize, protectionBandwidthUnit)
+	op.ServerBandwidthSize = serverBandwidthSize
+	op.ServerBandwidthUnit = serverBandwidthUnit
+	op.ServerBandwidthBits = userconfigs.ADPackageSizeBits(serverBandwidthSize, serverBandwidthUnit)
+
+	op.IsOn = isOn
+	return this.Save(tx, op)
+}
+
+// CountAllPackages 查询高防产品数量
+func (this *ADPackageDAO) CountAllPackages(tx *dbs.Tx, networkId int64) (int64, error) {
+	var query = this.Query(tx).
+		State(ADPackageStateEnabled)
+
+	if networkId > 0 {
+		query.Attr("networkId", networkId)
+	}
+
+	query.Where("networkId IN (SELECT id FROM " + SharedADNetworkDAO.Table + " WHERE state=1)")
+
+	return query.Count()
+}
+
+// ListPackages 列出单页高防产品
+func (this *ADPackageDAO) ListPackages(tx *dbs.Tx, networkId int64, offset int64, size int64) (result []*ADPackage, err error) {
+	var query = this.Query(tx).
+		State(ADPackageStateEnabled)
+	if networkId > 0 {
+		query.Attr("networkId", networkId)
+	}
+	_, err = query.
+		Where("networkId IN (SELECT id FROM " + SharedADNetworkDAO.Table + " WHERE state=1)").
+		Asc("networkId").
+		Asc("protectionBandwidthSize").
+		Asc("serverBandwidthSize").
+		AscPk().
+		Offset(offset).
+		Limit(size).
+		Slice(&result).
+		FindAll()
+	return
+}
+
+// CountAllIdlePackages 查询可用的产品数量
+func (this *ADPackageDAO) CountAllIdlePackages(tx *dbs.Tx) (int64, error) {
+	return this.Query(tx).
+		State(ADPackageStateEnabled).
+		Attr("isOn", true).
+		Where("networkId IN (SELECT id FROM "+SharedADNetworkDAO.Table+" WHERE state=1 AND isOn=1)").
+		Where("id IN (SELECT DISTINCT packageId FROM "+SharedADPackageInstanceDAO.Table+" WHERE state=1 AND isOn=1 AND (userId=0 OR userDayTo IS NULL OR userDayTo<:endDay))"). // 确保没有正在使用
+		Param("endDay", timeutil.Format("Ymd")).
+		Where("id IN (SELECT DISTINCT packageId FROM " + SharedADPackagePriceDAO.Table + " WHERE price>0 )"). // 已设定价格
+		Count()
+}
+
+// FindAllIdlePackages 列出所有可用的高防产品
+func (this *ADPackageDAO) FindAllIdlePackages(tx *dbs.Tx) (result []*ADPackage, err error) {
+	_, err = this.Query(tx).
+		State(ADPackageStateEnabled).
+		Attr("isOn", true).
+		Where("networkId IN (SELECT id FROM "+SharedADNetworkDAO.Table+" WHERE state=1 AND isOn=1)").
+		Where("id IN (SELECT DISTINCT packageId FROM "+SharedADPackageInstanceDAO.Table+" WHERE state=1 AND isOn=1 AND (userId=0 OR userDayTo IS NULL OR userDayTo>:endDay))"). // 确保没有正在使用
+		Param("endDay", timeutil.Format("Ymd")).
+		Where("id IN (SELECT DISTINCT packageId FROM " + SharedADPackagePriceDAO.Table + " WHERE price>0 )"). // 已设定价格
+		Asc("protectionBandwidthSize").
+		Asc("serverBandwidthSize").
+		AscPk().
+		Slice(&result).
+		FindAll()
+
+	return
+}
+
+// FindIdlePackage 查找符合条件的高防产品
+func (this *ADPackageDAO) FindIdlePackage(tx *dbs.Tx, networkId int64, protectionBandwidthSize int64, protectionBandwidthUnit string, serverBandwidthSize int64, serverBandwidthUnit string) (p *ADPackage, err error) {
+	one, err := this.Query(tx).
+		State(ADPackageStateEnabled).
+		Attr("isOn", true).
+		Attr("networkId", networkId).
+		Attr("protectionBandwidthSize", protectionBandwidthSize).
+		Attr("protectionBandwidthUnit", protectionBandwidthUnit).
+		Attr("serverBandwidthSize", serverBandwidthSize).
+		Attr("serverBandwidthUnit", serverBandwidthUnit).
+		Where("id IN (SELECT DISTINCT packageId FROM "+SharedADPackageInstanceDAO.Table+" WHERE state=1 AND isOn=1 AND (userId=0 OR userDayTo IS NULL OR userDayTo<:endDay))"). // 确保没有正在使用
+		Param("endDay", timeutil.Format("Ymd")).
+		Where("id IN (SELECT DISTINCT packageId FROM " + SharedADPackagePriceDAO.Table + " WHERE price>0 )"). // 已设定价格
+		Find()
+	if err != nil || one == nil {
+		return nil, err
+	}
+	return one.(*ADPackage), nil
+}