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("用户没有余量信息")
 		}