com.huawei.discovery.entity.PlugEffectStrategyCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of springboot-registry-plugin Show documentation
The plugin module of registry from springboot
There is a newer version: 1.4.1
Show newest version
/*
 * Copyright (C) 2022-2022 Huawei Technologies Co., Ltd. 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.huawei.discovery.entity;

import com.huawei.discovery.config.PlugEffectWhiteBlackConstants;

import com.huaweicloud.sermant.core.operation.OperationManager;
import com.huaweicloud.sermant.core.operation.converter.api.YamlConverter;
import com.huaweicloud.sermant.core.service.dynamicconfig.common.DynamicConfigEventType;
import com.huaweicloud.sermant.core.utils.StringUtils;

import java.util.Collections;
import java.util.HashSet;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 插件生效策略缓存
 *
 * @author chengyouling
 * @since 2022-10-08
 */
public enum PlugEffectStrategyCache {
    /**
     * 实例
     */
    INSTANCE;

    private final Map caches = new ConcurrentHashMap<>();

    private final YamlConverter yamlConverter = OperationManager.getOperation(YamlConverter.class);

    /**
     * 当前value对应的服务名集合
     */
    private Set curServices = new HashSet<>();

    /**
     * 将动态配置放入缓存中
     *
     * @param eventType 事件类型
     * @param content 事件内容
     */
    public void resolve(DynamicConfigEventType eventType, String content) {
        final Optional> dataMap = yamlConverter.convert(content, Map.class);
        if (eventType == DynamicConfigEventType.DELETE) {
            caches.clear();
        } else {
            if (dataMap.isPresent()) {
                Map map = dataMap.get();
                final String strategy = map.get(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_STRATEGY);
                if (strategy != null && checkStrategy(map.get(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_STRATEGY))) {
                    caches.put(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_STRATEGY, strategy);
                }
                final String value = map.get(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_VALUE);
                if (value != null) {
                    caches.put(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_VALUE,
                            map.get(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_VALUE));
                }
            } else {
                caches.clear();
            }
        }
        resolveServices();
    }

    /**
     * 解析服务名
     */
    private void resolveServices() {
        final String value = caches.get(PlugEffectWhiteBlackConstants.DYNAMIC_CONFIG_VALUE);
        if (value == null) {
            curServices = Collections.emptySet();
            return;
        }
        final String[] parts = value.split(",");
        final HashSet services = new HashSet<>();
        for (String part : parts) {
            if (part != null) {
                services.add(part.trim());
            }
        }
        curServices = services;
    }

    public Set getCurServices() {
        return curServices;
    }

    private boolean checkStrategy(String strategy) {
        return StringUtils.equalsIgnoreCase(strategy, PlugEffectWhiteBlackConstants.STRATEGY_ALL)
                || StringUtils.equalsIgnoreCase(strategy, PlugEffectWhiteBlackConstants.STRATEGY_NONE)
                || StringUtils.equalsIgnoreCase(strategy, PlugEffectWhiteBlackConstants.STRATEGY_WHITE)
                || StringUtils.equalsIgnoreCase(strategy, PlugEffectWhiteBlackConstants.STRATEGY_BLACK);
    }

    /**
     * 获取对应key的配置
     *
     * @param key 键
     * @return 动态配置值
     */
    public String getConfigContent(String key) {
        return caches.get(key);
    }
}