java编辑yml_java工具类11-读取yml配置文件

importcom.alibaba.fastjson.JSON;importcom.google.common.base.Preconditions;importcom..common.constant.CommonConstant;importcom..common.entity.EsConfigEntity;importcom..common.entity.KafkaConfigEntity;importcom..common.tool.ConfigUtil;importcom..common.tool.FileUtil;importjava.io.File;importjava.io.FileInputStream;importjava.io.IOException;importjava.util.Arrays;importjava.util.List;importjava.util.Map;importjava.util.Optional;importjava.util.Properties;importjava.util.function.Function;importjava.util.function.Predicate;importorg.apache.commons.lang3.StringUtils;importorg.slf4j.Logger;importorg.slf4j.LoggerFactory;public classProjectConfig {private static final Logger log = LoggerFactory.getLogger(ProjectConfig.class);private static final String SPARK_BATCH_CONF_FILE = "spark-batch.properties";private static final String SPARK_STREAMING_CONF_FILE = "spark-streaming.properties";private static final String JOB_CONF_FILE = "job.yaml";private static final String SERVICE_CONF_FILE = "service.yaml";private static final String COMMON_CONF_FILE = "application.yaml";private static Map sparkBatchConf = loadConfig("spark-batch.properties");private static Map sparkStreamingConf = loadConfig("spark-streaming.properties");publicProjectConfig() {

}public staticString getProjectHome() {returnSystem.getenv(CommonConstant.PROJECT_HOME);

}public staticString getProjectConfDir() {return System.getenv(CommonConstant.PROJECT_HOME) + File.separator + "conf";

}public staticString getDpsJarFile() {return System.getenv(CommonConstant.PROJECT_HOME) + File.separator + "jars" + File.separator + "libra-dps.jar";

}public staticString getCommonConfFile() {return getProjectConfDir() + File.separator + "application.yaml";

}public static MapgetSparkBatchConf() {returnsparkBatchConf;

}public static MapgetSparkStreamingConf() {returnsparkStreamingConf;

}public staticEsConfigEntity getESConf() {return (EsConfigEntity)JSON.parseObject(JSON.toJSONString(ConfigUtil.readYamlByPrefix("application.yaml", "es")), EsConfigEntity.class);

}public staticKafkaConfigEntity getKafkaConf() {return (KafkaConfigEntity)JSON.parseObject(JSON.toJSONString(ConfigUtil.readYamlByPrefix("application.yaml", "kafka")), KafkaConfigEntity.class);

}public staticString getIndexLimit() {return ConfigUtil.readYamlByPrefix("application.yaml", "es").get("splitLimitNum").toString();

}public staticObject getCommonValue(String key) {return ConfigUtil.readYaml("application.yaml").get(key);

}public static MapgetRedisConf() {return ConfigUtil.readYamlByPrefix("application.yaml", "redis");

}public staticObject getStreamingJobValue(String key) {return ConfigUtil.readYaml("job.yaml").get(key);

}public staticMap getStreamingJobConf() {return ConfigUtil.readYaml("job.yaml");

}public staticMap getStreamingServiceConf() {return ConfigUtil.readYaml("service.yaml");

}public static ListgetProtocols() {return Arrays.asList(ConfigUtil.readYaml("application.yaml").get("protocol").toString().split(","));

}public static OptionalgetEsMapping(String indexName) {

Preconditions.checkArgument(StringUtils.isNotBlank(indexName));

File mappingDir= new File(getProjectConfDir() + File.separator + "mapping");return mappingDir.isDirectory() ? (Optional)Arrays.asList(mappingDir.listFiles()).stream().filter((file) ->{return indexName.equals(file.getName().split("\\.")[0]);

}).map((file)->{return FileUtil.fileToString(mappingDir + File.separator +file.getName());

}).findAny().get() : FileUtil.fileToString(indexName+ ".json");

}public staticMap loadConfig(String fileName) {

Preconditions.checkArgument(StringUtils.isNotBlank(fileName));

String confPath=getProjectConfDir();

File file= newFile(confPath.concat(File.separator).concat(fileName));if (!file.exists()) {

file= newFile(fileName);

}

Preconditions.checkArgument(file.exists(),"未找到配置文件");

Properties properties= newProperties();try{

properties.load(newFileInputStream(file));returnproperties;

}catch(IOException var5) {

log.error("读取配置文件异常：{}/{}", newObject[]{confPath, fileName, var5});throwExceptionUtil.buildRuntimeException();

}

}

}