450 lines
12 KiB
Go
450 lines
12 KiB
Go
package handlers
|
|
|
|
import (
|
|
"gorm.io/gen/field"
|
|
"platform/pkg/u"
|
|
"platform/web/auth"
|
|
"platform/web/core"
|
|
resource2 "platform/web/domains/resource"
|
|
g "platform/web/globals"
|
|
"platform/web/globals/orm"
|
|
q "platform/web/queries"
|
|
s "platform/web/services"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
type ListResourceShortReq struct {
|
|
core.PageReq
|
|
ResourceNo *string `json:"resource_no"`
|
|
Active *bool `json:"active"`
|
|
Type *int `json:"type"`
|
|
CreateAfter *time.Time `json:"create_after"`
|
|
CreateBefore *time.Time `json:"create_before"`
|
|
ExpireAfter *time.Time `json:"expire_after"`
|
|
ExpireBefore *time.Time `json:"expire_before"`
|
|
}
|
|
|
|
func ListResourceShort(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []auth.PayloadType{auth.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(ListResourceShortReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 查询套餐列表
|
|
do := q.Resource.Where(
|
|
q.Resource.UserID.Eq(authContext.Payload.Id),
|
|
q.Resource.Type.Eq(int32(resource2.TypeShort)),
|
|
)
|
|
if req.ResourceNo != nil && *req.ResourceNo != "" {
|
|
do.Where(q.Resource.ResourceNo.Eq(*req.ResourceNo))
|
|
}
|
|
if req.Active != nil {
|
|
do.Where(q.Resource.Active.Is(*req.Active))
|
|
}
|
|
if req.Type != nil {
|
|
do.Where(q.ResourceShort.As(q.Resource.Short.Name()).Type.Eq(int32(*req.Type)))
|
|
}
|
|
if req.CreateAfter != nil {
|
|
do.Where(q.Resource.CreatedAt.Gte(orm.LocalDateTime(*req.CreateAfter)))
|
|
}
|
|
if req.CreateBefore != nil {
|
|
do.Where(q.Resource.CreatedAt.Lte(orm.LocalDateTime(*req.CreateBefore)))
|
|
}
|
|
if req.ExpireAfter != nil {
|
|
do.Where(q.ResourceShort.As(q.Resource.Short.Name()).Expire.Gte(orm.LocalDateTime(*req.ExpireAfter)))
|
|
}
|
|
if req.ExpireBefore != nil {
|
|
do.Where(q.ResourceShort.As(q.Resource.Short.Name()).Expire.Lte(orm.LocalDateTime(*req.ExpireBefore)))
|
|
}
|
|
|
|
resource, err := q.Resource.Debug().Where(do).
|
|
Joins(q.Resource.Short).
|
|
Order(q.Resource.CreatedAt.Desc()).
|
|
Offset(req.GetOffset()).
|
|
Limit(req.GetLimit()).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var total int64
|
|
if len(resource) < req.GetLimit() {
|
|
total = int64(len(resource) + req.GetOffset())
|
|
} else {
|
|
total, err = q.Resource.
|
|
Where(do).
|
|
Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return c.JSON(core.PageResp{
|
|
Total: int(total),
|
|
Page: req.GetPage(),
|
|
Size: req.GetSize(),
|
|
List: resource,
|
|
})
|
|
}
|
|
|
|
type ListResourceLongReq struct {
|
|
core.PageReq
|
|
ResourceNo *string `json:"resource_no"`
|
|
Active *bool `json:"active"`
|
|
Type *int `json:"type"`
|
|
CreateAfter *time.Time `json:"create_after"`
|
|
CreateBefore *time.Time `json:"create_before"`
|
|
ExpireAfter *time.Time `json:"expire_after"`
|
|
ExpireBefore *time.Time `json:"expire_before"`
|
|
}
|
|
|
|
func ListResourceLong(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []auth.PayloadType{auth.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(ListResourceLongReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 查询套餐列表
|
|
do := q.Resource.Where(
|
|
q.Resource.UserID.Eq(authContext.Payload.Id),
|
|
q.Resource.Type.Eq(int32(resource2.TypeLong)),
|
|
)
|
|
if req.ResourceNo != nil && *req.ResourceNo != "" {
|
|
do.Where(q.Resource.ResourceNo.Eq(*req.ResourceNo))
|
|
}
|
|
if req.Active != nil {
|
|
do.Where(q.Resource.Active.Is(*req.Active))
|
|
}
|
|
if req.Type != nil {
|
|
do.Where(q.ResourceLong.As(q.Resource.Long.Name()).Type.Eq(int32(*req.Type)))
|
|
}
|
|
if req.CreateAfter != nil {
|
|
do.Where(q.Resource.CreatedAt.Gte(orm.LocalDateTime(*req.CreateAfter)))
|
|
}
|
|
if req.CreateBefore != nil {
|
|
do.Where(q.Resource.CreatedAt.Lte(orm.LocalDateTime(*req.CreateBefore)))
|
|
}
|
|
if req.ExpireAfter != nil {
|
|
do.Where(q.ResourceLong.As(q.Resource.Long.Name()).Expire.Gte(orm.LocalDateTime(*req.ExpireAfter)))
|
|
}
|
|
if req.ExpireBefore != nil {
|
|
do.Where(q.ResourceLong.As(q.Resource.Long.Name()).Expire.Lte(orm.LocalDateTime(*req.ExpireBefore)))
|
|
}
|
|
|
|
resource, err := q.Resource.Debug().Where(do).
|
|
Joins(q.Resource.Long).
|
|
Order(q.Resource.CreatedAt.Desc()).
|
|
Offset(req.GetOffset()).
|
|
Limit(req.GetLimit()).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var total int64
|
|
if len(resource) < req.GetLimit() {
|
|
total = int64(len(resource) + req.GetOffset())
|
|
} else {
|
|
total, err = q.Resource.
|
|
Where(do).
|
|
Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return c.JSON(core.PageResp{
|
|
Total: int(total),
|
|
Page: req.GetPage(),
|
|
Size: req.GetSize(),
|
|
List: resource,
|
|
})
|
|
}
|
|
|
|
type AllResourceReq struct {
|
|
}
|
|
|
|
func AllActiveResource(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
authCtx, err := auth.NewProtect(c).Payload(auth.PayloadUser).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 查询套餐列表
|
|
var now = time.Now()
|
|
|
|
var short = q.ResourceShort.As(q.Resource.Short.Name())
|
|
var long = q.ResourceLong.As(q.Resource.Long.Name())
|
|
resources, err := q.Resource.
|
|
Joins(
|
|
q.Resource.Short,
|
|
q.Resource.Long,
|
|
).
|
|
Where(
|
|
q.Resource.UserID.Eq(authCtx.Payload.Id),
|
|
q.Resource.Active.Is(true),
|
|
q.Resource.Where(
|
|
q.Resource.Type.Eq(int32(resource2.TypeShort)),
|
|
q.ResourceShort.As(q.Resource.Short.Name()).Where(
|
|
short.Type.Eq(int32(resource2.ModeTime)),
|
|
short.Expire.Gte(orm.LocalDateTime(now)),
|
|
q.ResourceShort.As(q.Resource.Short.Name()).
|
|
Where(short.DailyLast.Lt(orm.LocalDateTime(u.Today()))).
|
|
Or(short.DailyLimit.GtCol(short.DailyUsed)),
|
|
).Or(
|
|
short.Type.Eq(int32(resource2.ModeCount)),
|
|
short.Quota.GtCol(short.Used),
|
|
),
|
|
).Or(
|
|
q.Resource.Type.Eq(int32(resource2.TypeLong)),
|
|
q.ResourceLong.As(q.Resource.Long.Name()).Where(
|
|
long.Type.Eq(int32(resource2.ModeTime)),
|
|
long.Expire.Gte(orm.LocalDateTime(now)),
|
|
q.ResourceLong.As(q.Resource.Long.Name()).
|
|
Where(long.DailyLast.Lt(orm.LocalDateTime(u.Today()))).
|
|
Or(long.DailyLimit.GtCol(long.DailyUsed)),
|
|
).Or(
|
|
long.Type.Eq(int32(resource2.ModeCount)),
|
|
long.Quota.GtCol(long.Used),
|
|
),
|
|
),
|
|
).
|
|
Order(q.Resource.CreatedAt.Desc()).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.JSON(resources)
|
|
}
|
|
|
|
type StatisticPersonalResp struct {
|
|
Short StatisticShort `json:"short"`
|
|
Long StatisticLong `json:"long"`
|
|
}
|
|
|
|
type StatisticShort struct {
|
|
ResourceCount int `json:"resource_count"`
|
|
ResourceQuotaSum int `json:"resource_quota_sum"`
|
|
ResourceDailyFreeSum int `json:"resource_daily_free_sum"`
|
|
}
|
|
|
|
type StatisticLong struct {
|
|
ResourceCount int `json:"resource_count"`
|
|
ResourceQuotaSum int `json:"resource_quota_sum"`
|
|
ResourceDailyFreeSum int `json:"resource_daily_free_sum"`
|
|
}
|
|
|
|
func StatisticResourceFree(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
session, err := auth.NewProtect(c).Payload(auth.PayloadUser).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 统计套餐剩余数量
|
|
resources, err := q.Resource.Debug().
|
|
Preload(
|
|
q.Resource.Short,
|
|
q.Resource.Long,
|
|
).
|
|
Where(
|
|
q.Resource.UserID.Eq(session.Payload.Id),
|
|
q.Resource.Active.Is(true),
|
|
).
|
|
Select(q.Resource.ID, q.Resource.Type).
|
|
Find()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var shortCount, shortQuotaSum, shortDailyFreeSum int
|
|
var longCount, longQuotaSum, longDailyFreeSum int
|
|
for _, resource := range resources {
|
|
switch {
|
|
|
|
// 短效包量
|
|
case resource2.Type(resource.Type) == resource2.TypeShort && resource2.Mode(resource.Short.Type) == resource2.ModeCount:
|
|
if u.Z(resource.Short.Quota) > resource.Short.Used {
|
|
shortCount++
|
|
shortQuotaSum += int(u.Z(resource.Short.Quota) - resource.Short.Used)
|
|
}
|
|
|
|
// 长效包量
|
|
case resource2.Type(resource.Type) == resource2.TypeLong && resource2.Mode(resource.Long.Type) == resource2.ModeCount:
|
|
if u.Z(resource.Long.Quota) > resource.Long.Used {
|
|
longCount++
|
|
longQuotaSum += int(u.Z(resource.Long.Quota) - resource.Long.Used)
|
|
}
|
|
|
|
// 短效包时
|
|
case resource2.Type(resource.Type) == resource2.TypeShort && resource2.Mode(resource.Short.Type) == resource2.ModeTime:
|
|
if time.Time(*resource.Short.Expire).After(time.Now()) {
|
|
if resource.Short.DailyLast == nil || u.SameDate(time.Time(*resource.Short.DailyLast)) == false {
|
|
shortCount++
|
|
shortDailyFreeSum += int(resource.Short.DailyLimit)
|
|
} else if resource.Short.DailyLimit > resource.Short.DailyUsed {
|
|
shortCount++
|
|
shortDailyFreeSum += int(resource.Short.DailyLimit - resource.Short.DailyUsed)
|
|
}
|
|
}
|
|
|
|
// 长效包时
|
|
case resource2.Type(resource.Type) == resource2.TypeLong && resource2.Mode(resource.Long.Type) == resource2.ModeTime:
|
|
if time.Time(*resource.Long.Expire).After(time.Now()) {
|
|
if resource.Long.DailyLast == nil || u.SameDate(time.Time(*resource.Long.DailyLast)) == false {
|
|
longCount++
|
|
longDailyFreeSum += int(resource.Long.DailyLimit)
|
|
} else if resource.Long.DailyLimit > resource.Long.DailyUsed {
|
|
longCount++
|
|
longDailyFreeSum += int(resource.Long.DailyLimit - resource.Long.DailyUsed)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return c.JSON(StatisticPersonalResp{
|
|
Short: StatisticShort{
|
|
ResourceCount: shortCount,
|
|
ResourceQuotaSum: shortQuotaSum,
|
|
ResourceDailyFreeSum: shortDailyFreeSum,
|
|
},
|
|
Long: StatisticLong{
|
|
ResourceCount: longCount,
|
|
ResourceQuotaSum: longQuotaSum,
|
|
ResourceDailyFreeSum: longDailyFreeSum,
|
|
},
|
|
})
|
|
}
|
|
|
|
type StatisticResourceUsageReq struct {
|
|
ResourceNo *string `json:"resource_no"`
|
|
TimeAfter *time.Time `json:"time_after"`
|
|
TimeBefore *time.Time `json:"time_before"`
|
|
}
|
|
|
|
type StatisticResourceUsageResp []struct {
|
|
Date time.Time `json:"date"`
|
|
Count int `json:"count"`
|
|
}
|
|
|
|
func StatisticResourceUsage(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
session, err := auth.NewProtect(c).Payload(auth.PayloadUser).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
var req = new(StatisticResourceUsageReq)
|
|
if err := g.Validator.Validate(c, req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 统计套餐提取数量
|
|
do := q.LogsUserUsage.Where(q.LogsUserUsage.UserID.Eq(session.Payload.Id))
|
|
if req.ResourceNo != nil && *req.ResourceNo != "" {
|
|
var resourceID int32
|
|
err := q.Resource.
|
|
Where(
|
|
q.Resource.UserID.Eq(session.Payload.Id),
|
|
q.Resource.ResourceNo.Eq(*req.ResourceNo),
|
|
).
|
|
Select(q.Resource.ID).
|
|
Scan(&resourceID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
do.Where(q.LogsUserUsage.ResourceID.Eq(resourceID))
|
|
}
|
|
if req.TimeAfter != nil {
|
|
do.Where(q.LogsUserUsage.Time.Gte(orm.LocalDateTime(*req.TimeAfter)))
|
|
}
|
|
if req.TimeBefore != nil {
|
|
do.Where(q.LogsUserUsage.Time.Lte(orm.LocalDateTime(*req.TimeBefore)))
|
|
}
|
|
|
|
var data = new(StatisticResourceUsageResp)
|
|
err = q.LogsUserUsage.Debug().
|
|
Select(
|
|
q.LogsUserUsage.Count_.Sum().As("count"),
|
|
field.NewUnsafeFieldRaw("date_trunc('day', time)").As("date"),
|
|
).
|
|
Where(do).
|
|
Group(
|
|
field.NewUnsafeFieldRaw("date_trunc('day', time)"),
|
|
).
|
|
Order(
|
|
field.NewField("", "date").Desc(),
|
|
).
|
|
Scan(&data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.JSON(data)
|
|
}
|
|
|
|
type CreateResourceReq struct {
|
|
*s.CreateResourceData
|
|
}
|
|
|
|
func CreateResource(c *fiber.Ctx) error {
|
|
|
|
// 检查权限
|
|
authCtx, err := auth.NewProtect(c).Payload(auth.PayloadUser).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
var req = new(CreateResourceReq)
|
|
if err := g.Validator.Validate(c, req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 创建套餐
|
|
err = s.Resource.CreateResourceByBalance(authCtx.Payload.Id, time.Now(), req.CreateResourceData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func ResourcePrice(c *fiber.Ctx) error {
|
|
// 检查权限
|
|
_, err := auth.NewProtect(c).Payload(auth.PayloadInternalServer).Do()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
var req = new(CreateResourceReq)
|
|
if err := g.Validator.Validate(c, req); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 获取套餐价格
|
|
return c.JSON(fiber.Map{
|
|
"price": req.GetAmount().StringFixed(2),
|
|
})
|
|
}
|