Spring ProtocolResolver策略接口的用法

這篇文章主要講解了Spring ProtocolResolver策略接口的用法，內容清晰明了，對此有興趣的小伙伴可以學習一下，相信大家閱讀完之后會有幫助。

ProtocolResolver是一個策略接口，可以用于自定義協議解析， 比如spring就有一個 “classpath:”開頭的特定協議（但是spring并不是自定義ProtocolResolver 實現來完成這個功能的）

@FunctionalInterface
public interface ProtocolResolver {
	@Nullable
	Resource resolve(String location, ResourceLoader resourceLoader);

}

@FunctionalInterface:聲明該接口是一個函數式接口,主要用于編譯級錯誤檢查，加上該注解，該接口中只能定義唯一一個方法。

spring提供了ProtocolResolver機制，用于匹配自定義的文件schema來加載文件；而且不干擾ResourceLoader的機制，最重要的是它會添加到spring環境下的所有的loader中。我們只需要擴展一個ProtocolResolver類，并將它在合適的實際加入到ResourceLoader即可，此后加載properties文件時我們的ProtocolResolver總會被執行

具體用法如下：

自定義MyProtocolResolver,加載資源時，當我們輸入的資源路徑以path:開頭時，就會加載相對路徑下的config/下相應的資源。

public class MyProtocolResolver implements ProtocolResolver {
  public static final String PATH = "path:";
  @Override
  public Resource resolve(String location, ResourceLoader resourceLoader) {
    if (!location.startsWith(PATH))
      return null;
    String realPath = location.substring(5);
    String classPath = "classpath:config/" + realPath;
  
    return resourceLoader.getResource(classPath);
  }

}

測試：

DefaultResourceLoader resourceLoader=new DefaultResourceLoader();
resourceLoader.addProtocolResolver(new MyProtocolResolver());
Resource resource = resourceLoader.getResource("path:config.txt");
InputStream inputStream = resource.getInputStream();
StringBuffer out = new StringBuffer();
byte[] b = new byte[4096];
for (int n; (n = inputStream.read(b)) != -1; ) {
out.append(new String(b, 0, n));
}
System.out.println(out);

由此我們可以通過實現ProtocolResolver接口來定義我們自己的加載資源路徑，也可以自定義自己加載資源路徑的優先級。

看完上述內容，是不是對Spring ProtocolResolver策略接口的用法有進一步的了解，如果還想學習更多內容，歡迎關注億速云行業資訊頻道。