新增功能 -- 产品订阅 年度/月度

src/main/java/com/ai/da/controller/StripeController.java

@@ -1,17 +1,21 @@
 package com.ai.da.controller;
 
 import com.ai.da.common.response.Response;
+import com.ai.da.model.dto.ProductPurchaseDTO;
 import com.ai.da.service.StripeService;
 import com.paypal.http.HttpResponse;
 import com.paypal.payments.Refund;
+import com.stripe.exception.StripeException;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
+import org.simpleframework.xml.core.Validate;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 
 @Api(tags = "Stripe模块")
@@ -24,19 +28,19 @@ public class StripeController {
     private StripeService stripeService;
 
     @ApiOperation("创建支付链接")
-    @PostMapping("/createOrder/{amount}")
-    public Response<String> pay(@PathVariable Integer amount, @RequestParam String returnUrl) {
-        return Response.success(stripeService.pay(amount, returnUrl));
+    @PostMapping("/createOrder")
+    public Response<String> pay(@Validate @RequestBody ProductPurchaseDTO productPurchaseDTO) {
+        return Response.success(stripeService.pay(productPurchaseDTO));
     }
 
     @ApiOperation("支付通知")
     @PostMapping("/trade/notify")
-    public Response<String> callback(HttpServletRequest request) throws ServletException, IOException {
+    public void callback(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
         Boolean result =  stripeService.notify(request);
         if (result){
-            return Response.success(200,"success");
+            response.setStatus(HttpServletResponse.SC_OK);
         }else {
-            return Response.fail(400,"failure");
+            response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
         }
     }
 
@@ -51,4 +55,20 @@ public class StripeController {
         }
     }
 
+    @ApiOperation("获取订阅")
+    @PostMapping("/getSubscription")
+    public void getSubscription() {
+        try {
+            stripeService.getSubscription("xp", "xupei3360@163.com");
+        } catch (StripeException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    @ApiOperation("取消订阅")
+    @PostMapping("/cancelSubscription")
+    public Response<String> cancelSubscription(@RequestParam String subscriptionId) {
+        stripeService.cancelSubscription(subscriptionId);
+        return Response.success("success");
+    }
 }