com.huaweicloud.sermant.premain.common.BootArgsBuilder Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sermant-agentcore-premain Show documentation
Show all versions of sermant-agentcore-premain Show documentation
The starting module for Sermant.
The newest version!
/*
* Copyright (C) 2021-2021 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.huaweicloud.sermant.premain.common;
import com.huaweicloud.sermant.premain.utils.LoggerUtils;
import com.huaweicloud.sermant.premain.utils.StringUtils;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Locale;
import java.util.Map;
import java.util.Properties;
import java.util.logging.Logger;
/**
* 启动参数构建器
*
* @author HapThorin
* @version 1.0.0
* @since 2021-11-12
*/
public abstract class BootArgsBuilder {
private static final Logger LOGGER = LoggerUtils.getLogger();
/**
* 构建启动参数
*
* @param argsMap 入参集
* @param agentPath agent路径
*/
public static void build(Map argsMap, String agentPath) {
final Properties configMap = loadConfig(agentPath);
addNotNullEntries(argsMap, configMap);
addNormalEntries(argsMap, configMap);
addPathEntries(argsMap, agentPath);
}
/**
* 读取启动配置
*
* @param agentPath agent路径
* @return 启动配置
*/
private static Properties loadConfig(String agentPath) {
String realPath = StringUtils.isBlank(agentPath) ? PathDeclarer.getAgentPath() : agentPath;
final Properties properties = new Properties();
try (InputStream configStream = Files.newInputStream(
Paths.get(PathDeclarer.getBootConfigPath(realPath)))) {
properties.load(configStream);
} catch (IOException ioException) {
LOGGER.severe(String.format(Locale.ROOT, "Exception occurs when close config InputStream , %s .",
ioException.getMessage()));
}
return properties;
}
/**
* 通用获取参数值方式，优先从配置获取，在从环境变量获取
*
* @param key 参数键
* @param configMap 配置集
* @return 参数值
*/
private static String getCommonValue(String key, Properties configMap) {
String value = configMap.getProperty(key);
return getActualValue(value);
}
/**
* 添加非空的键值对，涉及的参数均不可为空 appName、serviceName、instanceName、appType
*
* @param argsMap 参数集
* @param configMap 配置集
*/
private static void addNotNullEntries(Map argsMap, Properties configMap) {
String key = BootConstant.ARTIFACT_NAME_KEY;
String defaultValue = "default";
if (!argsMap.containsKey(key)) {
final String value = getCommonValue(key, configMap);
argsMap.put(key, value == null ? defaultValue : value);
}
key = BootConstant.APP_NAME_KEY;
if (!argsMap.containsKey(key)) {
final String value = getCommonValue(key, configMap);
argsMap.put(key, value == null ? defaultValue : value);
}
key = BootConstant.SERVICE_NAME_KEY;
if (!argsMap.containsKey(key)) {
final String value = getCommonValue(key, configMap);
argsMap.put(key, value == null ? defaultValue : value);
}
key = BootConstant.APP_TYPE_KEY;
if (!argsMap.containsKey(key)) {
final String value = getCommonValue(key, configMap);
argsMap.put(key, value == null ? defaultValue : value);
}
}
/**
* 添加普通键值对，为空时不添加
*
* @param argsMap 参数集
* @param configMap 配置集
*/
private static void addNormalEntries(Map argsMap, Properties configMap) {
for (Object key : configMap.keySet()) {
if (!argsMap.containsKey((String) key)) {
final String value = configMap.getProperty((String) key);
if (value != null) {
argsMap.put((String) key, getActualValue(value));
}
}
}
}
/**
* 获取形如"${}"的配置，环境变量或系统变量
*
* @param configVal 配置值
* @return 真实配置
*/
private static String getActualValue(String configVal) {
if (configVal != null && configVal.matches("^.*\\$\\{[\\w.]+(:.*)?}.*$")) {
final int startIndex = configVal.indexOf("${") + 2;
final int endIndex = configVal.indexOf('}', startIndex);
final String envKey = configVal.substring(startIndex, endIndex);
final int separatorIndex = envKey.indexOf(':');
final String key = separatorIndex >= 0 ? envKey.substring(0, separatorIndex) : envKey;
final String defaultValue = separatorIndex >= 0 ? envKey.substring(separatorIndex + 1) : "";
// 优先级为环境变量 > 系统变量
if (!StringUtils.isBlank(System.getenv(key))) {
return System.getenv(key);
}
if (!StringUtils.isBlank(System.getProperty(key))) {
return System.getProperty(key);
}
return defaultValue;
}
return configVal;
}
/**
* 添加路径键值对
*
* @param argsMap 参数集
* @param agentPath agent路径
*/
public static void addPathEntries(Map argsMap, String agentPath) {
// 如果指定的agent路径为空，则通过ProtectionDomain获取
String realPath = StringUtils.isBlank(agentPath) ? PathDeclarer.getAgentPath() : agentPath;
argsMap.put(BootConstant.AGENT_ROOT_DIR_KEY, realPath);
argsMap.put(BootConstant.CORE_IMPLEMENT_DIR_KEY, PathDeclarer.getImplementPath(realPath));
argsMap.put(BootConstant.CORE_CONFIG_FILE_KEY, PathDeclarer.getConfigPath(realPath));
argsMap.put(BootConstant.PLUGIN_SETTING_FILE_KEY, PathDeclarer.getPluginSettingPath(realPath));
argsMap.put(BootConstant.PLUGIN_PACKAGE_DIR_KEY, PathDeclarer.getPluginPackagePath(realPath));
argsMap.put(BootConstant.LOG_SETTING_FILE_KEY, PathDeclarer.getLogbackSettingPath(realPath));
argsMap.put(BootConstant.COMMON_DEPENDENCY_DIR_KEY, PathDeclarer.getCommonLibPath(realPath));
}
}