Java中实时读取yaml文件

注意此种方式在修改yml文件之后就会立即读取到新的值,不需要重启服务,但不重启会导致后台和前台拿到的数据不一致,谨慎使用

引入snakeyaml

<dependency>
     <groupId>org.yaml</groupId>
     <artifactId>snakeyaml</artifactId>
     <version>1.23</version>
</dependency>

application.yml

spring:
    profiles:
        #环境
        active: dev

application-dev.yml

#微信公众号 (测试号)
wechat:
  appId: wxh123242hh3h25

读取yaml工具类

package com.ainsg.apps.aop.common.utils;

import org.apache.ibatis.io.Resources;
import org.yaml.snakeyaml.Yaml;

import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

/**
 * 读取 yml 工具类
 *
 * @author: ChenRuiYao
 * @since: 2020-07-30 09:35
 * @Version: 1.0
 */
public class YamlUtil {

  /**
   * 读取 ymal
   *
   * @param path 需要读取的文件路径
   * @return Map
   */
  public static Map getReadAbleYaml(String path) {
    Yaml yaml = new Yaml();
    Map obj = null;
    InputStream ins = null;
    try {
      ins = Resources.getResourceAsStream(path);
      obj = (Map) yaml.load(ins);
    } catch (Exception e) {
      e.printStackTrace();
    } finally {
      try {
        if (ins != null) {
          ins.close();
        }
      } catch (IOException e) {
        e.printStackTrace();
      }
    }
    return obj;
  }
}

应用

package com.ainsg.apps.aop.modules.sys.controller;

import com.ainsg.apps.aop.common.utils.YamlUtil;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;

@RestController
@RequestMapping("/yaml")
public class YamlController {
  /**
   * 获取对应环境中的 appid (wechat.appId)
   *
   * @author: ChenRuiYao
   * @since: 2020/7/30  16:04
   */
  @GetMapping("/getAppId")
  @ResponseBody
  public String getAppId() {
    String active = getEnv();
    Map env = YamlUtil.getReadAbleYaml("application-" + active + ".yml");
    Map wechat = (Map) env.get("wechat");
    String appId = wechat.get("appId").toString();
    return appId;
  }

  /**
   * 获取 application.yml 中的环境信息 (spring.profiles.active)
   *
   * @author: ChenRuiYao
   * @since: 2020/7/30  16:04
   */
  @GetMapping("/getEnv")
  @ResponseBody
  public String getEnv() {
    Map properties = YamlUtil.getReadAbleYaml("application.yml");
    Map spring = (Map) properties.get("spring");
    Map profiles = (Map) spring.get("profiles");
    String active = profiles.get("active").toString();
    return active;
  }
}
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Java本身没有直接读取YAML文件包括注释的API,但可以使用第三方库SnakeYAML实现。 以下是一个示例代码,使用SnakeYAML读取YAML文件内容包括注释: ```java import org.yaml.snakeyaml.Yaml; import org.yaml.snakeyaml.constructor.Constructor; import org.yaml.snakeyaml.nodes.Node; import org.yaml.snakeyaml.nodes.Tag; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.InputStream; import java.util.LinkedHashMap; import java.util.Map; public class YamlReader { public static void main(String[] args) throws FileNotFoundException { Yaml yaml = new Yaml(new CommentedConstructor()); InputStream inputStream = new FileInputStream("config.yml"); Map<String, Object> obj = yaml.load(inputStream); System.out.println(obj); } } class CommentedConstructor extends Constructor { public CommentedConstructor() { this.yamlConstructors.put(null, new CommentedConstruct()); } private class CommentedConstruct extends Constructor.ConstructYamlMap { @Override public Object construct(Node node) { Object obj = super.construct(node); if (node instanceof org.yaml.snakeyaml.nodes.MappingNode) { org.yaml.snakeyaml.nodes.MappingNode mnode = (org.yaml.snakeyaml.nodes.MappingNode) node; Map<String, Object> map = (Map<String, Object>) obj; for (org.yaml.snakeyaml.nodes.NodeTuple tuple : mnode.getValue()) { org.yaml.snakeyaml.nodes.Node keyNode = tuple.getKeyNode(); if (keyNode instanceof org.yaml.snakeyaml.nodes.ScalarNode) { org.yaml.snakeyaml.nodes.ScalarNode scalarNode = (org.yaml.snakeyaml.nodes.ScalarNode) keyNode; String key = scalarNode.getValue(); map.put(key, tuple.getValueNode().getRawValue()); // 将注释保存到map if (tuple.getValueNode().getTag().equals(Tag.COMMENT)) { map.put("#" + key, tuple.getValueNode().getValue()); } } } } return obj; } } } ``` 这里使用了CommentedConstructor继承自SnakeYAML的Constructor类,重写了构造函数,并在其对节点进行了处理。使用时只需要将YAML文件的路径修改为自己的文件路径即可。 需要注意的是,这里的注释是以键值对的方式保存在Map的,键值为"#"+原键名。因此在获取键值对时需要注意。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值