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 {