| | |
| | | package com.yeshi.fanli.util.mq.cmq; |
| | | |
| | | import com.google.gson.Gson; |
| | | import com.qcloud.cmq.Message; |
| | | import com.yeshi.fanli.dto.push.PushContentDetailDTO; |
| | | import com.yeshi.fanli.util.Constant; |
| | | import org.yeshi.utils.CMQUtil; |
| | | import com.yeshi.fanli.util.mq.rabbit.RabbitmqSenderUtil; |
| | | import org.springframework.amqp.rabbit.core.RabbitTemplate; |
| | | import org.springframework.stereotype.Component; |
| | | |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import javax.annotation.Resource; |
| | | |
| | | @Component |
| | | public class PushCMQManager { |
| | | |
| | | private static String secretId = "AKIDTlpgJhLjOozvd6QI2XnpfGbgV4NQJk25"; |
| | | private static String secretKey = "xhCSUHo55oHUQ6XicFcmfIgspX0EEzWo"; |
| | | private static PushCMQManager pushCMQManager; |
| | | private static CMQUtil cmqUtil; |
| | | @Resource |
| | | private RabbitTemplate rabbitTemplate; |
| | | |
| | | |
| | | public static String PUSH_HW = "push-hw"; |
| | | |
| | | static { |
| | | cmqUtil = CMQUtil.getInstance(secretId, secretKey); |
| | | TDMQUtil.getInstance().init(secretId, secretKey, Constant.TDMQ_PUBLIC); |
| | | |
| | | PUSH_HW += "-" + Constant.systemCommonConfig.getProjectName(); |
| | | TDMQUtil.getInstance().createQueue(PUSH_HW); |
| | | } |
| | | |
| | | public static PushCMQManager getInstance() { |
| | | if (pushCMQManager == null) |
| | | pushCMQManager = new PushCMQManager(); |
| | | return pushCMQManager; |
| | | } |
| | | |
| | | /** |
| | | * @param dto |
| | |
| | | if (dto == null) |
| | | return; |
| | | String json = new Gson().toJson(dto); |
| | | TDMQUtil.getInstance().sendMsg(PUSH_HW, json); |
| | | } |
| | | |
| | | /** |
| | | * 消费队列消息 |
| | | * |
| | | * @param count |
| | | * @return |
| | | */ |
| | | public Map<String, PushContentDetailDTO> consumeHWPush(int count) { |
| | | List<Message> list = Constant.TDMQ_CONSUMER ? TDMQUtil.getInstance().recieveMsg(count, PUSH_HW) : cmqUtil.recieveMsg(count, PUSH_HW); |
| | | Map<String, PushContentDetailDTO> map = new HashMap<>(); |
| | | if (list != null) |
| | | for (Message msg : list) { |
| | | String result = msg.msgBody; |
| | | PushContentDetailDTO dto = new Gson().fromJson(result, PushContentDetailDTO.class); |
| | | map.put(msg.receiptHandle, dto); |
| | | } |
| | | return map; |
| | | } |
| | | |
| | | /** |
| | | * 删除京东队列 |
| | | * |
| | | * @param receiptHandle |
| | | */ |
| | | public void deleteHWPush(String receiptHandle) { |
| | | if (Constant.TDMQ_CONSUMER) { |
| | | TDMQUtil.getInstance().deleteMsg(PUSH_HW, receiptHandle); |
| | | } else { |
| | | cmqUtil.deleteMsg(PUSH_HW, receiptHandle); |
| | | } |
| | | RabbitmqSenderUtil.sendQueueMsg(rabbitTemplate, PUSH_HW, json); |
| | | } |
| | | } |