gradle項目中資源文件的相對路徑打包處理技巧 - yjmyzz－IT工程師數位筆記本

文章出處

開發java application時，不管是用ant/maven/gradle中的哪種方式來構建，通常最后都會打包成一個可執行的jar包程序，而程序運行所需的一些資源文件(配置文件)，比如jdbc.properties, log4j2.xml，spring-xxx.xml這些，可以一起打包到jar中，程序運行時用類似classpath*:xxx.xml的去加載，大多數情況下，這樣就能工作得很好了。

但是，如果有一天，需要修正配置，比如：一個應用上線初期，為了調試方便，可能會把log的日志級別設置低一些，比如：INFO級別，運行一段時間穩定以后，只需要記錄WARN或ERROR級別的日志，這時候就需要修改log4j2.xml之類的配置文件，如果把配置文件打包在jar文件內部，改起來就比較麻煩，要把重新打包部署，要么在線上，先用jar命令將jar包解壓，改好后，再打包回去，比較繁瑣。

面對這種需求，更好的方式是把配置文件放在jar文件的外部相對目錄下，程序啟動時去加載相對目錄下的配置文件，這樣改起來，就方便多了，下面演示如何實現：(以gradle項目為例)

主要涉及以下幾點：

1、如何不將配置文件打包到jar文件內

既然配置文件放在外部目錄了，jar文件內部就沒必要再重復包含這些文件了，可以修改build.gradle文件，參考下面這樣：

processResources {
    exclude { "**/*.*" }
}

相當于覆蓋了默認的processResouces task，這樣gradle打包時，資源目錄下的任何文件都將排除。

2、log4j2的配置加載處理

log4j2加載配置文件時，默認情況下會找classpath下的log4j2.xml文件，除非手動給它指定配置文件的位置，分析它的源碼，可以找到下面這段：org.apache.logging.log4j.core.config.ConfigurationFactory.Factory#getConfiguration(java.lang.String, java.net.URI)

 1 public Configuration getConfiguration(final String name, final URI configLocation) {
 2 
 3             if (configLocation == null) {
 4                 final String configLocationStr = this.substitutor.replace(PropertiesUtil.getProperties()
 5                         .getStringProperty(CONFIGURATION_FILE_PROPERTY));
 6                 if (configLocationStr != null) {
 7                     ConfigurationSource source = null;
 8                     try {
 9                         source = getInputFromUri(NetUtils.toURI(configLocationStr));
10                     } catch (final Exception ex) {
11                         // Ignore the error and try as a String.
12                         LOGGER.catching(Level.DEBUG, ex);
13                     }
14                     if (source == null) {
15                         final ClassLoader loader = LoaderUtil.getThreadContextClassLoader();
16                         source = getInputFromString(configLocationStr, loader);
17                     }
18                     if (source != null) {
19                         for (final ConfigurationFactory factory : getFactories()) {
20                             final String[] types = factory.getSupportedTypes();
21                             if (types != null) {
22                                 for (final String type : types) {
23                                     if (type.equals("*") || configLocationStr.endsWith(type)) {
24                                         final Configuration config = factory.getConfiguration(source);
25                                         if (config != null) {
26                                             return config;
27                                         }
28                                     }
29                                 }
30                             }
31                         }
32                     }
33                 } else {
34                     for (final ConfigurationFactory factory : getFactories()) {
35                         final String[] types = factory.getSupportedTypes();
36                         if (types != null) {
37                             for (final String type : types) {
38                                 if (type.equals("*")) {
39                                     final Configuration config = factory.getConfiguration(name, configLocation);
40                                     if (config != null) {
41                                         return config;
42                                     }
43                                 }
44                             }
45                         }
46                     }
47                 }
48             } else {
49                 // configLocation != null
50                 final String configLocationStr = configLocation.toString();
51                 for (final ConfigurationFactory factory : getFactories()) {
52                     final String[] types = factory.getSupportedTypes();
53                     if (types != null) {
54                         for (final String type : types) {
55                             if (type.equals("*") || configLocationStr.endsWith(type)) {
56                                 final Configuration config = factory.getConfiguration(name, configLocation);
57                                 if (config != null) {
58                                     return config;
59                                 }
60                             }
61                         }
62                     }
63                 }
64             }
65 
66             Configuration config = getConfiguration(true, name);
67             if (config == null) {
68                 config = getConfiguration(true, null);
69                 if (config == null) {
70                     config = getConfiguration(false, name);
71                     if (config == null) {
72                         config = getConfiguration(false, null);
73                     }
74                 }
75             }
76             if (config != null) {
77                 return config;
78             }
79             LOGGER.error("No log4j2 configuration file found. Using default configuration: logging only errors to the console.");
80             return new DefaultConfiguration();
81         }

View Code

其中常量CONFIGURATION_FILE_PROPERTY的定義為：

public static final String CONFIGURATION_FILE_PROPERTY = "log4j.configurationFile";

View Code

從這段代碼可以看出，只要在第一次調用log4j2的getLogger之前設置系統屬性，將其指到配置文件所在的位置即可。

3、其它一些配置文件(比如spring配置)的相對路徑加載

這個比較容易，spring本身就支持從文件目錄加載配置的能力。

綜合以上分析，可以封裝一個工具類：

 1 import org.springframework.context.ConfigurableApplicationContext;
 2 import org.springframework.context.support.FileSystemXmlApplicationContext;
 3 
 4 import java.io.File;
 5 
 6 
 7 public class ApplicationContextUtil {
 8 
 9     private static ConfigurableApplicationContext context = null;
10 
11     private static ApplicationContextUtil instance = null;
12 
13     public static ApplicationContextUtil getInstance() {
14         if (instance == null) {
15             synchronized (ApplicationContextUtil.class) {
16                 if (instance == null) {
17                     instance = new ApplicationContextUtil();
18                 }
19             }
20         }
21         return instance;
22     }
23 
24     public ConfigurableApplicationContext getContext() {
25         return context;
26     }
27 
28     private ApplicationContextUtil() {
29 
30     }
31 
32 
33     static {
34 
35         //加載log4j2.xml
36         String configLocation = "resources/log4j2.xml";
37         File configFile = new File(configLocation);
38         if (!configFile.exists()) {
39             System.err.println("log4j2 config file:" + configFile.getAbsolutePath() + " not exist");
40             System.exit(0);
41         }
42         System.out.println("log4j2 config file:" + configFile.getAbsolutePath());
43 
44         try {
45             //注:這一句必須放在整個應用第一次LoggerFactory.getLogger(XXX.class)前執行
46             System.setProperty("log4j.configurationFile", configFile.getAbsolutePath());
47         } catch (Exception e) {
48             System.err.println("log4j2 initialize error:" + e.getLocalizedMessage());
49             System.exit(0);
50         }
51 
52         //加載spring配置文件
53         configLocation = "resources/spring-context.xml";
54         configFile = new File(configLocation);
55 
56         if (!configFile.exists()) {
57             System.err.println("spring config file:" + configFile.getAbsolutePath() + " not exist");
58             System.exit(0);
59         }
60 
61         System.out.println("spring config file:" + configFile.getAbsolutePath());
62 
63         if (context == null) {
64             context = new FileSystemXmlApplicationContext(configLocation);
65             System.out.println("spring load success!");
66         }
67 
68     }
69 
70 
71 }

View Code

注：這里約定了配置文件放在相對目錄resources下，而且log4j2的配置文件名為log4j2.xml，spring的入口配置文件為spring-context.xml(如果不想按這個約定來，可參考這段代碼自行修改)

有了這個工具類，mainclass入口程序上可以這么用：

 1 import org.slf4j.Logger;
 2 import org.slf4j.LoggerFactory;
 3 import org.springframework.context.ApplicationContext;
 4 
 5 /**
 6  * Created by yangjunming on 12/15/15.
 7  * author: yangjunming@huijiame.com
 8  */
 9 public class App {
10 
11     private static ApplicationContext context;
12     private static Logger logger;
13 
14 
15     public static void main(String[] args) {
16 
17         context = ApplicationContextUtil.getInstance().getContext();
18         logger = LoggerFactory.getLogger(App.class);
19 
20         System.out.println("start ...");
21 
22         logger.debug("debug message");
23 
24         logger.info("info message");
25 
26         logger.warn("warn message");
27 
28         logger.error("error message");
29 
30         System.out.println(context.getBean(SampleObject.class));
31 
32     }
33 }

View Code

再次友情提醒：logger的實例化，一定要放在ApplicationContextUtil.getInstance().getContext();之后，否則logger在第一次初始化時，仍然嘗試會到classpath下去找log4j2.xml文件，實例化之后，后面再設置系統屬性就沒用了。

4、gradle 打包的處理

代碼寫完了，還有最后一個工作沒做，既然配置文件不打包到jar里了，那就得復制到jar包的相對目錄resources下，可以修改build.gradle腳本，讓計算機處理處理，在代替手動復制配置文件。

task pack(type: Copy, dependsOn: [clean, installDist]) {
    sourceSets.main.resources.srcDirs.each {
        from it
        into "$buildDir/install/$rootProject.name/bin/resources"
    }
}

增加這個task后，直接用gradle pack 就可以實現打包，并自動復制配置文件到相對目錄resources目錄下了，參考下圖：

最后國際慣例，給個示例源碼：https://github.com/yjmyzz/config-load-demo

gradle pack 后，可進入build/install/config-load-demo/bin 目錄，運行./config-load-demo (windows下運行config-load-demo.bat) 查看效果，然后嘗試修改resources/log4j2.xml里的日志級別，再次運行，觀察變化。

文章列表