package com.framework.plugin.token;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.ElementType;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Token {
boolean save() default false;
boolean remove() default false;
}
package com.framework.plugin.token;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.lang.reflect.Method;
import java.util.UUID;
public class TokenInterceptor extends HandlerInterceptorAdapter {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
if (handler instanceof HandlerMethod) {
HandlerMethod handlerMethod = (HandlerMethod) handler;
Method method = handlerMethod.getMethod();
Token annotation = method.getAnnotation(Token.class); //判断是否有标注
if (annotation != null) {
boolean needSaveSession = annotation.save();
if (needSaveSession) {
//生成uuid随机token,反射到jsp
request.getSession(false).setAttribute("token", UUID.randomUUID().toString());
}
boolean needRemoveSession = annotation.remove(); //saveentity方法触发
if (needRemoveSession) {
if (isRepeatSubmit(request)) {
return false;
}
//防止下一次进入,去掉token,
request.getSession(false).removeAttribute("token");
}
}
return true;
} else {
return super.preHandle(request, response, handler);
}
}
/**
* 判断是否重复点击
* @author 李品良
*创建时间:2016年9月10日 下午2:26:05
* @param request
* @return
*/
private boolean isRepeatSubmit(HttpServletRequest request) {
//那么下一次,这个方法就失败了.
String serverToken = (String) request.getSession(false).getAttribute("token");
if (serverToken == null) {
return true;
}
String clinetToken = request.getParameter("token");
if (clinetToken == null) {
return true;
}
if (!serverToken.equals(clinetToken)) {
return true;
}
return false;
}
}
<!-- token拦截器 -->
<mvc:interceptors>
<mvc:interceptor>
<mvc:mapping path="/**" />
<bean class="com.framework.plugin.token.TokenInterceptor" />
</mvc:interceptor>
</mvc:interceptors>
jsp要加上
<input type="hidden" name="token" value="${token}"/>
controllor 加上
/**
* 跳转到新增界面
*
* @return
*/
@RequestMapping("addUI")
@Token(save = true)
public String addUI(Model model) {
return Common.BACKGROUND_PATH + "/system/producttree/add";
}
/**
* 添加菜单
*
* @param producttree
* @return Map
* @throws Exception
*/
@RequestMapping("addEntity")
@ResponseBody
@Token(remove = true)
@Transactional(readOnly = false) // 需要事务操作必须加入此注解
@SystemLog(module = "系统管理", methods = "产品管理-新增产品") // 凡需要处理业务逻辑的.都需要记录操作日志
public String addEntity() throws Exception {
ProductTreeFormMap productTreeFormMap = getFormMap(ProductTreeFormMap.class);
productTreeFormMap.save();
return "success";
}