实现余额购买接口 & 实现全局 id 生成器

web/handlers/resource.go

@@ -0,0 +1,140 @@
+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

web/handlers/trade.go

@@ -0,0 +1,61 @@
+package handlers
+
+import (
+	"platform/web/auth"
+	m "platform/web/models"
+	q "platform/web/queries"
+	"platform/web/services"
+	"strconv"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+// region CreateTrade
+
+type CreateTradeReq struct {
+	Subject string `json:"subject" validate:"required"`
+	Remark  string `json:"remark"`
+	Amount  int    `json:"amount" validate:"required"`
+	Method  int    `json:"method" validate:"required"` // 支付方式：1.支付宝，2.微信
+}
+
+func CreateTrade(c *fiber.Ctx) error {
+	// 检查权限
+	authContext, err := auth.Protect(c, []services.PayloadType{services.PayloadUser}, []string{})
+	if err != nil {
+		return err
+	}
+
+	// 解析请求参数
+	req := new(CreateTradeReq)
+	if err := c.BodyParser(req); err != nil {
+		return err
+	}
+
+	// 创建交易订单
+	num, err := services.ID.GenSerial(c.Context())
+	if err != nil {
+		return err
+	}
+	var trade = m.Trade{
+		UserID:  authContext.Payload.Id,
+		InnerNo: strconv.FormatUint(num, 10),
+		Subject: req.Subject,
+		Remark:  req.Remark,
+		Amount:  float64(req.Amount) / 100,
+		Method:  int32(req.Method),
+	}
+
+	// 调用外部接口
+
+	// 保存交易订单
+	err = q.Trade.Create(&trade)
+	if err != nil {
+		return err
+	}
+
+	// 返回结果，外部支付链接
+	return nil
+}
+
+// endregion