diff --git a/src/main/java/im/zhaojun/common/model/constant/ZFileConstant.java b/src/main/java/im/zhaojun/common/model/constant/ZFileConstant.java index 63013b2..39d03ce 100644 --- a/src/main/java/im/zhaojun/common/model/constant/ZFileConstant.java +++ b/src/main/java/im/zhaojun/common/model/constant/ZFileConstant.java @@ -1,5 +1,10 @@ package im.zhaojun.common.model.constant; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.annotation.Configuration; + +@Configuration public class ZFileConstant { public final static String USER_HOME = System.getProperty("user.home"); @@ -9,16 +14,31 @@ public class ZFileConstant { /** * 页面头部文件 */ - public static final String HEADER_FILE_NAME = "header.md"; + public static String HEADER_FILE_NAME = "header.md"; /** * 页面尾部文件 */ - public static final String FOOTER_FILE_NAME = "footer.md"; + public static String FOOTER_FILE_NAME = "footer.md"; /** * 密码文件 */ - public static final String PASSWORD_FILE_NAME = "password.txt"; + public static String PASSWORD_FILE_NAME = "password.txt"; + + @Autowired(required = false) + public void setHeaderFileName(@Value("${zfile.constant.header}") String headerFileName) { + ZFileConstant.HEADER_FILE_NAME = headerFileName; + } + + @Autowired(required = false) + public void setFooterFileName(@Value("${zfile.constant.footer}") String footerFileName) { + ZFileConstant.FOOTER_FILE_NAME = footerFileName; + } + + @Autowired(required = false) + public void setPasswordFileName(@Value("${zfile.constant.password}") String passwordFileName) { + ZFileConstant.PASSWORD_FILE_NAME = passwordFileName; + } } diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml index 33233fb..4057bef 100644 --- a/src/main/resources/application.yml +++ b/src/main/resources/application.yml @@ -50,4 +50,8 @@ spring: password: 12345 zfile: cache: - timeout: 300 \ No newline at end of file + timeout: 300 + constant: + header: header.md + footer: footer.md + password: password.txt \ No newline at end of file