钉钉报警接入代码

@Service@Slf4jpublic class DingTalkUtil { @Value("${dingTalk.robot.url}") private String robotUrl; @Value("${dingTalk.robot.me}") private String me; // 钉钉密钥 @Value("${dingTalk.robot.secret}") private String secret; @Value("${dingTalk.enabled}") private Boolean enabled; private OkHttpClient okHttpClient; private static final ObjectMapper objectMapper = new ObjectMapper(); private static final MediaType jsonMediaType = MediaType.parse("application/json"); @PostConstruct public void init() { ExecutorService executorService = new ThreadPoolExecutor( 1, 5, 1, TimeUnit.MINUTES, new ArrayBlockingQueue<>(100), ThreadFactoryBuilder.create().setNamePrefix("dingTalk-").build(), new ThreadPoolExecutor.CallerRunsPolicy() ); Dispatcher dispatcher = new Dispatcher(executorService); dispatcher.setMaxRequests(5); dispatcher.setMaxRequestsPerHost(5); okHttpClient = new OkHttpClient.Builder() .readTimeout(Duration.ofSeconds(1)) .connectTimeout(Duration.ofSeconds(1)) .callTimeout(Duration.ofSeconds(1)) .writeTimeout(Duration.ofSeconds(1)) .dispatcher(dispatcher) .build(); } /** * 异步发送钉钉机器人文本消息. */ public void sendTextMessage(String content) { doSendTextMessage(content, textMessage -> { }); } /** * 异步发送文本消息并@自己. */ public void sendTextMessageWithAtMe(String content) { doSendTextMessage(content, textMessage -> textMessage.getAt().getAtMobiles().add(me)); } /** * 异步发送文本消息并@所有人. */ public void sendTextMessageWithAtAll(String content) { doSendTextMessage(content, textMessage -> textMessage.getAt().setAtAll(true)); } private void doSendTextMessage(String content, Consumer<TextMessage> messageConfigurator) { if (!enabled) { return; } if (StringUtils.isBlank(content)) { throw new IllegalArgumentException("文本消息内容不能为空"); } TextMessage textMessage = new TextMessage(); textMessage.setText(new TextMessage.Content(content)); messageConfigurator.accept(textMessage); long timestamp = System.currentTimeMillis(); String sign = sign(timestamp); try { Request request = new Request.Builder() .url((robotUrl + "×tamp=" + timestamp + "&sign=" + sign)) .post(RequestBody.create(objectMapper.writeValueAsString(textMessage), jsonMediaType)) .build(); Call call = okHttpClient.newCall(request); call.enqueue(new Callback() { @Override public void onFailure(@NotNull Call call, @NotNull IOException e) { log.error("发送钉钉消息失败, 请求: {}.", call, e); } @Override public void onResponse(@NotNull Call call, @NotNull Response response) { ResponseBody responseBody = response.body(); log.debug("钉钉发送成功, call: {}, resp: {}.", call.request().body(), responseBody); if (responseBody != null) responseBody.close(); } }); } catch (JsonProcessingException e) { throw ExceptionUtil.wrapRuntime(e); } } private String sign(long timestamp) { final String seed = (timestamp + "\n" + secret); try { Mac mac = Mac.getInstance("HmacSHA256"); mac.init(new SecretKeySpec(secret.getBytes(StandardCharsets.UTF_8), "HmacSHA256")); byte[] result = mac.doFinal(seed.getBytes(StandardCharsets.UTF_8)); return URLEncoder.encode(Base64.getEncoder().encodeToString(result), StandardCharsets.UTF_8.displayName()); } catch (NoSuchAlgorithmException | InvalidKeyException | UnsupportedEncodingException e) { throw ExceptionUtil.wrapRuntime(e); } } @Getter private static class TextMessage { private final String msgtype = "text"; @Setter private Content text; private final At at = new At(); @Data @AllArgsConstructor private static class Content { private String content; } private static class At { @Setter private boolean isAtAll = false; @Getter private final List<String> atMobiles = new LinkedList<>(); // 不能删除,否则会导致生成的json字段名是atAll, 导致@所有人不生效 public boolean getIsAtAll() { return isAtAll; } } }}

@Service@Slf4jpublic class DingTalkUtil {    @Value("${dingTalk.robot.url}")    private String robotUrl;    @Value("${dingTalk.robot.me}")    private String me;    // 钉钉密钥    @Value("${dingTalk.robot.secret}")    private String secret;    @Value("${dingTalk.enabled}")    private Boolean enabled;    private OkHttpClient okHttpClient;    private static final ObjectMapper objectMapper = new ObjectMapper();    private static final MediaType jsonMediaType = MediaType.parse("application/json");    @PostConstruct    public void init() {        ExecutorService executorService = new ThreadPoolExecutor(                1,                5,                1,                TimeUnit.MINUTES,                new ArrayBlockingQueue<>(100),                ThreadFactoryBuilder.create().setNamePrefix("dingTalk-").build(),                new ThreadPoolExecutor.CallerRunsPolicy()        );        Dispatcher dispatcher = new Dispatcher(executorService);        dispatcher.setMaxRequests(5);        dispatcher.setMaxRequestsPerHost(5);        okHttpClient = new OkHttpClient.Builder()                .readTimeout(Duration.ofSeconds(1))                .connectTimeout(Duration.ofSeconds(1))                .callTimeout(Duration.ofSeconds(1))                .writeTimeout(Duration.ofSeconds(1))                .dispatcher(dispatcher)                .build();    }    /**     * 异步发送钉钉机器人文本消息.     */    public void sendTextMessage(String content) {        doSendTextMessage(content, textMessage -> {        });    }    /**     * 异步发送文本消息并@自己.     */    public void sendTextMessageWithAtMe(String content) {        doSendTextMessage(content, textMessage -> textMessage.getAt().getAtMobiles().add(me));    }    /**     * 异步发送文本消息并@所有人.     */    public void sendTextMessageWithAtAll(String content) {        doSendTextMessage(content, textMessage -> textMessage.getAt().setAtAll(true));    }    private void doSendTextMessage(String content, Consumer<TextMessage> messageConfigurator) {        if (!enabled) {            return;        }        if (StringUtils.isBlank(content)) {            throw new IllegalArgumentException("文本消息内容不能为空");        }        TextMessage textMessage = new TextMessage();        textMessage.setText(new TextMessage.Content(content));        messageConfigurator.accept(textMessage);        long timestamp = System.currentTimeMillis();        String sign = sign(timestamp);        try {            Request request = new Request.Builder()                    .url((robotUrl + "×tamp=" + timestamp + "&sign=" + sign))                    .post(RequestBody.create(objectMapper.writeValueAsString(textMessage), jsonMediaType))                    .build();            Call call = okHttpClient.newCall(request);            call.enqueue(new Callback() {                @Override                public void onFailure(@NotNull Call call, @NotNull IOException e) {                    log.error("发送钉钉消息失败, 请求: {}.", call, e);                }                @Override                public void onResponse(@NotNull Call call, @NotNull Response response) {                    ResponseBody responseBody = response.body();                    log.debug("钉钉发送成功, call: {}, resp: {}.", call.request().body(), responseBody);                    if (responseBody != null) responseBody.close();                }            });        } catch (JsonProcessingException e) {            throw ExceptionUtil.wrapRuntime(e);        }    }    private String sign(long timestamp) {        final String seed = (timestamp + "\n" + secret);        try {            Mac mac = Mac.getInstance("HmacSHA256");            mac.init(new SecretKeySpec(secret.getBytes(StandardCharsets.UTF_8), "HmacSHA256"));            byte[] result = mac.doFinal(seed.getBytes(StandardCharsets.UTF_8));            return URLEncoder.encode(Base64.getEncoder().encodeToString(result), StandardCharsets.UTF_8.displayName());        } catch (NoSuchAlgorithmException | InvalidKeyException | UnsupportedEncodingException e) {            throw ExceptionUtil.wrapRuntime(e);        }    }    @Getter    private static class TextMessage {        private final String msgtype = "text";        @Setter        private Content text;        private final At at = new At();        @Data        @AllArgsConstructor        private static class Content {            private String content;        }        private static class At {            @Setter            private boolean isAtAll = false;            @Getter            private final List<String> atMobiles = new LinkedList<>();            // 不能删除,否则会导致生成的json字段名是atAll, 导致@所有人不生效            public boolean getIsAtAll() {                return isAtAll;            }        }    }}

内容来源网络,如有侵权,联系删除,本文地址:https://www.230890.com/zhan/100298.html

(0)

相关推荐