RocketMQ消息实体改造

admin

2019-11-02 5e8907f1797a0f2028ab0731755941cf9ff7fc08

 fanli/src/main/java/com/yeshi/fanli/util/rocketmq/consumer/order/InviteOrderSubsidyMessageListener.java

@@ -1,11 +1,34 @@
package com.yeshi.fanli.util.rocketmq.consumer.order;



import java.util.Date;



import javax.annotation.Resource;



import org.springframework.stereotype.Component;



import com.aliyun.openservices.ons.api.Action;

import com.aliyun.openservices.ons.api.ConsumeContext;

import com.aliyun.openservices.ons.api.Message;

import com.aliyun.openservices.ons.api.MessageListener;

import com.aliyun.openservices.ons.api.Producer;

import com.aliyun.openservices.ons.api.SendResult;

import com.google.gson.Gson;

import com.yeshi.fanli.dto.order.OrderMQMsgDTO;

import com.yeshi.fanli.dto.order.OrderMoneyRecievedMQMsgDTO;

import com.yeshi.fanli.dto.user.coupon.UserSystemCouponUseMQMsgDTO;

import com.yeshi.fanli.entity.bus.user.UserSystemCouponRecord;

import com.yeshi.fanli.entity.mq.MQUnSendInfo;

import com.yeshi.fanli.entity.system.SystemCoupon;

import com.yeshi.fanli.exception.money.OrderMoneySettleException;

import com.yeshi.fanli.exception.mq.MQUnSendInfoException;

import com.yeshi.fanli.exception.order.InviteOrderSubsidyException;

import com.yeshi.fanli.log.LogHelper;

import com.yeshi.fanli.service.inter.mq.MQUnSendInfoService;

import com.yeshi.fanli.service.inter.order.InviteOrderSubsidyService;

import com.yeshi.fanli.service.inter.order.OrderMoneySettleService;

import com.yeshi.fanli.service.inter.user.UserSystemCouponRecordService;

import com.yeshi.fanli.util.Constant;

import com.yeshi.fanli.util.rocketmq.MQTopicName;



/**

 * 邀请订单补贴消息消费

@@ -15,6 +38,20 @@
 */

@Component

public class InviteOrderSubsidyMessageListener implements MessageListener {

   @Resource

   private InviteOrderSubsidyService inviteOrderSubsidyService;



   @Resource

   private UserSystemCouponRecordService userSystemCouponRecordService;



   @Resource(name = "producer")

   private Producer producer;



   @Resource

   private MQUnSendInfoService mqUnSendInfoService;



   @Resource

   private OrderMoneySettleService orderMoneySettleService;



   public InviteOrderSubsidyMessageListener() {



@@ -22,9 +59,118 @@


   @Override

   public Action consume(Message message, ConsumeContext context) {

      if (message.getTag().startsWith("")) {



      // if (!Constant.IS_TASK) {// 非任务服务器不能消费消息

      // return Action.ReconsumeLater;

      // }

      LogHelper.mqInfo("consumer:InviteOrderSubsidyMessageListener", message.getTopic(), message.getTag(),

            new String(message.getBody()));



      if (MQTopicName.TOPIC_ORDER.name().equalsIgnoreCase(message.getTopic())) {

         // 查询上级红包

         OrderMQMsgDTO dto = new Gson().fromJson(new String(message.getBody()), OrderMQMsgDTO.class);

         if (dto != null) {

            try {

               inviteOrderSubsidyService.addOrUpdateByOrder(dto.getOrderId(), dto.getType());

            } catch (InviteOrderSubsidyException e) {

               return Action.ReconsumeLater;

            }

         }

      } else if (MQTopicName.TOPIC_FANLI.name().equalsIgnoreCase(message.getTopic())) {

         if (message.getTag().equalsIgnoreCase("actual")) {

            OrderMoneyRecievedMQMsgDTO dto = new Gson().fromJson(new String(message.getBody()),

                  OrderMoneyRecievedMQMsgDTO.class);

            if (dto != null) {

               if (dto.getType() == OrderMoneyRecievedMQMsgDTO.TYPE_ZIGOU) {// 自购到账

                  // 判断订单号是否在补贴中

                  long count = inviteOrderSubsidyService.countByOrderIdAndSourceType(dto.getOrderId(),

                        dto.getSourceType());

                  if (count > 0) {// 延时消息发送

                     sendOrderSubsidyUpdateLater(dto, 5);

                  }

               } else if (dto.getType() == OrderMoneyRecievedMQMsgDTO.TYPE_INVITE) {// 邀请到账

                  // 补贴到账

                  try {

                     orderMoneySettleService.inviteSubsidySettle(dto.getUid(), message.getKey());

                     return Action.CommitMessage;

                  } catch (OrderMoneySettleException e) {

                     return Action.ReconsumeLater;

                  }

               }

            }

         } else if (message.getTag().equalsIgnoreCase("delay")) {

            OrderMoneyRecievedMQMsgDTO dto = new Gson().fromJson(new String(message.getBody()),

                  OrderMoneyRecievedMQMsgDTO.class);

            if (dto != null) {

               if (dto.getType() == OrderMoneyRecievedMQMsgDTO.TYPE_ZIGOU) {// 延时时间已到

                  // 判断订单号是否在补贴中

                  long count = inviteOrderSubsidyService.countByOrderIdAndSourceType(dto.getOrderId(),

                        dto.getSourceType());

                  if (count > 0) {

                     // 判断是否要使用奖励券

                     UserSystemCouponRecord record = userSystemCouponRecordService

                           .getRecordByOrderNo(dto.getOrderId(), UserSystemCouponRecord.STATE_SUCCESS);

                     if (record != null

                           && record.getCouponType()

                                 .equalsIgnoreCase(SystemCoupon.CouponTypeEnum.rebatePercentCoupon.name())

                           && record.getGoodSource() == dto.getSourceType()) {// 成功使用了奖励券

                        inviteOrderSubsidyService.invalidByOrderIdAndSourceType(dto.getOrderId(),

                              dto.getSourceType());

                     } else {

                        // 使数据有效

                        inviteOrderSubsidyService.validByOrderIdAndSourceType(dto.getOrderId(),

                              dto.getSourceType());

                     }

                  }

               }

            }

         }



      } else if (MQTopicName.TOPIC_SYSTEM_COUPON_USE.name().equalsIgnoreCase(message.getTopic())) {// 奖励券使用成功

         UserSystemCouponUseMQMsgDTO dto = new Gson().fromJson(new String(message.getBody()),

               UserSystemCouponUseMQMsgDTO.class);

         if (dto != null) {

            long count = inviteOrderSubsidyService.countByOrderIdAndSourceType(dto.getOrderId(),

                  dto.getSourceType());

            if (count > 0) {

               // 使相关订单失效

               inviteOrderSubsidyService.invalidByOrderIdAndSourceType(dto.getOrderId(), dto.getSourceType());

            }

            return Action.CommitMessage;

         }

      }

      return Action.CommitMessage;

   }



   /**

    * 发送订单延时消息

    * 
    * @param day

    */

   private boolean sendOrderSubsidyUpdateLater(OrderMoneyRecievedMQMsgDTO dto, int day) {



      Message msg = new Message(MQTopicName.TOPIC_FANLI.name(), "delay", new Gson().toJson(dto).getBytes());

      msg.setKey(dto.getSourceType() + "_" + dto.getOrderId());

      long delayTime = System.currentTimeMillis() + (Constant.IS_TEST ? 1000 * 30L : 1000 * 60 * 60 * 24L * day);

      msg.setStartDeliverTime(delayTime);

      SendResult sendResult = producer.send(msg);

      if (sendResult != null) {

         return true;

      } else {

         MQUnSendInfo info = new MQUnSendInfo();

         info.setBody(new String(msg.getBody()));

         if (msg.getStartDeliverTime() > 0)

            info.setDeliverTime(new Date(msg.getStartDeliverTime()));

         info.setKey(msg.getKey());

         info.setTag(msg.getTag());

         info.setTopic(msg.getTopic());

         try {

            mqUnSendInfoService.addMQUnSendInfo(info);

         } catch (MQUnSendInfoException e) {

            e.printStackTrace();

         }

         return false;

      }

   }



}