cn.twelvet.xss.core.XssHolder Maven / Gradle / Ivy

Go to download
package cn.twelvet.xss.core;

import cn.twelvet.xss.annotation.XssCleanIgnore;

/**
 * 利用 ThreadLocal 缓存线程间的数据
 *
 * @author twelvet
 */
public class XssHolder {

	private static final ThreadLocal TL = new ThreadLocal<>();

	private static final ThreadLocal TL_IGNORE = new ThreadLocal<>();

	/**
	 * 是否开启
	 * @return boolean
	 */
	public static boolean isEnabled() {
		return Boolean.TRUE.equals(TL.get());
	}

	/**
	 * 标记为开启
	 */
	static void setEnable() {
		TL.set(Boolean.TRUE);
	}

	/**
	 * 保存接口上的 XssCleanIgnore
	 * @param xssCleanIgnore XssCleanIgnore
	 */
	public static void setXssCleanIgnore(XssCleanIgnore xssCleanIgnore) {
		TL_IGNORE.set(xssCleanIgnore);
	}

	/**
	 * 获取接口上的 XssCleanIgnore
	 * @return XssCleanIgnore
	 */
	public static XssCleanIgnore getXssCleanIgnore() {
		return TL_IGNORE.get();
	}

	/**
	 * 关闭 xss 清理
	 */
	public static void remove() {
		TL.remove();
		TL_IGNORE.remove();
	}

}