From 53bf31865a850151eb6fddfc9a6d2c264ed9c7ea Mon Sep 17 00:00:00 2001
From: lzh <1625167628@qq.com>
Date: Wed, 15 Oct 2025 15:01:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8B=93=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/dao/bundleExtend.go | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/internal/dao/bundleExtend.go b/internal/dao/bundleExtend.go
index fae8539..521cb37 100644
--- a/internal/dao/bundleExtend.go
+++ b/internal/dao/bundleExtend.go
@@ -7,7 +7,6 @@ import (
 	"micro-bundle/pb/bundle"
 	"micro-bundle/pkg/app"
 	"micro-bundle/pkg/utils"
-	"strconv"
 	"time"
 
 	"gorm.io/gorm"
@@ -20,10 +19,10 @@ func AddBundleExtendRecord(data model.BundleExtensionRecords) error {
 		}
 		if data.AvailableDurationAdditional != 0 && data.TimeUnit != 0 {
 			record := model.BundleBalance{}
-			if err := tx.Model(&model.BundleBalance{}).Where(&model.BundleOrderRecords{CustomerID: strconv.Itoa(data.UserId)}).Order("created_at desc").First(&record).Error; err != nil {
+			if err := tx.Model(&model.BundleBalance{}).Where(&model.BundleBalance{UserId: data.UserId}).Order("month desc").First(&record).Error; err != nil {
 				return err
 			}
-			record.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.AvailableDurationAdditional))
+			record.ExpiredAt = record.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.AvailableDurationAdditional))
 			return tx.Save(record).Error
 		}
 		return nil