From 77bb52fe882df5aeb793462344a155d35d026da8 Mon Sep 17 00:00:00 2001
From: lwl0608 <lwl0608@foxmail.com>
Date: Tue, 14 May 2024 09:52:23 +0800
Subject: [PATCH] create_order

---
 app/mix/dao/mysql.go         | 26 ++++++++++++++++++++++++++
 app/mix/service/logic/vas.go | 16 ++++++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/app/mix/dao/mysql.go b/app/mix/dao/mysql.go
index a2497d7f..903df2f0 100644
--- a/app/mix/dao/mysql.go
+++ b/app/mix/dao/mysql.go
@@ -270,6 +270,32 @@ func (m *Mysql) GetOrders(ctx *gin.Context, tx *sqlx.Tx, mid, st, et int64, offs
 	return
 }
 
+// 获取订单数
+func (m *Mysql) GetOrderCnt(ctx *gin.Context, tx *sqlx.Tx, mid, st, et int64, orderStatus int32) (int, error) {
+	type S struct {
+		Cnt int `json:"cnt"`
+	}
+	var (
+		err error
+		s   = S{}
+	)
+	sqlStr := fmt.Sprintf("select count(1) as cnt from %s where mid=? and order_status=? and ct>=? and ct<?", TableOrder)
+	args := []any{mid, orderStatus, st, et}
+	if tx != nil {
+		err = tx.GetContext(ctx, &s, sqlStr, args...)
+	} else {
+		db := m.getDBVas()
+		err = db.SelectContext(ctx, &s, sqlStr, args...)
+	}
+	if err == sql.ErrNoRows {
+		return 0, nil
+	}
+	if err != nil {
+		return 0, err
+	}
+	return s.Cnt, nil
+}
+
 // 更新订单状态
 func (m *Mysql) UpdateOrderStatus(ctx *gin.Context, tx *sqlx.Tx, orderId string, preStatus, aftStatus int32) error {
 	var err error
diff --git a/app/mix/service/logic/vas.go b/app/mix/service/logic/vas.go
index b161e623..0a1bc919 100644
--- a/app/mix/service/logic/vas.go
+++ b/app/mix/service/logic/vas.go
@@ -107,6 +107,22 @@ func (v *Vas) GetMembershipProductList(ctx *gin.Context, req *vasproto.GetMember
 }
 
 func (v *Vas) CreateOrder(ctx *gin.Context, req *vasproto.CreateOrderReq) (data *vasproto.CreateOrderData, err error) {
+	// 检查退订单次数
+	var (
+		et = time.Now().Unix()
+		st = et - 86400*30
+	)
+	cnt, err := v.store.GetOrderCnt(ctx, nil, req.Mid, st, et, dbstruct.VasOrderStatusRefund)
+	if err != nil {
+		logger.Error("GetOrderCnt fail, mid: %v, err: %v", req.Mid, err)
+		return
+	}
+	logger.Info("GetOrderCnt Refund, mid: %v, cnt: %v", req.Mid, cnt)
+	if cnt >= 8 {
+		err = fmt.Errorf("账号已受限，解限请联系客服")
+		return
+	}
+
 	var (
 		productId = req.ProductId
 		order     *dbstruct.Order