服务端接入个推 推送

austin-common/src/main/java/com/java3y/austin/common/dto/model/PushContentModel.java

@@ -1,9 +1,23 @@
 package com.java3y.austin.common.dto.model;
 
 
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
 /**
  * @author 3y
+ *
+ * 通知栏消息推送
  */
+@Data
+@Builder
+@AllArgsConstructor
+@NoArgsConstructor
 public class PushContentModel extends ContentModel {
 
+    private String title;
+    private String content;
+    private String url;
 }

austin-handler/src/main/java/com/java3y/austin/handler/domain/push/PushParam.java

@@ -0,0 +1,33 @@
+package com.java3y.austin.handler.domain.push;
+
+
+import com.java3y.austin.common.domain.TaskInfo;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.Set;
+
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+@Builder
+public class PushParam {
+
+    /**
+     * 调用 接口时需要的token
+     */
+    private String token;
+
+    /**
+     * 调用接口时需要的appId
+     */
+    private String appId;
+
+    /**
+     * 消息模板的信息
+     */
+    private TaskInfo taskInfo;
+
+}

austin-handler/src/main/java/com/java3y/austin/handler/domain/push/getui/BatchSendPushParam.java

@@ -0,0 +1,54 @@
+package com.java3y.austin.handler.domain.push.getui;
+
+import com.alibaba.fastjson.annotation.JSONField;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.Set;
+
+
+/**
+ * 批量推送消息的param
+ *
+ * @author 3y
+ * https://docs.getui.com/getui/server/rest_v2/push/
+ */
+@NoArgsConstructor
+@AllArgsConstructor
+@Data
+@Builder
+public class BatchSendPushParam {
+
+    /**
+     * audience
+     */
+    @JSONField(name = "audience")
+    private AudienceVO audience;
+    /**
+     * taskid
+     */
+    @JSONField(name = "taskid")
+    private String taskId;
+    /**
+     * isAsync
+     */
+    @JSONField(name = "is_async")
+    private Boolean isAsync;
+
+    /**
+     * AudienceVO
+     */
+    @NoArgsConstructor
+    @Data
+    @Builder
+    @AllArgsConstructor
+    public static class AudienceVO {
+        /**
+         * cid
+         */
+        @JSONField(name = "cid")
+        private Set<String> cid;
+    }
+}

austin-handler/src/main/java/com/java3y/austin/handler/domain/push/getui/SendPushParam.java

@@ -1,4 +1,4 @@
-package com.java3y.austin.handler.domain.getui;
+package com.java3y.austin.handler.domain.push.getui;
 
 import com.alibaba.fastjson.annotation.JSONField;
 import lombok.AllArgsConstructor;
@@ -6,7 +6,7 @@ import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
-import java.util.List;
+import java.util.Set;
 
 /**
  * 推送消息的param
@@ -65,7 +65,7 @@ public class SendPushParam {
          * cid
          */
         @JSONField(name = "cid")
-        private List<String> cid;
+        private Set<String> cid;
     }
 
     /**

austin-handler/src/main/java/com/java3y/austin/handler/domain/push/getui/SendPushResult.java

@@ -1,4 +1,4 @@
-package com.java3y.austin.handler.domain.getui;
+package com.java3y.austin.handler.domain.push.getui;
 
 
 import com.alibaba.fastjson.JSONObject;

austin-handler/src/main/java/com/java3y/austin/handler/handler/impl/PushHandler.java

@@ -1,12 +1,32 @@
 package com.java3y.austin.handler.handler.impl;
 
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.IdUtil;
+import cn.hutool.http.ContentType;
+import cn.hutool.http.Header;
+import cn.hutool.http.HttpRequest;
+import com.alibaba.fastjson.JSON;
+import com.google.common.base.Throwables;
+import com.java3y.austin.common.constant.SendAccountConstant;
 import com.java3y.austin.common.domain.TaskInfo;
+import com.java3y.austin.common.dto.account.GeTuiAccount;
+import com.java3y.austin.common.dto.model.PushContentModel;
 import com.java3y.austin.common.enums.ChannelType;
+
+import com.java3y.austin.handler.domain.push.PushParam;
+import com.java3y.austin.handler.domain.push.getui.BatchSendPushParam;
+import com.java3y.austin.handler.domain.push.getui.SendPushParam;
+import com.java3y.austin.handler.domain.push.getui.SendPushResult;
 import com.java3y.austin.handler.handler.BaseHandler;
 import com.java3y.austin.handler.handler.Handler;
+import com.java3y.austin.support.utils.AccountUtils;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.core.StringRedisTemplate;
 import org.springframework.stereotype.Component;
 
+import java.util.Set;
+
 /**
  * 通知栏消息发送处理
  * <p>
@@ -18,13 +38,127 @@ import org.springframework.stereotype.Component;
 @Slf4j
 public class PushHandler extends BaseHandler implements Handler {
 
+    private static final String BASE_URL = "https://restapi.getui.com/v2/";
+    private static final String SINGLE_PUSH_PATH = "/push/single/cid";
+    private static final String BATCH_PUSH_CREATE_TASK_PATH = "/push/list/message";
+    private static final String BATCH_PUSH_PATH = "/push/list/cid";
+
     public PushHandler() {
         channelCode = ChannelType.PUSH.getCode();
     }
 
+    @Autowired
+    private AccountUtils accountUtils;
+    @Autowired
+    private StringRedisTemplate redisTemplate;
+
+
     @Override
     public boolean handler(TaskInfo taskInfo) {
-        return true;
+
+        try {
+            GeTuiAccount account = accountUtils.getAccount(taskInfo.getSendAccount(), SendAccountConstant.GE_TUI_ACCOUNT_KEY, SendAccountConstant.GE_TUI_ACCOUNT_PREFIX, GeTuiAccount.class);
+            String token = redisTemplate.opsForValue().get(SendAccountConstant.GE_TUI_ACCESS_TOKEN_PREFIX + taskInfo.getSendAccount());
+            PushParam pushParam = PushParam.builder().token(token).appId(account.getAppId()).taskInfo(taskInfo).build();
+
+            String result;
+            if (taskInfo.getReceiver().size() == 1) {
+                result = singlePush(pushParam);
+            } else {
+                result = batchPush(createTaskId(pushParam), pushParam);
+            }
+            SendPushResult sendPushResult = JSON.parseObject(result, SendPushResult.class);
+            if (sendPushResult.getCode().equals(0)) {
+                return true;
+            }
+            // 常见的错误 应当 关联至 AnchorState,由austin后台统一透出失败原因
+            log.error("PushHandler#handler fail!result:{},params:{}", JSON.toJSONString(sendPushResult), JSON.toJSONString(taskInfo));
+        } catch (Exception e) {
+            log.error("PushHandler#handler fail!e:{},params:{}", Throwables.getStackTraceAsString(e), JSON.toJSONString(taskInfo));
+        }
+        return false;
+    }
+
+
+    /**
+     * 单推
+     * @param pushParam
+     * @return http result
+     */
+    private String singlePush(PushParam pushParam) {
+        String url = BASE_URL + pushParam.getAppId() + SINGLE_PUSH_PATH;
+        SendPushParam sendPushParam = assembleParam((PushContentModel) pushParam.getTaskInfo().getContentModel(), pushParam.getTaskInfo().getReceiver());
+        String body = HttpRequest.post(url).header(Header.CONTENT_TYPE.getValue(), ContentType.JSON.getValue())
+                .header("token", pushParam.getToken())
+                .body(JSON.toJSONString(sendPushParam))
+                .timeout(2000)
+                .execute().body();
+        return body;
+    }
+
+
+    /**
+     * 批量推送
+     *
+     * @param taskId  个推 返回的任务Id
+     * @param pushParam
+     * @return
+     */
+    private String batchPush(String taskId, PushParam pushParam) {
+        String url = BASE_URL + pushParam.getAppId() + BATCH_PUSH_PATH;
+        BatchSendPushParam batchSendPushParam = BatchSendPushParam.builder()
+                .taskId(taskId)
+                .isAsync(true)
+                .audience(BatchSendPushParam.AudienceVO.builder().cid(pushParam.getTaskInfo().getReceiver()).build()).build();
+        String body = HttpRequest.post(url).header(Header.CONTENT_TYPE.getValue(), ContentType.JSON.getValue())
+                .header("token", pushParam.getToken())
+                .body(JSON.toJSONString(batchSendPushParam))
+                .timeout(2000)
+                .execute().body();
+        return body;
+    }
+
+
+    /**
+     * 群推前需要构建taskId
+     * @param pushParam
+     * @return http result
+     */
+    private String createTaskId(PushParam pushParam) {
+        String url = BASE_URL + pushParam.getAppId() + BATCH_PUSH_CREATE_TASK_PATH;
+        SendPushParam param = assembleParam((PushContentModel) pushParam.getTaskInfo().getContentModel());
+        String taskId = "";
+        try {
+            String body = HttpRequest.post(url).header(Header.CONTENT_TYPE.getValue(), ContentType.JSON.getValue())
+                    .header("token", pushParam.getToken())
+                    .body(JSON.toJSONString(param))
+                    .timeout(2000)
+                    .execute().body();
+
+            taskId = JSON.parseObject(body, SendPushResult.class).getData().getString("taskId");
+        } catch (Exception e) {
+            log.error("PushHandler#createTaskId fail :{},params:{}", Throwables.getStackTraceAsString(e), JSON.toJSONString(pushParam.getTaskInfo()));
+        }
+
+        return taskId;
+    }
+
+
+    private SendPushParam assembleParam(PushContentModel pushContentModel) {
+        return assembleParam(pushContentModel, null);
+    }
+
+    private SendPushParam assembleParam(PushContentModel pushContentModel, Set<String> cid) {
+        SendPushParam param = SendPushParam.builder()
+                .requestId(String.valueOf(IdUtil.getSnowflake().nextId()))
+                .pushMessage(SendPushParam.PushMessageVO.builder().notification(SendPushParam.PushMessageVO.NotificationVO.builder()
+                                .title(pushContentModel.getTitle()).body(pushContentModel.getContent()).clickType("startapp").build())
+                        .build())
+                .build();
+        if (CollUtil.isNotEmpty(cid)) {
+            param.setAudience(SendPushParam.AudienceVO.builder().cid(cid).build());
+        }
+        return param;
     }
 
 }