BeetlConfiguration扩展配置

1beetl拓展配置类,绑定一些工具类,方便在模板中直接调用
2
3BeetlConfiguration.java
4
5public class BeetlConfiguration extends BeetlGroupUtilConfiguration {
6
7    @Autowired
8    private Environment env;
9
10    @Autowired
11    private DictSelectorTag dictSelectorTag;
12
13    @Override
14    public void initOther() {
15        groupTemplate.registerFunctionPackage("shiro", new ShiroExt());
16        groupTemplate.registerFunctionPackage("tool", new ToolUtil());
17        groupTemplate.registerFunctionPackage("kaptcha", new KaptchaUtil());
18        groupTemplate.registerTagFactory("dictSelector", () -> dictSelectorTag);
19
20        groupTemplate.registerFunction("env", new Function() {
21            @Override
22            public String call(Object[] paras, Context ctx) {
23                String key = (String) paras[0];
24                String value = env.getProperty(key);
25                if (value != null) {
26                    return getStr(value);
27                }
28                if (paras.length == 2) {
29                    return (String) paras[1];
30                }
31                return null;
32            }
33
34            String getStr(String str) {
35                try {
36                    return new String(str.getBytes("iso8859-1"), StandardCharsets.UTF_8);
37                } catch (UnsupportedEncodingException e) {
38                    throw new RuntimeException(e);
39                }
40            }
41        });
42    }
43}