批量空投

liquidnet-bus-api/liquidnet-service-goblin-api/src/main/java/com/liquidnet/service/goblin/service/IGoblinNftExCodeService.java

@@ -43,5 +43,5 @@ public interface IGoblinNftExCodeService {
      * @param activityId
      * @return
      */
-    ResponseDto<Boolean> addAirdrops(MultipartFile file, String activityId);
+    ResponseDto<Boolean> addAirdrops(MultipartFile file, GoblinNftExCodeParam goblinNftExCodeParam);
 }

liquidnet-bus-common/liquidnet-common-service-base/src/main/java/com/liquidnet/service/base/constant/MQConst.java

@@ -274,6 +274,7 @@ public class MQConst {
         GOBLIN_STORE_ORDER_OPERA("goblin:stream:order:store", "group.order:store", "商铺订单操作"),
         GOBLIN_USER_ORDER_OPERA("goblin:stream:order:user", "group.order:user", "用户订单操作"),
         GOBLIN_XLS_OPERA("goblin:stream:xls", "group.xls", "xls文件操作"),
+        GOBLIN_PHONE_CODE_OPERA("goblin:stream:phone:code","goblin.phone.code","批量空投操作"),
 
         GOBLIN_UN_PAY_0("goblin:stream:order:back:0", "group.order:back", "回滚关闭订单库存队列"),
         GOBLIN_UN_PAY_1("goblin:stream:order:back:1", "group.order:back", "回滚关闭订单库存队列"),

liquidnet-bus-do/liquidnet-service-adam-do/src/main/java/com/liquidnet/service/adam/mapper/AdamUserMapper.java

@@ -49,8 +49,17 @@ public interface AdamUserMapper extends BaseMapper<AdamUser> {
 
     /**
      * 根据手机号获取 用户信息
+     *
      * @param mobile
      * @return
      */
     AdamUserInfoDto selectByPhone(@Param("mobile") String mobile);
+
+    /**
+     * 根据手机号批量获取
+     *
+     * @param mobiles
+     * @return
+     */
+    List<AdamUserInfoDto> selectByPhones(@Param("mobiles") String mobiles);
 }

liquidnet-bus-do/liquidnet-service-adam-do/src/main/resources/com/liquidnet/service/adam/mapper/AdamUserMapper.xml

@@ -7,7 +7,16 @@
     </resultMap>
 
     <select id="selectByPhone" resultMap="BaseResult">
-        select uid,mobile from adam_user where mobile = #{mobile,jdbcType=VARCHAR}
+        select uid, mobile
+        from adam_user
+        where mobile = #{mobile,jdbcType=VARCHAR}
+    </select>
+
+    <select id="selectByPhones" resultMap="BaseResult">
+        select uid,mobile from adam_user where mobile in
+        <foreach collection="mobiles.split(',')" item="mobile" open="(" separator="," close=")">
+            #{mobile}
+        </foreach>
     </select>
 
     <update id="executeForDM">

liquidnet-bus-do/liquidnet-service-goblin-do/src/main/java/com/liquidnet/service/goblin/mapper/GoblinNftExCodeMapper.java

@@ -72,6 +72,7 @@ public interface GoblinNftExCodeMapper extends BaseMapper<GoblinNftExCode> {
 
     /**
      * 修改操作用户
+     *
      * @param goblinNftExCode
      * @return
      */
@@ -79,9 +80,18 @@ public interface GoblinNftExCodeMapper extends BaseMapper<GoblinNftExCode> {
 
     /**
      * 根据活动id获取可用的兑换码
+     *
      * @param activityId
      * @return
      */
     List<GoblinNftExCode> selectCodeAvailableByActivityId(@Param("activityId") String activityId);
 
+    /**
+     * 批量修改
+     *
+     * @param codes
+     * @param adminUid
+     * @return
+     */
+    int updateCodeAdminUids(@Param("codes") String codes, @Param("adminUid") String adminUid);
 }

liquidnet-bus-do/liquidnet-service-goblin-do/src/main/resources/com.liquidnet.service.goblin.mapper/GoblinGoodsSkuMapper.xml

@@ -11,6 +11,7 @@
         <result column="status" property="status" jdbcType="VARCHAR"/>
         <result column="shelves_status" property="shelvesStatus" jdbcType="VARCHAR"/>
         <result column="sku_appear" property="skuAppear" jdbcType="VARCHAR"/>
+        <result column="sku_canbuy" property="skuCanbuy" jdbcType="VARCHAR"/>
         <result column="del_flg" property="delFlg" jdbcType="VARCHAR"/>
         <result column="soldout_status" property="soldoutStatus" jdbcType="VARCHAR"/>
         <result column="sale_start_time" property="saleStartTime" jdbcType="TIMESTAMP"/>
@@ -24,7 +25,7 @@
     </select>
 
     <select id="selectBySkuIds" resultMap="BaseResult">
-        select sku_id,name,sku_pic,hit_ratio,unbox,sku_type,status,shelves_status,sku_appear,del_flg,soldout_status,sale_start_time from goblin_goods_sku where
+        select sku_id,name,sku_pic,hit_ratio,unbox,sku_type,status,shelves_status,sku_canbuy,sku_appear,del_flg,soldout_status,sale_start_time from goblin_goods_sku where
         sku_id in
         <foreach collection="skuIds.split(',')" item="skuId" open="(" separator="," close=")">
             #{skuId,jdbcType=VARCHAR}
@@ -32,7 +33,7 @@
     </select>
 
     <select id="selectBySpuIds" resultMap="BaseResult">
-        select spu_id,sku_id,hit_ratio,unbox,sku_type,status,shelves_status,sku_appear,del_flg,soldout_status,sale_start_time from goblin_goods_sku where
+        select spu_id,sku_id,hit_ratio,unbox,sku_type,status,shelves_status,sku_appear,del_flg,sku_canbuy,soldout_status,sale_start_time from goblin_goods_sku where
         spu_id in
         <foreach collection="spuIds.split(',')" item="spuId" open="(" separator="," close=")">
             #{spuId,jdbcType=VARCHAR}

liquidnet-bus-do/liquidnet-service-goblin-do/src/main/resources/com.liquidnet.service.goblin.mapper/GoblinNftExCodeMapper.xml

@@ -65,6 +65,17 @@
         </set>
         where code_id = #{codeId}
     </update>
+    
+    <update id="updateCodeAdminUids">
+        update goblin_nft_ex_code
+        <set>
+            admin_uid = #{adminUid}
+        </set>
+        where code in
+        <foreach collection="codes.split(',')" item="code" open="(" separator="," close=")">
+            #{code}
+        </foreach>
+    </update>
 
     <select id="selectGoblinNftCode" parameterType="com.liquidnet.service.goblin.entity.GoblinNftExCode"
             resultMap="BeseResult">

liquidnet-bus-service/liquidnet-service-consumer-all/liquidnet-service-consumer-kylin/src/main/java/com/liquidnet/service/consumer/kylin/receiver/ConsumerGoblinPhoneCodeReceiver.java

+package com.liquidnet.service.consumer.kylin.receiver;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.liquidnet.commons.lang.util.*;
+import com.liquidnet.service.base.ResponseDto;
+import com.liquidnet.service.base.constant.MQConst;
+import com.liquidnet.service.goblin.mapper.GoblinNftExCodeMapper;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+import org.springframework.util.LinkedMultiValueMap;
+import org.springframework.util.MultiValueMap;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.Map;
+
+@Slf4j
+@Component
+public class ConsumerGoblinPhoneCodeReceiver extends AbstractBizRedisReceiver{
+
+    @Autowired
+    private GoblinNftExCodeMapper goblinNftExCodeMapper;
+    @Value("${liquidnet.service.order.url}")
+    private String orderUrl;
+
+    @Override
+    protected boolean consumerMessageHandler(String msg) {
+        Boolean bol = false;
+        try{
+            JSONObject jsonObject = JSON.parseObject(msg);
+            String codestr = jsonObject.getString("codes");
+            String userIds = jsonObject.getString("userIds");
+            String adminUid = jsonObject.getString("adminUid");
+            List<String> codeList =  Arrays.asList(codestr.split(","));
+            List<String> userIdList = Arrays.asList(userIds.split(","));
+
+
+            StringBuffer codes = new StringBuffer();
+            for (int i = 0; i < userIdList.size(); i++) {
+                MultiValueMap<String, String> params = new LinkedMultiValueMap();
+                params.add("code", codeList.get(i));
+                params.add("userId", userIdList.get(i));
+                MultiValueMap<String, String> headers = CollectionUtil.linkedMultiValueMapStringString();
+                headers.add("Accept", "application/json;charset=UTF-8");
+                headers.add("Authorization", "Bearer " + CurrentUtil.getToken());
+                String post = HttpUtil.post(orderUrl + "/order/goblin/nft/airdrop", params, headers);
+                ResponseDto<Boolean> rsp = JsonUtils.fromJson(post, new TypeReference<ResponseDto<Boolean>>() {
+                });
+                if (rsp.isSuccess()) {
+                    codes.append(codeList.get(i)).append(",");
+                }
+            }
+            if (StringUtil.isNotBlank(codes)){
+                codes.deleteCharAt(codes.length()-1);
+            }
+            goblinNftExCodeMapper.updateCodeAdminUids(codes.toString(),adminUid);
+        }catch (Exception e){
+            log.error("发送空投失败!");
+        }
+        return bol;
+    }
+
+    @Override
+    protected String getRedisStreamKey() {
+        return MQConst.GoblinQueue.GOBLIN_PHONE_CODE_OPERA.getKey();
+    }
+
+    @Override
+    protected String getRedisStreamGroup() {
+        return MQConst.GoblinQueue.GOBLIN_PHONE_CODE_OPERA.getGroup();
+    }
+}

liquidnet-bus-service/liquidnet-service-platform/liquidnet-service-platform-impl/src/main/java/com/liquidnet/service/platform/controller/goblin/code/GoblinNftExCodeController.java

@@ -56,9 +56,9 @@ public class GoblinNftExCodeController {
 
     @ApiOperationSupport(order = 4)
     @ApiOperation(value = "excel空投批量发送")
-    @PostMapping("addAirdrops/{activityId}")
-    public ResponseDto<Boolean> addAirdrops(MultipartFile file,@PathVariable("activityId")String activityId) {
-        return iGoblinNftExCodeService.addAirdrops(file,activityId);
+    @PostMapping("addAirdrops")
+    public ResponseDto<Boolean> addAirdrops(MultipartFile file,@RequestBody GoblinNftExCodeParam goblinNftExCodeParam) {
+        return iGoblinNftExCodeService.addAirdrops(file,goblinNftExCodeParam);
     }
 
 

liquidnet-bus-service/liquidnet-service-platform/liquidnet-service-platform-impl/src/main/java/com/liquidnet/service/platform/service/impl/goblin/GoblinNftExCodeServiceImpl.java

@@ -5,12 +5,15 @@ import com.alibaba.excel.read.listener.PageReadListener;
 import com.alibaba.excel.support.ExcelTypeEnum;
 import com.alibaba.excel.write.metadata.style.WriteCellStyle;
 import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.github.pagehelper.PageHelper;
 import com.liquidnet.commons.lang.util.*;
 import com.liquidnet.service.adam.dto.AdamUserInfoDto;
 import com.liquidnet.service.adam.mapper.AdamUserMapper;
 import com.liquidnet.service.base.ResponseDto;
+import com.liquidnet.service.base.constant.MQConst;
 import com.liquidnet.service.goblin.dto.vo.CodeExcelVo;
 import com.liquidnet.service.goblin.dto.vo.GoblinNftExCodeFetchExcelVo;
 import com.liquidnet.service.goblin.dto.vo.GoblinNftExCodeVo;
@@ -25,6 +28,7 @@ import com.liquidnet.service.goblin.param.GoblinNftExCodeParam;
 import com.liquidnet.service.goblin.service.IGoblinNftExCodeService;
 import com.liquidnet.service.platform.utils.GoblinRedisUtils;
 import com.liquidnet.service.platform.utils.ObjectUtil;
+import com.liquidnet.service.platform.utils.QueueUtils;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.poi.ss.usermodel.HorizontalAlignment;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -41,6 +45,8 @@ import java.net.URLEncoder;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
 import java.util.List;
 
 /**
@@ -64,6 +70,8 @@ public class GoblinNftExCodeServiceImpl implements IGoblinNftExCodeService {
     private AdamUserMapper adamUserMapper;
     @Value("${liquidnet.service.order.url}")
     private String orderUrl;
+    @Autowired
+    private QueueUtils queueUtils;
 
     @Override
     public PageInfoVo selectCodePageList(GoblinNftExCodeParam goblinNftExCodeParam) {
@@ -232,7 +240,7 @@ public class GoblinNftExCodeServiceImpl implements IGoblinNftExCodeService {
                 return ResponseDto.failure(rsp.getMessage());
             }
         } catch (Exception e) {
-            e.printStackTrace();
+            // e.printStackTrace();
             return ResponseDto.failure("发送公投失败！");
         }
     }
@@ -296,8 +304,6 @@ public class GoblinNftExCodeServiceImpl implements IGoblinNftExCodeService {
                             break;
                         }
                     }
-
-
                 }
             }
         }
@@ -368,31 +374,71 @@ public class GoblinNftExCodeServiceImpl implements IGoblinNftExCodeService {
     }
 
     @Override
-    public ResponseDto<Boolean> addAirdrops(MultipartFile file, String activityId) {
-        try{
+    public ResponseDto<Boolean> addAirdrops(MultipartFile file, GoblinNftExCodeParam goblinNftExCodeParam) {
+        try {
             // 手机号集合
-            ArrayList<String> phones = CollectionUtil.arrayListString();
-            EasyExcel.read(file.getInputStream(), GoblinNftExCodeFetchExcelVo.class, new PageReadListener<GoblinNftExCodeFetchExcelVo>(dataList->{
+            StringBuffer phonestr = new StringBuffer();
+            EasyExcel.read(file.getInputStream(), GoblinNftExCodeFetchExcelVo.class, new PageReadListener<GoblinNftExCodeFetchExcelVo>(dataList -> {
                 for (GoblinNftExCodeFetchExcelVo data : dataList) {
                     if (data.getPhone() == null) {
                         continue;
                     }
-                    phones.add(data.getPhone());
+                    phonestr.append(data.getPhone()).append(",");
                 }
             })).sheet().doRead();
+
+            if (StringUtil.isNotBlank(phonestr)) {
+                phonestr.deleteCharAt(phonestr.length() - 1);
+            }
+
+            // 根据手机号获取用户
+            List<AdamUserInfoDto> adamUserMappers = adamUserMapper.selectByPhones(phonestr.toString());
+
             // 根据活动id获取可用兑换码
-            List<GoblinNftExCode> goblinNftExCodes = goblinNftExCodeMapper.selectCodeAvailableByActivityId(activityId);
-            if (goblinNftExCodes.size() < phones.size()){
+            List<GoblinNftExCode> goblinNftExCodes = goblinNftExCodeMapper.selectCodeAvailableByActivityId(goblinNftExCodeParam.getActivityId());
+            if (goblinNftExCodes.size() < adamUserMappers.size()) {
                 return ResponseDto.failure("可用兑换码数量不足!");
             }
 
 
-        }catch (Exception e){
+            // 页条数
+            int pageSize = 2;
+            ArrayList<String> userIdList = CollectionUtil.arrayListString();
+            StringBuffer userIds = new StringBuffer();
+            int i = 1;
+            for (AdamUserInfoDto adamUserInfoDto : adamUserMappers) {
+                userIdList.add(adamUserInfoDto.getUid());
+                userIds.append(adamUserInfoDto.getUid()).append(",");
+                if (userIdList.size() == pageSize) {
+                    StringBuffer codes = new StringBuffer();
+                    PageHelper.startPage(i, pageSize, true);
+                    List<GoblinNftExCode> goblinNftExCodeList = goblinNftExCodeMapper.selectCodeAvailableByActivityId(goblinNftExCodeParam.getActivityId());
+                    for (GoblinNftExCode goblinNftExCode : goblinNftExCodeList) {
+                        codes.append(goblinNftExCode.getCode()).append(",");
+                    }
+                    JSONObject jsonObject = new JSONObject();
+                    codes.deleteCharAt(codes.length() - 1);
+                    userIds.deleteCharAt(userIds.length() - 1);
+                    jsonObject.put("codes", codes);
+                    jsonObject.put("userIds", userIds);
+                    jsonObject.put("adminUid",goblinNftExCodeParam.getAdminUid());
+                    queueUtils.sendMssPhoneCodeRedis(MQConst.GoblinQueue.GOBLIN_PHONE_CODE_OPERA.getKey(), jsonObject.toJSONString());
+                    userIdList.clear();
+                    i++;
+                }
+            }
+
+
+        } catch (Exception e) {
             e.printStackTrace();
         }
         return null;
     }
 
+    public static void main(String[] args) {
+
+
+    }
 
     /**
      * 导出

liquidnet-bus-service/liquidnet-service-platform/liquidnet-service-platform-impl/src/main/java/com/liquidnet/service/platform/service/impl/goblin/GoblinNftExSkuServiceImpl.java

@@ -230,6 +230,7 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
         // Map<String, Map<String, Object>> mapMap = new HashMap<>();
 
         // sku --> 概率/库存
+        Integer countStockNumber = 10;
         HashMap<String, Map<String, Object>> mapMap = CollectionUtil.mapHashMap();
         for (String key : map.keySet()) {
             BigDecimal skuHitRatio = BigDecimal.valueOf(Double.valueOf(map.get(key).toString()));
@@ -240,6 +241,7 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
             hitRatioCount = hitRatioCount.add(skuHitRatio);
             objectMap.put("hitRatio", skuHitRatio);
             objectMap.put("stock", goblinRedisUtils.getSkuStock(key));
+            countStockNumber += goblinRedisUtils.getSkuStock(key);
             mapMap.put(key, objectMap);
 
         }
@@ -247,7 +249,7 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
 
         // sku ---> 库存
         HashMap<String, Integer> stockMap = CollectionUtil.mapStringInteger();
-        getStock(exStock, hitRatioCount, mapMap, stockMap);
+        getStock(exStock, hitRatioCount, mapMap, stockMap,countStockNumber);
         return stockMap;
     }
 
@@ -259,19 +261,22 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
 
         Map<String, Map<String, Object>> mapMap = new HashMap<>();
         Map<String, Object> objectMap = new HashMap<>();
-        objectMap.put("hitRatio", 33.33);
-        objectMap.put("stock", 5);
+        objectMap.put("hitRatio", 20);
+        objectMap.put("stock", 3);
         mapMap.put("1", objectMap);
         Map<String, Object> objectMap1 = new HashMap<>();
-        objectMap1.put("hitRatio", 33.33);
-        objectMap1.put("stock", 100);
+        objectMap1.put("hitRatio", 20);
+        objectMap1.put("stock", 10);
         mapMap.put("2", objectMap1);
         Map<String, Object> objectMap2 = new HashMap<>();
-        objectMap2.put("hitRatio", 33.33);
-        objectMap2.put("stock", 100);
+        objectMap2.put("hitRatio", 20);
+        objectMap2.put("stock", 2);
         mapMap.put("3", objectMap2);
         Map<String, Integer> stockMap = new HashMap<>();
-        getStock(10, new BigDecimal(99.99), mapMap, stockMap);
+        getStock(20, new BigDecimal(60), mapMap, stockMap,15);
+        for (String key : stockMap.keySet()) {
+            System.out.println("skuId" + key + "***********数量" + stockMap.get(key));
+        }
         System.out.println("kkk");
     }
 
@@ -283,8 +288,8 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
      * @param mapMap
      * @return
      */
-    private static void getStock(Integer exStock, BigDecimal hitRatioCount, Map<String, Map<String, Object>> mapMap, Map<String, Integer> stockMap) {
-        if (exStock <= 0) {
+    private static void getStock(Integer exStock, BigDecimal hitRatioCount, Map<String, Map<String, Object>> mapMap, Map<String, Integer> stockMap,Integer countStockNumber) {
+        if (exStock <= 0 || countStockNumber == 0) {
             return;
         }
 
@@ -306,6 +311,7 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
 
             if (subStock > Integer.valueOf(stock.toString())) {
                 subNumber += Integer.valueOf(stock.toString());
+                countStockNumber -= Integer.valueOf(stock.toString());
                 mapMap.get(key).put("stock", 0);
                 if (stockMap.get(key) == null) {
                     stockMap.put(key, Integer.valueOf(stock.toString()));
@@ -314,6 +320,8 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
                 }
 
             } else {
+                countStockNumber -= subStock;
+                mapMap.get(key).put("stock", Integer.valueOf(stock.toString()) - subStock);
                 subNumber += subStock;
                 if (stockMap.get(key) == null) {
                     stockMap.put(key, subStock);
@@ -356,7 +364,7 @@ public class GoblinNftExSkuServiceImpl implements IGoblinNftExSkuService {
                     hitRatioCount = hitRatioCount.subtract(new BigDecimal(hitRatio.toString()));
                 }
             }
-            getStock(exStock, hitRatioCount, mapMap, stockMap);
+            getStock(exStock, hitRatioCount, mapMap, stockMap,countStockNumber);
         }
     }
 

liquidnet-bus-service/liquidnet-service-platform/liquidnet-service-platform-impl/src/main/java/com/liquidnet/service/platform/utils/QueueUtils.java

@@ -12,7 +12,7 @@ import java.util.HashMap;
 @Component
 public class QueueUtils {
 
-//    @Autowired
+    //    @Autowired
 //    private RabbitTemplate rabbitTemplate;
     @Autowired
     RedisDataSourceUtil redisDataSourceUtil;
@@ -64,4 +64,16 @@ public class QueueUtils {
         map.put(MQConst.QUEUE_MESSAGE_KEY, jsonMsg);
         redisDataSourceUtil.getRedisGoblinUtil().getStringRedisTemplate().opsForStream().add(StreamRecords.mapBacked(map).withStreamKey(streamKey));
     }
+
+    /**
+     * 发送消息 -REDIS
+     *
+     * @param
+     */
+    public void sendMssPhoneCodeRedis(String streamKey, String jsonMsg) {
+        HashMap<String, String> map = CollectionUtil.mapStringString();
+        map.put(MQConst.QUEUE_MESSAGE_KEY, jsonMsg);
+        redisDataSourceUtil.getRedisGoblinUtil().getStringRedisTemplate().opsForStream().add(StreamRecords.mapBacked(map).withStreamKey(streamKey));
+    }
+
 }