溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點(diǎn)擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

在Spring Boot項(xiàng)目中如何實(shí)現(xiàn)自定義PropertySourceLoader

發(fā)布時(shí)間：2020-11-18 16:24:44 來源：億速云 閱讀：256 作者：Leah 欄目：編程語言

今天就跟大家聊聊有關(guān)在Spring Boot項(xiàng)目中如何實(shí)現(xiàn)自定義PropertySourceLoader，可能很多人都不太了解，為了讓大家更加了解，小編給大家總結(jié)了以下內(nèi)容，希望大家根據(jù)這篇文章可以有所收獲。

SpringBoot 的配置文件內(nèi)置支持 properties、xml、yml、yaml 幾種格式，其中 properties和xml 對應(yīng)的Loader類為 PropertiesPropertySourceLoader ，yml和yaml 對應(yīng)的Loader類為 YamlPropertySourceLoader。

觀察這2個(gè)類可以發(fā)現(xiàn)，都實(shí)現(xiàn)自接口 PropertySourceLoader 。所以我們要新增支持別的格式的配置文件，就可以通過實(shí)現(xiàn)接口 PropertySourceLoader 來實(shí)現(xiàn)了。

下面實(shí)現(xiàn)了一個(gè) json 格式的配置文件 Loader類：

package com.shanhy.sboot.property;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.springframework.boot.env.PropertySourceLoader;
import org.springframework.boot.json.JsonParser;
import org.springframework.boot.json.JsonParserFactory;
import org.springframework.core.env.MapPropertySource;
import org.springframework.core.env.PropertySource;
import org.springframework.core.io.Resource;

/**
 * JSON格式配置文件加載器
 * 
 * @author 單紅宇(CSDN CATOOP)
 * @create 2017年4月20日
 */
public class JsonPropertySourceLoader implements PropertySourceLoader {

  public String[] getFileExtensions() {
    // 配置文件格式（擴(kuò)展名）
    return new String[] { "json" };
  }

  public PropertySource<&#63;> load(String name, Resource resource, String profile) throws IOException {
    // 處理機(jī)制參考PropertiesPropertySourceLoader
    // 無論profile有沒有值，底層都會嘗試先執(zhí)行 load(String name, Resource resource, null)，所以這個(gè)地方之間判斷等于null即可。
    // 當(dāng)前版本springboot-1.5.2（后續(xù)版本未知）詳見 ConfigFileApplicationListener 的 445 行
    if (profile == null) {
      Map<String, Object> result = mapPropertySource(resource);
      return new MapPropertySource(name, result);
    }
    return null;
  }

  /**
   * 解析Resource為Map
   *
   * @param resource
   * @return
   * @throws IOException
   * 
   * @author 單紅宇(CSDN CATOOP)
   * @create 2017年4月20日
   */
  private Map<String, Object> mapPropertySource(Resource resource) throws IOException {
    if (resource == null) {
      return null;
    }
    Map<String, Object> result = new HashMap<String, Object>();
    JsonParser parser = JsonParserFactory.getJsonParser();
    Map<String, Object> map = parser.parseMap(readFile(resource));
    nestMap("", result, map);
    return result;
  }

  /**
   * 讀取Resource文件內(nèi)容為字符串
   *
   * @param resource
   * @return
   * @throws IOException
   * 
   * @author 單紅宇(CSDN CATOOP)
   * @create 2017年4月20日
   */
  private String readFile(Resource resource) throws IOException {
    InputStream inputStream = resource.getInputStream();
    List<Byte> byteList = new LinkedList<Byte>();
    byte[] readByte = new byte[1024];
    int length;
    while ((length = inputStream.read(readByte)) > 0) {
      for (int i = 0; i < length; i++) {
        byteList.add(readByte[i]);
      }
    }
    byte[] allBytes = new byte[byteList.size()];
    int index = 0;
    for (Byte soloByte : byteList) {
      allBytes[index] = soloByte;
      index += 1;
    }
    return new String(allBytes, "UTF-8");
  }

  /**
   * 處理map（map中可能還嵌套map，遞歸處理），最終輸出一個(gè)非嵌套的map
   *
   * @param prefix
   *      前綴
   * @param result
   *      處理后的map
   * @param map
   *      處理前的map
   * 
   * @author 單紅宇(CSDN CATOOP)
   * @create 2017年4月20日
   */
  @SuppressWarnings("unchecked")
  private void nestMap(String prefix, Map<String, Object> result, Map<String, Object> map) {
    if (prefix.length() > 0) {
      prefix += ".";
    }
    for (Map.Entry<String, Object> entrySet : map.entrySet()) {
      if (entrySet.getValue() instanceof Map) {
        nestMap(prefix + entrySet.getKey(), result, (Map<String, Object>) entrySet.getValue());
      } else {
        result.put(prefix + entrySet.getKey().toString(), entrySet.getValue());
      }
    }
  }
}

然后在 src/main/resources 中創(chuàng)建 META-INF/spring.factories 文件，內(nèi)容為：

org.springframework.boot.env.PropertySourceLoader=\
com.shanhy.sboot.property.JsonPropertySourceLoader

創(chuàng)建測試的配置文件 application.json

{
  "custom": {
    "property": {
      "message": "測試數(shù)據(jù)"
    }
  }
}

創(chuàng)建驗(yàn)證結(jié)果的 HelloController.Java

@RestController
public class HelloController {

  @Value("${custom.property.message:}")
  private String customProperty;

  @RequestMapping("/test")
  public String test() {
    return customProperty;
  }
}

啟動工程服務(wù)，瀏覽器訪問 http://localhost:8080/test 即可查看輸出的結(jié)果為 “測試數(shù)據(jù)”；

看完上述內(nèi)容，你們對在Spring Boot項(xiàng)目中如何實(shí)現(xiàn)自定義PropertySourceLoader有進(jìn)一步的了解嗎？如果還想了解更多知識或者相關(guān)內(nèi)容，請關(guān)注億速云行業(yè)資訊頻道，感謝大家的支持。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手