开通aiVip

src/main/java/com/peanut/modules/taihumed/controller/AiVipController.java

@@ -0,0 +1,140 @@
+package com.peanut.modules.taihumed.controller;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.IdWorker;
+import com.peanut.common.utils.R;
+import com.peanut.common.utils.ShiroUtils;
+import com.peanut.config.Constants;
+import com.peanut.config.DelayQueueConfig;
+import com.peanut.modules.book.service.TransactionDetailsService;
+import com.peanut.modules.common.entity.BuyOrder;
+import com.peanut.modules.common.entity.MyUserEntity;
+import com.peanut.modules.common.service.AiVipLogService;
+import com.peanut.modules.common.service.BuyOrderService;
+import com.peanut.modules.common.service.JfTransactionDetailsService;
+import com.peanut.modules.common.service.MyUserService;
+import com.peanut.modules.pay.weChatPay.dto.WechatPaymentInfo;
+import com.peanut.modules.pay.weChatPay.service.WxpayService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.amqp.core.MessagePostProcessor;
+import org.springframework.amqp.rabbit.core.RabbitTemplate;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.transaction.Transactional;
+import java.math.BigDecimal;
+
+@Slf4j
+@RestController("taihumedAiVip")
+@RequestMapping("taihumed/aiVip")
+public class AiVipController {
+
+    @Autowired
+    private BuyOrderService buyOrderService;
+    @Autowired
+    private MyUserService myUserService;
+    @Autowired
+    private TransactionDetailsService transactionDetailsService;
+    @Autowired
+    private JfTransactionDetailsService jfTransactionDetailsService;
+    @Autowired
+    private RabbitTemplate rabbitTemplate;
+    @Autowired
+    private WxpayService wxpayService;
+    @Autowired
+    private AiVipLogService aiVipLogService;
+
+    //aivip下单
+    @RequestMapping("/placeAiVipOrder")
+    @Transactional
+    public R placeVipOrder(@RequestBody BuyOrder buyOrder){
+        int uid = ShiroUtils.getUId();
+        buyOrder.setOrderStatus("0");
+        buyOrder.setOrderType("aiVip");
+        String timeId = IdWorker.getTimeId().substring(0, 32);
+        buyOrder.setOrderSn(timeId);
+        buyOrder.setUserId(uid);
+        buyOrderService.save(buyOrder);
+        BigDecimal totalPrice = buyOrder.getRealMoney();
+        if (Constants.PAYMENT_METHOD_VIRTUAL.equals(buyOrder.getPaymentMethod())) {
+            buyOrder.setOrderStatus(Constants.ORDER_STATUS_TO_BE_SHIPPED);
+            MyUserEntity user = myUserService.getById(buyOrder.getUserId());
+            if (usePeanutCoin(user, totalPrice)&&useJfCoin(user,buyOrder.getJfDeduction())) {
+                // 更新订单状态
+                buyOrderService.updateOrderStatus(user.getId(), buyOrder.getOrderSn(), "2");
+                //记录用户虚拟币消费
+                if(totalPrice.compareTo(BigDecimal.ZERO)>0){
+                    transactionDetailsService.recordTransaction(buyOrder, user, totalPrice);
+                }
+                //记录用户积分消费情况
+                if(buyOrder.getJfDeduction()!=null&&buyOrder.getJfDeduction().compareTo(BigDecimal.ZERO) > 0){
+                    jfTransactionDetailsService.recordJfTransaction(buyOrder, user, buyOrder.getJfDeduction());
+                }
+                //开通用户aiVip
+                aiVipLogService.openAiVip(buyOrder.getAiBuyConfigId());
+            } else {
+                rabbitTemplate.convertAndSend(
+                        DelayQueueConfig.ORDER_TO_BE_PAY_EXCHANGE,
+                        DelayQueueConfig.ORDER_TO_BE_PAY_ROUTING_KEY,
+                        buyOrder.getOrderId(),
+                        messagePostProcessor()
+                );
+                return R.error(500, "余额不足！");
+            }
+        }
+        //下单微信支付预付款订单
+        if(Constants.PAYMENT_METHOD_WECHAT_PAY.equals(buyOrder.getPaymentMethod())){
+            BuyOrder buyOrderEntity = buyOrderService.getBaseMapper().selectOne(new LambdaQueryWrapper<BuyOrder>().eq(BuyOrder::getOrderSn, timeId));
+            WechatPaymentInfo paymentInfo = new WechatPaymentInfo();
+            paymentInfo.setOrderSn(buyOrderEntity.getOrderSn());
+            paymentInfo.setBuyOrderId(buyOrderEntity.getOrderId());
+            paymentInfo.setTotalAmount(buyOrderEntity.getRealMoney());
+            paymentInfo.setAppName(buyOrder.getAppName());
+            wxpayService.prepay(paymentInfo);
+        }
+        rabbitTemplate.convertAndSend(
+                DelayQueueConfig.ORDER_TO_BE_PAY_EXCHANGE,
+                DelayQueueConfig.ORDER_TO_BE_PAY_ROUTING_KEY,
+                buyOrder.getOrderId(),
+                messagePostProcessor()
+        );
+        return R.ok().put("orderSn", timeId).put("money",buyOrder.getRealMoney());
+    }
+
+
+    private boolean usePeanutCoin(MyUserEntity user, BigDecimal totalPrice) {
+        if (user.getPeanutCoin().compareTo(totalPrice) >= 0) {
+            if(totalPrice.compareTo(BigDecimal.ZERO)==0){//纯积分支付，虚拟币不用支付
+                return true;
+            }
+            user.setPeanutCoin(user.getPeanutCoin().subtract(totalPrice));
+            myUserService.updateById(user);
+            return true;
+        }
+        return false;
+    }
+
+    private boolean useJfCoin(MyUserEntity user,BigDecimal jf){
+        if(jf==null){
+            return true;
+        }
+        if(user.getJf().compareTo(jf)>=0){
+            user.setJf(user.getJf().subtract(jf));
+            myUserService.updateById(user);
+            return true;
+        }else{
+            return false;
+        }
+    }
+
+    private MessagePostProcessor messagePostProcessor() {
+        return message -> {
+            //设置有效期30分钟
+            message.getMessageProperties().setExpiration(String.valueOf(30 * 60 * 1000));
+            return message;
+        };
+    }
+
+}