freemarker实现自定义标签

freemarker实现自定义标签

freemarker实现自定义标签其实并没有什么难度，这个功能我们叫自定义标签，在官网中称为指令，也并不是什么高级技术，只是大家没发现而已，参考下官网文档就能实现：https://freemarker.apache.org/docs/pgui_datamodel_directive.html
http://freemarker.foofun.cn/pgui_datamodel_directive.html
更多功能需要自己去研究，编写这编文章的目的是因为我在网上无法搜索到满意的结果。

一、依赖

    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-freemarker</artifactId>
    </dependency>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

二、首先自定义一个标签 auth

我们的标签是这个样子：

<!-- 提前将标签引入 -->
<#assign auth="com.lingkang.flynovel.config.tag.AuthTag"?new()>
<@auth security="isAuthenticated">
    已经登录
</@auth>

<@auth hasRole="ROLE_USER">
  存在角色 ROLE_USER
</@auth>

那么我们的Java代码这样写：

import freemarker.core.Environment;
import freemarker.template.TemplateDirectiveBody;
import freemarker.template.TemplateDirectiveModel;
import freemarker.template.TemplateException;
import freemarker.template.TemplateModel;
import java.io.IOException;
import java.util.Collection;
import java.util.Map;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;

/**
 * @author 绫小路
 * @date 2021/3/27
 * @description 自定义标签, 首先引入 <#assign auth="com.lingkang.flynovel.config.tag.AuthTag"?new()>
 * <p><@auth hasRole="ROLE_USER"> 存在角色 ROLE_USER </@auth>
 */
@Component
public class AuthTag implements TemplateDirectiveModel {

  @Override
  public void execute(Environment environment, Map map, TemplateModel[] templateModels, TemplateDirectiveBody templateDirectiveBody)
      throws TemplateException, IOException {
    // 判断是否授权标签：
    /**
     * <#assign auth="com.lingkang.flynovel.config.tag.AuthTag"?new()>
     * <@auth security="isAuthenticated";aaaa>
     *     已经登录
     * </@auth>
     */
    Object o = map.get("security");
    if (o != null && "isAuthenticated".equals(o.toString())) {
      if (SecurityContextHolder.getContext().getAuthentication() == null) {
      // 输出标签的内容为null
        templateDirectiveBody.render(null);
      } else {
      // 将标签的内容输出
        templateDirectiveBody.render(environment.getOut());
      }
      return;
    }

    //判断是否存在角色
    /**
     * <@auth hasRole="ROLE_USER">
     *   存在角色 ROLE_USER
     * </@auth>
     */
    Object hasRole = map.get("hasRole");
    if (hasRole != null) {
      String role = hasRole.toString();
      Collection<? extends GrantedAuthority> authorities = SecurityContextHolder.getContext().getAuthentication().getAuthorities();
      for (GrantedAuthority g : authorities) {
        if (role.equals(g.getAuthority())) {
          templateDirectiveBody.render(environment.getOut());
          return;
        }
      }
      return;
    }
  }
}

效果截图：