internal/logic/bundleExtendLogic.go

@@ -19,17 +19,6 @@ 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),
@@ -37,7 +26,7 @@ func BundleExtend(req *bundle.BundleExtendRequest) (*bundle.BundleExtendResponse
 			ImageNumber:        int(req.ImagesAdditional),
 			DataAnalysisNumber: int(req.DataAdditional),
 			AccountNumber:      int(req.AccountAdditional),
-			DurationNumber:     durationNumber,
+			DurationNumber:     int(req.AvailableDurationAdditional),
 		}); err != nil {
 			return nil, errors.New("用户没有余量信息")
 		}