如何基于JAVA读取yml配置文件指定key内容

 更新时间:2020年01月13日 10:56:02   作者:流逝、  
这篇文章主要介绍了如何基于JAVA读取yml配置文件指定key内容,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

这篇文章主要介绍了如何基于JAVA读取yml配置文件指定key内容,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

先引入需要的依赖

<!--读取yml文件-->
    <dependency>
      <groupId>org.yaml</groupId>
      <artifactId>snakeyaml</artifactId>
      <version>1.23</version>
    </dependency>

读取YML文件工具类的代码

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.ResourceUtils;
import org.yaml.snakeyaml.Yaml;

import java.io.*;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * @author hunmeng
 * @create 2020-01-10 20:34
 */
public class YmlUtils {

  private static final Logger LOGGER = LoggerFactory.getLogger(YmlUtils.class);

  private static String bootstrap_file = "classpath:application-test.yml";

  private static Map<String,String> result = new HashMap<>();

  /**
   * 根据文件名获取yml的文件内容
   * @param filePath
   * @param keys 第一个参数对应第一个key,第二个参数对应第二个key 比如spring.name下的所有 就是两个参数、
   *       getYmlByFileName(bootstrap_file,"spring", "name");
   * @return
   */
  public static Map<String,String> getYmlByFileName(String filePath, String... keys) {
    result = new HashMap<>();
    if(filePath == null) filePath = bootstrap_file;
    InputStream in = null;
    try {
      File file = ResourceUtils.getFile(filePath);
      in = new BufferedInputStream(new FileInputStream(file));
      Yaml props = new Yaml();
      Object obj = props.loadAs(in,Map.class);
      Map<String,Object> param = (Map<String, Object>) obj;

      for(Map.Entry<String,Object> entry:param.entrySet()){
        String key = entry.getKey();
        Object val = entry.getValue();
        if (keys.length != 0 && !keys[0].equals(key)){
          continue;
        }
        if(val instanceof Map){
          forEachYaml(key,(Map<String, Object>) val, 1, keys);
        }else{
          result.put(key, val.toString());
        }
      }
      return result;
    } catch (FileNotFoundException e) {
      LOGGER.error(e.getMessage(),e);
    }finally {
      if (in != null){
        try {
          in.close();
        } catch (IOException e) {
          LOGGER.error(e.getMessage(),e);
        }
      }
    }
    return null;
  }

  /**
   * 根据key获取值
   * @param key
   * @return
   */
  public static String getValue(String key) throws FileNotFoundException {
    Map<String,String> map = getYmlByFileName(null);
    if(map==null)return null;
    return map.get(key);
  }


  /**
   * 遍历yml文件,获取map集合
   * @param key_str
   * @param obj
   * @param i
   * @param keys
   * @return
   */
  public static Map<String,String> forEachYaml(String key_str,Map<String, Object> obj, int i, String... keys){
    for(Map.Entry<String,Object> entry:obj.entrySet()){
      String key = entry.getKey();
      Object val = entry.getValue();
      if (keys.length > i && !keys[i].equals(key)){
        continue;
      }
      String str_new = "";
      if(StringUtils.isNotEmpty(key_str)){
        str_new = key_str+ "."+key;
      }else{
        str_new = key;
      }
      if(val instanceof Map){
        forEachYaml(str_new,(Map<String, Object>) val, ++i, keys);
        i--;
      }else{

        result.put(str_new,val.toString());
      }
    }

    return result;
  }


  /**
   * 获取bootstrap.yml的name
   * @return
   */
  public static String getApplicationName() throws FileNotFoundException {
    return getYmlByFileName(bootstrap_file).get("server.port");
  }

  /**
   * 获取bootstrap.yml的name
   * @return
   */
  public static String getApplicationName1() throws FileNotFoundException {
    String name = getYmlByFileName(bootstrap_file).get("spring.application.name");
    return name + "center";
  }

  public static void main(String[] args) throws FileNotFoundException {

    Map<String, String> ymlByFileName = getYmlByFileName(bootstrap_file,"spring");
    Set<Map.Entry<String, String>> entries = ymlByFileName.entrySet();
    for (Map.Entry<String, String> entry : entries) {
      System.out.println(entry.getKey()+"==="+entry.getValue());
    }

    System.out.println(getApplicationName());
  }
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

相关文章

  • 编程入门:掌握Java运算符技巧

    编程入门:掌握Java运算符技巧

    掌握Java运算符技巧,能让你的编程之旅轻松许多,本指南将带你深入了解如何巧妙地使用这些强大的工具,让代码不仅高效,还充满乐趣,跟着我们一起,让你的Java代码在运算符的魔法下焕发新生!
    2023-12-12
  • SpringBoot项目中只执行一次的任务写法实现

    SpringBoot项目中只执行一次的任务写法实现

    有时候我们需要进行初始化工作,就说明只要进行一次的工作,本文主要介绍了SpringBoot项目中只执行一次的任务写法实现,感兴趣的可以了解一下
    2023-12-12
  • Java编写中容易搞错的一些东西

    Java编写中容易搞错的一些东西

    Java编写中容易搞错的一些东西...
    2006-12-12
  • 详解如何使用MongoDB+Springboot实现分布式ID的方法

    详解如何使用MongoDB+Springboot实现分布式ID的方法

    这篇文章主要介绍了详解如何使用MongoDB+Springboot实现分布式ID的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-09-09
  • Java实现文件夹中内容定时删除

    Java实现文件夹中内容定时删除

    这篇文章主要为大家详细介绍了Java实现文件夹中内容定时删除,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2021-08-08
  • Springcloud整合stream,rabbitmq实现消息驱动功能

    Springcloud整合stream,rabbitmq实现消息驱动功能

    官方定义SpringCloud Stream 是一个构建消息驱动微服务的框架。我们只需要搞清楚如何与Spring Cloud Stream 交互就可以方便使用消息驱动的方式。本文将通过Springcloud整合stream,rabbitmq实现消息驱动功能,需要的可以参考一下
    2022-02-02
  • SpringBoot中的Spring Cloud Hystrix原理和用法详解

    SpringBoot中的Spring Cloud Hystrix原理和用法详解

    在Spring Cloud中,Hystrix是一个非常重要的组件,Hystrix可以帮助我们构建具有韧性的分布式系统,保证系统的可用性和稳定性,在本文中,我们将介绍SpringBoot中的Hystrix,包括其原理和如何使用,需要的朋友可以参考下
    2023-07-07
  • Spring启动指定时区的两种方法

    Spring启动指定时区的两种方法

    最近项目启动,时间要修改成东七区时间,本文主要介绍了Spring启动指定时区的两种方法,具有一定的参考价值,感兴趣的可以了解一下
    2023-11-11
  • 浅析Java集合及LIst接口

    浅析Java集合及LIst接口

    这篇文章主要介绍了Java集合及LIst接口,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-03-03
  • java定义受限制的类型参数操作

    java定义受限制的类型参数操作

    这篇文章主要介绍了java定义受限制的类型参数操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-08-08

最新评论