freemarker自定义标签实现权限管理

目的：自定义一个freemarker标签，当满足时显示标签体中的内容，并输出动态内容，否则不进行显示。

实现：

1、实现 TemplateDirectiveModel 接口（freemarker实现自定义标签需要实现这个接口）

2、在freemarker的配置类中，在freemarker的Configuration中注册这个自定义标签

3、页面上进行使用

注:自定义标签每次渲染都会进入一次标签处理类

一、实现TemplateDirectiveModel接口

import freemarker.core.Environment;
import freemarker.core.StringArraySequence;
import freemarker.template.*;
import lombok.extern.log4j.Log4j2;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Log4j2
public class AuthDirective implements TemplateDirectiveModel {

    private static final String AUTH = "auth";

    @Override
    public void execute(Environment env, Map params, TemplateModel[] loopVars, TemplateDirectiveBody body) throws TemplateException, IOException {
        // 1.获取用户所有的权限
        List<String> list = new ArrayList<>();
        list.add("01");
        list.add("03");
        log.info("获取当前权限列表:{} AuthDirective-execute", JsonBinder.buildNormalBinder().toJson(list));
        
        // 2.获取标签中auth属性的值，此处我知道前台页面传递过来的是String类型，所以可以直接强制转换成SimpleScalar类型
        SimpleScalar auth = (SimpleScalar) params.get(AUTH);
        log.info("获取到标签需要的权限:{} AuthDirective-execute",JsonBinder.buildNormalBinder().toJson(auth));
        // 3、给页面上的循环变量设置值
        if (null != loopVars && loopVars.length > 0) {
            loopVars[0] = new SimpleScalar("<span style='color:red'>这个是后台返回的值</span>"); // 返回给前台一个String的类型
            loopVars[1] = new StringArraySequence(new String[]{"张三", "李四", "王五"});         // 返回给前台一个Sequence类型
        }
        // 4.判断用户是否拥有这个权限
        if (list.contains(auth.getAsString())) {
            log.info("拥有权限就输出显示 AuthDirective-execute");
            // 显示指令包含的中间的内容
            body.render(env.getOut());
        }else{
            log.info("没有权限不显示 AuthDirective-execute");
        }
    }
}

二、注册标签

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.extern.log4j.Log4j2;

import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

/**
 * @describe FreeMarker初始化配置
 *
 */
@Log4j2
@Data
@Configuration
@NoArgsConstructor
@AllArgsConstructor
public class FreeMarkerConfig implements InitializingBean {
	
	@Resource
	private freemarker.template.Configuration configuration;

	@Bean
	public AuthDirective authDirective() {
		return new AuthDirective();
	}

	@Override
	public void afterPropertiesSet() throws Exception {
		configuration.setSharedVariable("auth", authDirective());
	}
}

三、使用标签(其中的auth值为传到服务端进行匹配用的)

<@auth auth="03";loopVar01,loopVar02>
					我可以出来,获取到后台返回${loopVar01}<br/>
					<#list loopVar02 as loop>
						per : ${loop_index}-${loop}
					</#list>
				</@auth>