com.github.jaemon.dinger.multi.algorithm.DingerHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dinger-spring-boot-starter Show documentation
Dinger-SpringBoot集成钉钉/企业微信群机器人实现消息通知中间件
The newest version!
/*
 * Copyright ©2015-2023 Jaemon. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.github.jaemon.dinger.multi.algorithm;

import com.github.jaemon.dinger.core.DingerConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Dinger钉钉机器人算法 

 *
 * 
 *     VM options: {@code -Dmulti.dinger.minute.limit.count=2}
 * 
 *
 * 
 *     机器人发送消息频率限制
 *      *         https://ding-doc.dingtalk.com/doc#/serverapi3/pghqkk
 *     
 * 
 *
 * @author Jaemon
 * @since 1.0
 */
public class DingerHandler implements AlgorithmHandler {
    private static final Logger log = LoggerFactory.getLogger(DingerHandler.class);
    public static final String DINGTALK_MULTI_DINGER_COUNT = "multi.dinger.minute.limit.count";
    private static final DateTimeFormatter DATETIME_FMT = DateTimeFormatter.ofPattern("yyMMddHHmm");
    private static final ZoneId ZONE_ID = ZoneId.of("Asia/Shanghai");

    /** 每分钟内限制发送 COUNT 条 */
    private static final int COUNT_THRESHOLD;

    static {
        COUNT_THRESHOLD = System.getProperty(DINGTALK_MULTI_DINGER_COUNT) == null ?
                20 : Integer.parseInt(
                        System.getProperty(DINGTALK_MULTI_DINGER_COUNT)
        );
    }

    /** 索引号 */
    private volatile int index = DEFAULT_INDEX;
    /** 当前分钟 */
    private String currentMinite = null;
    /** 计数器 */
    private AtomicInteger counter = new AtomicInteger(DEFAULT_INDEX);


    @Override
    public DingerConfig handler(List dingerConfigs, DingerConfig defaultDingerConfig) {
        int size = dingerConfigs.size();

        synchronized (this) {
            if (currentMinite == null) {
                currentMinite = LocalDateTime.now(ZONE_ID).format(DATETIME_FMT);
            }

            int count = counter.getAndIncrement();

            boolean countBool = count >= COUNT_THRESHOLD;
            String now = LocalDateTime.now(ZONE_ID).format(DATETIME_FMT);
            boolean inMinute = now.equals(currentMinite);
            if (countBool) {
                if (inMinute) {
                    index++;
                    index = index >= size ? DEFAULT_INDEX : index;
                }
                currentMinite = now;
                counter.set(1);

                if (log.isDebugEnabled()) {
                    log.debug("#{}-{}# 在{}分内发送了{}次, 当前分钟={}, 下一个机器人={}.",
                            algorithmId(), COUNT_THRESHOLD, currentMinite, count, now, index);
                }

//                System.out.println(String.format("#%s-%d# 在%s分内发送了%d次, 当前分钟=%s, 下一个机器人=%d.",
//                        algorithmId(), COUNT_THRESHOLD, currentMinite, count, now, index));
            } else if (!countBool && !inMinute) {
                currentMinite = now;
                counter.set(1);

                if (log.isDebugEnabled()) {
                    log.debug("#{}-{}# 在{}分内发送了{}次, 当前分钟={}, 当前机器人={}.",
                            algorithmId(), COUNT_THRESHOLD, currentMinite, count, now, index);
                }
//                System.out.println(String.format("#%s-%d# 在%s分内发送了%d次, 当前分钟=%s, 当前机器人=%d.",
//                        algorithmId(), COUNT_THRESHOLD, currentMinite, count, now, index));
            }

        }

        return dingerConfigs.get(index);
    }
}