对公添加余额

manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/dto/PayOrderDto.java

@@ -25,6 +25,8 @@ import lombok.Setter;
 
 import javax.persistence.GeneratedValue;
 import javax.persistence.GenerationType;
+import javax.validation.constraints.DecimalMin;
+import javax.validation.constraints.Min;
 import javax.validation.constraints.NotNull;
 import java.io.Serializable;
 import java.util.Date;
@@ -68,6 +70,7 @@ public class PayOrderDto extends BaseDTO implements Serializable {
     private Date payTime;
 
     @NotNull
+    @DecimalMin(value = "0.01",message = "最小金额0.01")
     @ApiModelProperty(value = "金额")
     private Double amount;
 

manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/PayOrderServiceImpl.java

@@ -28,6 +28,7 @@ import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
+import org.springframework.security.core.parameters.P;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -102,12 +103,29 @@ public class PayOrderServiceImpl implements PayOrderService {
     public void create(PayOrderDto resources) {
         checkPer();
         PayOrder payOrder = new PayOrder();
-        BeanUtil.copyProperties(resources, payOrder);
-        payOrder.setPayTime(DateUtil.date());
-        payOrder.setUserId(resources.getUserId());
-        payOrder.setStatus(DefaultNumberConstants.ONE_NUMBER);
-        payOrder.setPayType(DefaultNumberConstants.TWO_NUMBER);
-        payOrderRepository.save(payOrder);
+        Long userId = resources.getUserId();
+        if (userId != null) {
+            BeanUtil.copyProperties(resources, payOrder);
+            UserDto userServiceById = userService.findById(resources.getUserId());
+            if (userServiceById != null && userServiceById.getCompanyId() != null) {
+                Long companyId = userServiceById.getCompanyId();
+                CompanyDto companyById = companyService.findCompanyById(companyId);
+                // 充值余额
+                Double balance =
+                        companyById.getUserBalance() != null ?
+                                companyById.getUserBalance() :
+                                    DefaultNumberConstants.ZERO_NUMBER;
+                companyService.updateUserBalanceByCompanyId
+                        (NumberUtil.add(balance, resources.getAmount()), companyId);
+                payOrder.setCompanyId(companyId);
+            }
+            payOrder.setUserId(userId);
+            payOrder.setPayTime(DateUtil.date());
+            payOrder.setUserId(resources.getUserId());
+            payOrder.setStatus(DefaultNumberConstants.ONE_NUMBER);
+            payOrder.setPayType(DefaultNumberConstants.TWO_NUMBER);
+            payOrderRepository.save(payOrder);
+        }
     }
 
     /**