充值赠送

src/main/java/com/edgec/browserbackend/account/controller/PayBackController.java

@@ -3,9 +3,7 @@ package com.edgec.browserbackend.account.controller;
 import com.edgec.browserbackend.account.service.PayBackService;
 import com.edgec.browserbackend.browser.domain.PayBack;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -26,4 +24,10 @@ public class PayBackController {
     public List<PayBack> getAll() {
         return payBackService.findAll();
     }
+
+    @PutMapping
+    public List<PayBack> putBack(@RequestBody PayBack payBack) {
+        return payBackService.updatePayBack(payBack);
+    }
+
 }

src/main/java/com/edgec/browserbackend/account/service/PayBackService.java

@@ -12,4 +12,6 @@ import java.util.List;
 public interface PayBackService {
 
     List<PayBack> findAll();
+
+    List<PayBack> updatePayBack(PayBack payBack);
 }

src/main/java/com/edgec/browserbackend/account/service/impl/PayBackServiceImpl.java

@@ -5,6 +5,7 @@ import com.edgec.browserbackend.browser.domain.PayBack;
 import com.edgec.browserbackend.browser.repository.PayBackRepository;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.web.bind.annotation.RequestBody;
 
 import java.util.List;
 
@@ -23,4 +24,14 @@ public class PayBackServiceImpl implements PayBackService {
     public List<PayBack> findAll() {
         return payBackRepository.findAll();
     }
+
+    @Override
+    public List<PayBack> updatePayBack(PayBack payBack) {
+        List<PayBack> list = payBackRepository.findByPay(payBack.getPay());
+        if(list.size() > 0 && payBack.getPay() > 0){
+            list.forEach(payBackRepository::delete);
+            payBackRepository.insert(payBack);
+        }
+        return payBackRepository.findAll();
+    }
 }

src/main/java/com/edgec/browserbackend/account/service/impl/PaymentServiceImpl.java

@@ -590,7 +590,7 @@ public class PaymentServiceImpl implements PaymentService {
                             balance.setUsername(byTradeNo.getUsername());
                         }
                         // 充值返现
-                        PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount());
+                        PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount()).get(0);
                         if (payBack == null) {
                             payBack = new PayBack();
                         }
@@ -658,7 +658,7 @@ public class PaymentServiceImpl implements PaymentService {
                         balance.setUsername(byTradeNo.getUsername());
                     }
                     // 充值返现
-                    PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount());
+                    PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount()).get(0);
                     if (payBack == null)
                         payBack = new PayBack();
                     balance.setBalanced(balance.getBalanced() + byTradeNo.getAmount() + payBack.getBack());
@@ -802,7 +802,7 @@ public class PaymentServiceImpl implements PaymentService {
         if (b) {
             log.info("订单 " + byTradeNo.getTradeNo() + " 的支付状态更新为成功");
             // 2.2 获取充值优惠赠送
-            PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount());
+            PayBack payBack = payBackRepository.findByPay(byTradeNo.getAmount()).get(0);
             if (payBack == null) {
                 payBack = new PayBack();
             }

src/main/java/com/edgec/browserbackend/browser/domain/PayBack.java

 package com.edgec.browserbackend.browser.domain;
 
 import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import org.springframework.data.annotation.Id;
 import org.springframework.data.mongodb.core.mapping.Document;
 
 @Document(collection = "paybackprice")
 public class PayBack {
+
+    @Id
+    private String id;
     private int pay;
     private int back = 0;
 
+    public String getId() {
+        return id;
+    }
+
+    public void setId(String id) {
+        this.id = id;
+    }
+
     public int getBack() {
         return back;
     }

src/main/java/com/edgec/browserbackend/browser/repository/PayBackRepository.java

@@ -3,6 +3,9 @@ package com.edgec.browserbackend.browser.repository;
 import com.edgec.browserbackend.browser.domain.PayBack;
 import org.springframework.data.mongodb.repository.MongoRepository;
 
+import java.util.List;
+
 public interface PayBackRepository extends MongoRepository<PayBack, String> {
-    PayBack findByPay(int pay);
+
+    List<PayBack> findByPay(int pay);
 }