From 17d458109258ccae524dda8d20ba1d2589557fe9 Mon Sep 17 00:00:00 2001
From: lzh <1625167628@qq.com>
Date: Tue, 21 Oct 2025 11:33:31 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=89=A9=E5=B1=95=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/logic/bundleExtendLogic.go | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/internal/logic/bundleExtendLogic.go b/internal/logic/bundleExtendLogic.go
index 919d023..259cbf4 100644
--- a/internal/logic/bundleExtendLogic.go
+++ b/internal/logic/bundleExtendLogic.go
@@ -19,15 +19,17 @@ func BundleExtend(req *bundle.BundleExtendRequest) (*bundle.BundleExtendResponse
 	if err := copier.CopyWithOption(&data, req, copier.Option{DeepCopy: true}); err != nil {
 		return nil, err
 	}
-	if err := dao.ExtendBundleBalanceByUserId(model.BundleBalanceExtendPo{
-		UserId:             int(req.UserId),
-		VideoNumber:        int(req.VideoAdditional),
-		ImageNumber:        int(req.ImagesAdditional),
-		DataAnalysisNumber: int(req.DataAdditional),
-		AccountNumber:      int(req.AccountAdditional),
-		DurationNumber:     int(req.DataAdditional),
-	}); err != nil {
-		return nil, errors.New("用户没有余量信息")
+	if req.Type == 1 {
+		if err := dao.ExtendBundleBalanceByUserId(model.BundleBalanceExtendPo{
+			UserId:             int(req.UserId),
+			VideoNumber:        int(req.VideoAdditional),
+			ImageNumber:        int(req.ImagesAdditional),
+			DataAnalysisNumber: int(req.DataAdditional),
+			AccountNumber:      int(req.AccountAdditional),
+			DurationNumber:     int(req.DataAdditional),
+		}); err != nil {
+			return nil, errors.New("用户没有余量信息")
+		}
 	}
 	err := dao.AddBundleExtendRecord(data)
 	if err != nil {