From 404ddc9cf8415bc89ceaedb2e5b2c81a7f50bbb2 Mon Sep 17 00:00:00 2001 From: lzh <1625167628@qq.com> Date: Tue, 21 Oct 2025 17:08:03 +0800 Subject: [PATCH] =?UTF-8?q?=E6=89=A9=E5=B1=95=E6=97=B6=E9=97=B4=E4=BF=AE?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/logic/bundleExtendLogic.go | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/internal/logic/bundleExtendLogic.go b/internal/logic/bundleExtendLogic.go index 5d6fd76..de0be2b 100644 --- a/internal/logic/bundleExtendLogic.go +++ b/internal/logic/bundleExtendLogic.go @@ -19,6 +19,17 @@ func BundleExtend(req *bundle.BundleExtendRequest) (*bundle.BundleExtendResponse if err := copier.CopyWithOption(&data, req, copier.Option{DeepCopy: true}); err != nil { return nil, err } + var durationNumber int + + switch req.TimeUnit { + case 1: + durationNumber = int(req.AvailableDurationAdditional) * 1 + case 2: + durationNumber = int(req.AvailableDurationAdditional) * 30 + case 3: + durationNumber = int(req.AvailableDurationAdditional) * 365 + } + if req.Type == 1 { if err := dao.ExtendBundleBalanceByUserId(model.BundleBalanceExtendPo{ UserId: int(req.UserId), @@ -26,7 +37,7 @@ func BundleExtend(req *bundle.BundleExtendRequest) (*bundle.BundleExtendResponse ImageNumber: int(req.ImagesAdditional), DataAnalysisNumber: int(req.DataAdditional), AccountNumber: int(req.AccountAdditional), - DurationNumber: int(req.AvailableDurationAdditional), + DurationNumber: durationNumber, }); err != nil { return nil, errors.New("用户没有余量信息") }