io.servicecomb.serviceregistry.config.ConfigurePropertyUtils.java Source code

Introduction

Here is the source code for io.servicecomb.serviceregistry.config.ConfigurePropertyUtils.java

Source

/*
 * Copyright 2017 Huawei Technologies Co., Ltd
 *
 * 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 io.servicecomb.serviceregistry.config;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.apache.commons.configuration.Configuration;
import org.apache.commons.lang.StringUtils;

import com.netflix.config.DynamicPropertyFactory;

import io.servicecomb.serviceregistry.api.Const;
import io.servicecomb.serviceregistry.api.registry.BasePath;

public final class ConfigurePropertyUtils {
    private ConfigurePropertyUtils() {
    }

    /**
     * ?key?prefix??
     */
    public static Map<String, String> getPropertiesWithPrefix(String prefix) {
        Object config = DynamicPropertyFactory.getBackingConfigurationSource();
        if (!Configuration.class.isInstance(config)) {
            return new HashMap<>();
        }

        return getPropertiesWithPrefix((Configuration) config, prefix);
    }

    // caller ensure configuration is valid
    public static Map<String, String> getPropertiesWithPrefix(Configuration configuration, String prefix) {
        Map<String, String> propertiesMap = new HashMap<>();

        Iterator<String> keysIterator = configuration.getKeys(prefix);
        while (keysIterator.hasNext()) {
            String key = keysIterator.next();
            propertiesMap.put(key.substring(prefix.length() + 1), String.valueOf(configuration.getProperty(key)));
        }
        return propertiesMap;
    }

    @SuppressWarnings("unchecked")
    public static List<BasePath> getMicroservicePaths(Configuration configuration) {
        List<BasePath> basePaths = new ArrayList<>();
        for (Object path : configuration.getList("service_description.paths")) {
            BasePath basePath = new BasePath();
            Map<String, ?> pathMap = (Map<String, ?>) path;
            basePath.setPath(buildPath((String) pathMap.get("path")));
            basePath.setProperty((Map<String, String>) pathMap.get("property"));
            basePaths.add(basePath);
        }
        return basePaths;
    }

    private static String buildPath(String path) {
        String prefix = System.getProperty(Const.URL_PREFIX);
        if (StringUtils.isNotEmpty(prefix)) {
            if (!path.startsWith(prefix)) {
                path = prefix + path;
            }
        }
        return path;
    }
}