WangHan
2024-09-12 d5855a4926926698b740bc6c7ba489de47adb68b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package tech.powerjob.server.remote.worker.utils;
 
import com.google.common.collect.Sets;
import org.apache.commons.lang3.StringUtils;
import tech.powerjob.server.common.SJ;
import tech.powerjob.server.common.module.WorkerInfo;
 
import java.util.Optional;
import java.util.Set;
 
/**
 * 指定工具
 *
 * @author tjq
 * @since 2024/2/24
 */
public class SpecifyUtils {
 
    private static final String TAG_EQUALS = "tagEquals:";
 
    private static final String TAG_IN = "tagIn:";
 
    public static boolean match(WorkerInfo workerInfo, String specifyInfo) {
 
        String workerTag = workerInfo.getTag();
 
        // tagIn 语法,worker 可上报多个tag,如 WorkerInfo#tag=tag1,tag2,tag3,配置中指定 tagIn=tag1 即可命中
        if (specifyInfo.startsWith(TAG_IN)) {
            String targetTag = specifyInfo.replace(TAG_IN, StringUtils.EMPTY);
            return Optional.ofNullable(workerTag).orElse(StringUtils.EMPTY).contains(targetTag);
        }
 
        // tagEquals 语法,字符串完全匹配,worker 只可上报一个 tag,如 WorkerInfo#tag=tag1,配置中指定 tagEquals=tag1 即可命中
        if (specifyInfo.startsWith(TAG_EQUALS)) {
            String targetTag = specifyInfo.replace(TAG_EQUALS, StringUtils.EMPTY);
            return Optional.ofNullable(workerTag).orElse(StringUtils.EMPTY).equals(targetTag);
        }
 
        // 默认情况,IP 和 tag 逗号分割后任意完全匹配即视为命中(兼容 4.3.8 版本前序逻辑)
        Set<String> designatedWorkersSet = Sets.newHashSet(SJ.COMMA_SPLITTER.splitToList(specifyInfo));
 
        for (String tagOrAddress : designatedWorkersSet) {
            if (tagOrAddress.equals(workerInfo.getTag()) || tagOrAddress.equals(workerInfo.getAddress())) {
                return true;
            }
        }
 
        return false;
    }
 
}