141 lines
3.0 KiB
Go
141 lines
3.0 KiB
Go
package handlers
|
|
|
|
import (
|
|
"errors"
|
|
"platform/web/auth"
|
|
m "platform/web/models"
|
|
q "platform/web/queries"
|
|
"platform/web/services"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
// region CreateResourceByBalance
|
|
|
|
type CreateResourceByBalanceReq struct {
|
|
Type int32 `json:"type" validate:"required"`
|
|
Live int32 `json:"live" validate:"required"`
|
|
Expire int32 `json:"expire" validate:"required"`
|
|
Quota int32 `json:"quota" validate:"required"`
|
|
DailyLimit int32 `json:"daily_limit" validate:"required"`
|
|
}
|
|
|
|
// CreateResourceByBalance 通过余额创建资源
|
|
func CreateResourceByBalance(c *fiber.Ctx) error {
|
|
|
|
// 检查权限
|
|
authContext, err := auth.Protect(c, []services.PayloadType{services.PayloadUser}, []string{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 解析请求参数
|
|
req := new(CreateResourceByBalanceReq)
|
|
if err := c.BodyParser(req); err != nil {
|
|
return err
|
|
}
|
|
|
|
err = q.Q.Transaction(func(q *q.Query) error {
|
|
// 检查用户
|
|
user, err := q.User.Where(q.User.ID.Eq(authContext.Payload.Id)).Take()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 计算价格
|
|
var amount = 0
|
|
var payment = 0
|
|
|
|
// 检查余额
|
|
if user.Balance < float64(req.Quota)/100 {
|
|
return errors.New("余额不足")
|
|
}
|
|
|
|
// 创建资源
|
|
resource := m.Resource{
|
|
UserID: authContext.Payload.Id,
|
|
}
|
|
err = q.Resource.Save(&resource)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resourcePss := m.ResourcePss{
|
|
ResourceID: resource.ID,
|
|
Type: req.Type,
|
|
Live: req.Live,
|
|
Quota: req.Quota,
|
|
Expire: time.Now().Add(time.Duration(req.Expire) * time.Second),
|
|
DailyLimit: req.DailyLimit,
|
|
}
|
|
err = q.ResourcePss.Save(&resourcePss)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 更新用户余额
|
|
user.Balance -= float64(payment)
|
|
_, err = q.User.
|
|
Where(q.User.ID.Eq(authContext.Payload.Id)).
|
|
Update(q.User.Balance, user.Balance)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 生成账单
|
|
bill := m.Bill{
|
|
UserID: authContext.Payload.Id,
|
|
ResourceID: resource.ID,
|
|
BillNo: services.ID.GenReadable("bil"),
|
|
Type: 1,
|
|
Info: "购买套餐",
|
|
Amount: float64(amount),
|
|
Payment: float64(payment),
|
|
}
|
|
err = q.Bill.Save(&bill)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region CreateResourceByAlipayCallback
|
|
|
|
type CreateResourceByAlipayCallbackReq struct {
|
|
}
|
|
|
|
// CreateResourceByAlipayCallback 支付宝支付回调
|
|
func CreateResourceByAlipayCallback(c *fiber.Ctx) error {
|
|
|
|
// 根据支付类型执行不同流程：
|
|
// 1. 支付宝或微信（即时支付）
|
|
// - 更新订单状态
|
|
// - 生成账单
|
|
// - 生成资源
|
|
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
// endregion
|
|
|
|
// region CreateResourceByWechatCallback
|
|
|
|
type CreateResourceByWechatCallbackReq struct {
|
|
}
|
|
|
|
// CreateResourceByWechatCallback 微信支付回调
|
|
func CreateResourceByWechatCallback(c *fiber.Ctx) error {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
// endregion
|