uni-push分组

guoshilong

2023-08-15 6d32f3ba247c42c675bd09b1b7aa9a37c926f65e

 skjcmanager/skjcmanager-service/skjcmanager-alerts/src/main/java/cn/gistack/alerts/push/service/impl/PushServiceImpl.java

@@ -14,16 +14,13 @@
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.*;

@Service
@AllArgsConstructor
public class PushServiceImpl implements IPushService {
   private static final String PUSH_CLIENT_ID = "push:client:id:";
   private static final String UNI_CLOUD_PUSH_URL = "https://fc-mp-ad1ed7e3-700e-4fca-abae-0d097cc222b5.next.bspapp.com/push";
   private static final String UNI_CLOUD_PUSH_URL = "https://fc-mp-ff11cc4d-ca8e-4b47-a57f-e26804379530.next.bspapp.com/push";

   private BladeRedis bladeRedis;
   private IMessageClient messageClient;
@@ -38,8 +35,7 @@
   @Override
   public Boolean pushMessageList(List<MessageRecord> messageRecordList) {
      // 加入测试人员
//      addSendMsgTest(messageRecordList);
      PushVO pushVO = new PushVO();
      addSendMsgTest(messageRecordList);
      List<String> cIdList = new ArrayList<>();
      //相同的消息推送给多用户
      messageRecordList.forEach(messageRecord -> {
@@ -47,11 +43,25 @@
         String clientId = getClientIdByUserId(messageRecord.getRecipient());
         cIdList.add(clientId);
      });
      pushVO.setCIds(cIdList);
      pushVO.setTitle(messageRecordList.get(0).getTheme());
      pushVO.setContent(messageRecordList.get(0).getContent());

      return pushMessage(pushVO);
      //uni-push 一次推送最多500名用户，所以分批推送
      List<List<String>> groupList = groupListByQuantity(cIdList, 400);
      groupList.forEach(item->{
         PushVO pushVO = new PushVO();

         pushVO.setCIds(cIdList);
         pushVO.setTitle(messageRecordList.get(0).getTheme());
         pushVO.setContent(messageRecordList.get(0).getContent());

         HashMap<String,Object> payload = new HashMap<>();
         payload.put("title",messageRecordList.get(0).getTheme());
         payload.put("content",messageRecordList.get(0).getContent());
         payload.put("payload","test");
         pushVO.setPayload(payload);
          pushMessage(pushVO);

      });
      return true;
   }

   /**
@@ -110,6 +120,13 @@
      cIdList.add(clientIdByUserId);
      pushVO.setCIds(cIdList);

      HashMap<String,Object> payload = new HashMap<>();
      payload.put("title",pushVO.getTitle());
      payload.put("content",pushVO.getContent());
      payload.put("payload","test");
      pushVO.setPayload(payload);


      return pushMessage(pushVO);
   }

@@ -141,18 +158,46 @@
    * @return
    */
   public Boolean pushMessage(PushVO pushVO) {

      RestTemplate template = new RestTemplate();
      Map<String, Object> params = new LinkedHashMap<>();
      //用户cId，可数组，可字符串
      params.put("push_clientid", pushVO.getCIds());
      params.put("pushClientid", pushVO.getCIds());
      //标题
      params.put("title", pushVO.getTitle());
      //内容
      params.put("content", pushVO.getContent());
//      UniCloudReturnRes uniCloudReturnRes = template.getForObject(UNI_CLOUD_PUSH_URL,UniCloudReturnRes.class,params);

      params.put("requestId", UUID.randomUUID());

      params.put("payload",pushVO.getPayload());
      UniCloudReturnRes uniCloudReturnRes = template.postForObject(UNI_CLOUD_PUSH_URL, params, UniCloudReturnRes.class);
      return uniCloudReturnRes.getErrCode().equals("0") ? true : false;

   }

   /**
    * 对集合按指定数量进行分组
    * @param list 分组的集合
    * @param quantity 数量
    * @param <T>
    * @return
    */
   public  <T> List<List<T>> groupListByQuantity(List list, int quantity) {
      if (list == null || list.size() == 0) {
         return list;
      }

      if (quantity <= 0) {
         new IllegalArgumentException("Wrong quantity.");
      }

      List wrapList = new ArrayList();
      int count = 0;
      while (count < list.size()) {
         wrapList.add(list.subList(count, (count + quantity) > list.size() ? list.size() : count + quantity));
         count += quantity;
      }

      return wrapList;
   }
}