送礼记录-雪花主键-无锁化唯一主键插入

accompany-base/accompany-sharding/accompany-sharding-sdk/src/main/java/com/accompany/sharding/model/GiftSendRecord.java

@@ -8,7 +8,7 @@ import java.util.Date;
 
 @Data
 public class GiftSendRecord {
-    @TableId(type = IdType.AUTO)
+    @TableId(type = IdType.INPUT)
     private Long sendRecordId;
 
     private Long uid;

accompany-base/accompany-sharding/accompany-sharding-service/src/main/java/com/accompany/sharding/mapper/GiftSendRecordMapper.java

@@ -2,7 +2,8 @@ package com.accompany.sharding.mapper;
 
 import com.accompany.sharding.model.GiftSendRecord;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Param;
 
 public interface GiftSendRecordMapper extends BaseMapper<GiftSendRecord> {
-
+    int insertIgnore(@Param("record") GiftSendRecord record);
-}
+}

accompany-base/accompany-sharding/accompany-sharding-service/src/main/resources/sharding/sqlmappers/GiftSendRecordMapper.xml

@@ -1,5 +1,43 @@
 <?xml version="1.0" encoding="UTF-8" ?>
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
 <mapper namespace="com.accompany.sharding.mapper.GiftSendRecordMapper">
-
+    <insert id="insertIgnore" parameterType="com.accompany.sharding.model.GiftSendRecord">
-</mapper>
+        INSERT IGNORE INTO gift_send_record (
+            send_record_id,
+            uid,
+            partition_id,
+            recive_uid,
+            recive_type,
+            send_env,
+            room_uid,
+            room_type,
+            gift_id,
+            gift_num,
+            gift_type,
+            play_effect,
+            total_gold_num,
+            total_diamond_num,
+            create_time,
+            gift_source,
+            mess_id
+        ) VALUES (
+            #{record.sendRecordId},
+            #{record.uid},
+            #{record.partitionId},
+            #{record.reciveUid},
+            #{record.reciveType},
+            #{record.sendEnv},
+            #{record.roomUid},
+            #{record.roomType},
+            #{record.giftId},
+            #{record.giftNum},
+            #{record.giftType},
+            #{record.playEffect},
+            #{record.totalGoldNum},
+            #{record.totalDiamondNum},
+            #{record.createTime},
+            #{record.giftSource},
+            #{record.messId}
+        )
+    </insert>
+</mapper>

accompany-business/accompany-business-sdk/src/main/java/com/accompany/business/message/GiftMessage.java

@@ -9,6 +9,7 @@ import lombok.Data;
  */
 @Data
 public class GiftMessage extends BaseMqMessage {
+    private Long id;
     private String messId;    // 消息唯一标识
     private Long messTime;    // 消息创建时间
     private Long sendUid;     // 发送者UID

accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/gift/GiftMessageService.java

@@ -1,7 +1,6 @@
 package com.accompany.business.service.gift;
 
 import com.accompany.business.event.ActivityOfDayConsumeEvent;
-import com.accompany.business.event.AnchorFansSendGiftEvent;
 import com.accompany.business.event.GiftMessageEvent;
 import com.accompany.business.event.RoomPKEvent;
 import com.accompany.business.message.ActivityOfDayConsumeMessage;
@@ -20,6 +19,7 @@ import com.accompany.core.service.base.BaseService;
 import com.accompany.core.vo.vip.VipBaseInfoVO;
 import com.accompany.sharding.model.GiftSendRecord;
 import com.alibaba.fastjson.JSON;
+import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.ApplicationContext;
@@ -48,6 +48,8 @@ public class GiftMessageService extends BaseService {
     private ActivityUserLevelExpService activityUserLevelExpService;
     @Autowired
     private GiftRateService giftRateService;
+    @Autowired
+    private IdentifierGenerator identifierGenerator;
 
     public void handleGiftMessage(GiftMessage giftMessage) {
         // 防止消息被重复消费
@@ -132,7 +134,92 @@ public class GiftMessageService extends BaseService {
         //发布事件
         applicationContext.publishEvent(new GiftMessageEvent(giftMessage));
         applicationContext.publishEvent(new RoomPKEvent(giftMessage));
-        applicationContext.publishEvent(new AnchorFansSendGiftEvent(giftSendRecord));
+        // applicationContext.publishEvent(new AnchorFansSendGiftEvent(giftSendRecord));
+
+        // 删除该标识，表示消息已经消费过
+        jedisService.hdel(RedisKey.mq_gift_status.getKey(), giftMessage.getMessId());
+    }
+
+    public void handleGiftMessageV2(GiftMessage giftMessage) {
+
+        logger.info("【处理礼物mq】 giftMessage: {}", giftMessage);
+
+        long totalGoldNum = giftMessage.getGoldNum();
+        byte giftType = giftMessage.getGiftType();
+        giftMessage.setRealGoldNum(totalGoldNum);
+        if (Constant.GiftType.SUPER_LUCKY == giftType
+                || Constant.GiftType.LUCKY_24 == giftType
+                || Constant.GiftType.BRAVO_GIFT == giftType
+                || Constant.GiftType.LUCKY_25 == giftType){
+            giftMessage.setGoldNum(0L);
+        }
+
+        //只用来insert,下面还会算一次普通礼物的钻石流水，这里的流水不给事件消费，因为幸运礼物是单独走的消费
+        Double diamondFlow = giftRateService.calDiamondFlow(giftMessage.getPartitionId(), giftMessage.getRoomUid(), giftType, giftMessage.getRealGoldNum());
+
+        GiftSendRecord giftSendRecord = buildGiftSendRecord(giftMessage, diamondFlow);
+
+        // 先插入giftSendRecord
+        int row = giftSendRecordService.insertRecordIgnore(giftSendRecord);
+        if (row <= 0){
+            return;
+        }
+
+        logger.info("【处理礼物mq】 giftMessageId {} 插入gift_send_record {}", giftMessage.getMessId(), JSON.toJSONString(giftSendRecord));
+
+        String objId = giftSendRecord.getSendRecordId().toString();
+        double amount = giftMessage.getRealGoldNum().doubleValue();
+        BillObjTypeEnum inEnum = BillObjTypeEnum.GIFT_PERSON_INCOME;
+        BillObjTypeEnum outEnum = BillObjTypeEnum.GIFT_PERSON_PAY;
+        if (giftMessage.getRoomUid() != null && giftMessage.getRoomUid() > 0) {
+            inEnum = BillObjTypeEnum.GIFT_ROOM_INCOME;
+            outEnum = BillObjTypeEnum.GIFT_ROOM_PAY;
+        }
+        if (giftMessage.getGiftType() == GiftTypeEnum.LUCKY_24.getType()){
+            outEnum = BillObjTypeEnum.LUCKY_24_GIFT_PAY;
+        } else if (giftMessage.getGiftType() == GiftTypeEnum.BRAVO.getType()) {
+            outEnum = BillObjTypeEnum.BRAVO_GIFT_PAY;
+        } else if (giftMessage.getGiftType() == GiftTypeEnum.LUCKY_25.getType()) {
+            outEnum = BillObjTypeEnum.LUCKY_25_GIFT_PAY;
+        }
+
+        Date giftSendTime = new Date(giftMessage.getMessTime());
+
+        // 背包送礼不算钱
+        if (null != giftMessage.getGiftSource() && Constant.GiftSource.BACKPACK == giftMessage.getGiftSource()) {
+            amount = 0D;
+            if (null != giftMessage.getAfterPurse()){
+                giftMessage.getAfterPurse().setUpdateTime(giftSendTime);
+            }
+        }
+
+        // 插入账单记录表（发送礼物用户）
+        billRecordService.insertGiftSendBillRecord(giftMessage.getSendUid(), giftMessage.getRecvUid(),
+                giftMessage.getRoomUid(), objId, outEnum, amount,
+                giftMessage.getGiftId(), giftMessage.getGiftNum(), totalGoldNum, giftSendTime,
+                giftMessage.getAfterPurse());
+
+        Double diamondNum = 0D;//钻石流水
+        //处理逻辑
+        if (Constant.GiftType.SUPER_LUCKY != giftType
+                && Constant.GiftType.LUCKY_24 != giftType
+                && Constant.GiftType.BRAVO_GIFT != giftType
+                && Constant.GiftType.LUCKY_25 != giftType) {
+            //按配置比例分配主播、 会长、房主收益
+            diamondNum = giftEarnAllotService.allotGiftEarn(giftSendRecord, objId, inEnum);
+            //增加送礼用户和收礼用户的财富或魅力经验
+            addUserExperience(giftMessage);
+
+            applicationContext.publishEvent(new ActivityOfDayConsumeEvent(ActivityOfDayConsumeMessage.builder()
+                    .sendUid(giftMessage.getSendUid())
+                    .messTime(new Date(giftMessage.getMessTime()))
+                    .totalDiamondNum(giftMessage.getGoldNum().doubleValue()).consumeType("Normal Gift").build()));
+        }
+        giftMessage.setDiamondNum(diamondNum);
+        //发布事件
+        applicationContext.publishEvent(new GiftMessageEvent(giftMessage));
+        applicationContext.publishEvent(new RoomPKEvent(giftMessage));
+        // applicationContext.publishEvent(new AnchorFansSendGiftEvent(giftSendRecord));
 
         // 删除该标识，表示消息已经消费过
         jedisService.hdel(RedisKey.mq_gift_status.getKey(), giftMessage.getMessId());
@@ -200,6 +287,35 @@ public class GiftMessageService extends BaseService {
         return new BigDecimal(goldNum).multiply(BigDecimal.valueOf(1 + Double.parseDouble(levelBuff.getItemValue()))).setScale(0, BigDecimal.ROUND_DOWN).longValue();
     }
 
+    /**
+     * 组装礼物赠送记录对象
+     *
+     * @param giftMessage 礼物消息
+     * @param diamondFlow 钻石流水
+     * @return GiftSendRecord对象
+     */
+    private GiftSendRecord buildGiftSendRecord(GiftMessage giftMessage, Double diamondFlow) {
+        // 插入送礼物记录
+        final GiftSendRecord giftSendRecord = new GiftSendRecord();
+        giftSendRecord.setSendRecordId(giftMessage.getId());
+        giftSendRecord.setGiftId(giftMessage.getGiftId());
+        giftSendRecord.setReciveUid(giftMessage.getRecvUid());
+        giftSendRecord.setRoomUid(giftMessage.getRoomUid());
+        giftSendRecord.setRoomType(giftMessage.getRoomType());
+        giftSendRecord.setGiftNum(giftMessage.getGiftNum());
+        giftSendRecord.setSendEnv(giftMessage.getSendType());
+        giftSendRecord.setUid(giftMessage.getSendUid());
+        giftSendRecord.setPartitionId(giftMessage.getPartitionId());
+        giftSendRecord.setTotalGoldNum(giftMessage.getGoldNum());
+        giftSendRecord.setTotalDiamondNum(diamondFlow);
+        giftSendRecord.setGiftType(giftMessage.getGiftType());
+        giftSendRecord.setCreateTime(new Date(giftMessage.getMessTime()));
+        giftSendRecord.setGiftSource(giftMessage.getGiftSource().byteValue());
+        giftSendRecord.setMessId(giftMessage.getMessId());
+
+        return giftSendRecord;
+    }
+
     /**
      * 插入礼物赠送记录
      *
@@ -208,6 +324,7 @@ public class GiftMessageService extends BaseService {
     private GiftSendRecord insertGiftSendRecord(GiftMessage giftMessage, Double diamondFlow) {
         // 插入送礼物记录
         final GiftSendRecord giftSendRecord = new GiftSendRecord();
+        giftSendRecord.setSendRecordId(null != giftMessage.getId()? giftMessage.getId(): identifierGenerator.nextId(null).longValue());
         giftSendRecord.setGiftId(giftMessage.getGiftId());
         giftSendRecord.setReciveUid(giftMessage.getRecvUid());
         giftSendRecord.setRoomUid(giftMessage.getRoomUid());
@@ -226,7 +343,7 @@ public class GiftMessageService extends BaseService {
         giftSendRecordService.insertGiftSendRecord(giftSendRecord);
 
         logger.info("[处理礼物mq] insert giftSendRecored finish, sendUid:{}, partitionId:{}, recvUid:{}, goldNum:{}, messId:{}" +
-                ", giftRecordId:{}", giftMessage.getSendUid(), giftMessage.getPartitionId(), giftMessage.getRecvUid(),
+                        ", giftRecordId:{}", giftMessage.getSendUid(), giftMessage.getPartitionId(), giftMessage.getRecvUid(),
                 giftMessage.getGoldNum(), giftMessage.getMessId(), giftSendRecord.getSendRecordId());
 
         return giftSendRecord;

accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/gift/GiftSendRecordService.java

@@ -1,6 +1,5 @@
 package com.accompany.business.service.gift;
 
-import com.accompany.business.model.userevent.UserEventData;
 import com.accompany.common.constant.Constant;
 import com.accompany.common.result.BusiResult;
 import com.accompany.common.status.BusiStatus;
@@ -36,6 +35,10 @@ public class GiftSendRecordService extends ServiceImpl<GiftSendRecordMapper, Gif
     @Autowired
     private GiftSendRecordMapperExpand giftSendRecordMapperExpand;
 
+    public int insertRecordIgnore(GiftSendRecord record) {
+        return baseMapper.insertIgnore(record);
+    }
+
     public void insertGiftSendRecord(GiftSendRecord giftSendRecord) {
         for (int i = 0; i < 3; i++) {
             try {

accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/gift/GiftSendService.java

@@ -51,6 +51,7 @@ import com.accompany.core.vo.vip.VipBaseInfoVO;
 import com.alibaba.fastjson.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.incrementer.DefaultIdentifierGenerator;
+import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import lombok.extern.slf4j.Slf4j;
@@ -107,6 +108,8 @@ public class GiftSendService extends BaseService {
     private BravoGiftSendService bravoGiftSendService;
     @Autowired
     private Lucky25GiftSendService lucky25GiftSendService;
+    @Autowired
+    private IdentifierGenerator identifierGenerator;
 
 
     /**
@@ -206,6 +209,7 @@ public class GiftSendService extends BaseService {
                                          Integer giftSource, Boolean luckyBagGift, String giftSendUuid, Integer luckyBagId,
                                          Date sendGiftTime, Integer partitionId, UserPurse afterPurse) {
         GiftMessage message = new GiftMessage();
+        message.setId(identifierGenerator.nextId( null).longValue());
         message.setSendUid(sendUid);
         message.setRecvUid(recvUid);
         message.setRoomUid(roomUid);

accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/mq/RocketMQService.java

@@ -43,7 +43,7 @@ public class RocketMQService {
         List<Message<String>> messageList = giftMessages.stream()
                 .map(giftMessage -> MessageBuilder.withPayload(JSON.toJSONString(giftMessage)).build())
                 .collect(Collectors.toList());
-        mqMessageProducer.send(MqConstant.GIFT_TOPIC, messageList,
+        mqMessageProducer.send(MqConstant.GIFT_TOPIC_V2, messageList,
                 sendResult -> log.info("sendGiftMessage success result: {} message: {}", JSON.toJSONString(sendResult), messageList),
                 throwable -> log.error("sendGiftMessage fail message: {}", messageList, throwable));
     }

accompany-mq/accompany-mq-sdk/src/main/java/com/accompany/mq/constant/MqConstant.java

@@ -5,6 +5,9 @@ public interface MqConstant {
     String GIFT_TOPIC = "gift_topic";
     String GIFT_CONSUME_GROUP = "gift_consume_group";
 
+    String GIFT_TOPIC_V2 = "gift_topic_v2";
+    String GIFT_CONSUME_V2_GROUP = "gift_consume_v2_group";
+
     String OPEN_BOX_TOPIC = "open_box_topic";
     String OPEN_BOX_CONSUME_GROUP = "open_box_consume_group";
 

accompany-mq/accompany-mq-web/src/main/java/com/accompany/mq/consumer/GiftMessageV2Consumer.java

@@ -0,0 +1,29 @@
+package com.accompany.mq.consumer;
+
+import com.accompany.business.message.GiftMessage;
+import com.accompany.business.service.gift.GiftMessageService;
+import com.accompany.mq.constant.MqConstant;
+import com.accompany.mq.listener.AbstractMessageListener;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.stereotype.Component;
+
+@Slf4j
+@Component
+@ConditionalOnProperty(name = "spring.application.name", havingValue = "web")
+@RocketMQMessageListener(topic = MqConstant.GIFT_TOPIC_V2, consumerGroup = MqConstant.GIFT_CONSUME_V2_GROUP)
+//@RocketMQMessageListener(topic = MqConstant.GIFT_TOPIC, consumerGroup = MqConstant.GIFT_CONSUME_GROUP, consumeMode = ConsumeMode.ORDERLY)
+public class GiftMessageV2Consumer extends AbstractMessageListener<GiftMessage> {
+
+    @Autowired
+    private GiftMessageService giftMessageService;
+
+    @Override
+    public void onMessage(GiftMessage giftMessage) {
+        log.info("onMessage giftMessage: {}", giftMessage.toString());
+        giftMessageService.handleGiftMessageV2(giftMessage);
+    }
+
+}