1. 程式人生 > 程式設計 >Java後臺防止客戶端重複請求、提交表單實現原理

Java後臺防止客戶端重複請求、提交表單實現原理

這篇文章主要介紹了Java後臺防止客戶端重複請求、提交表單實現原理,文中通過示例程式碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

前言

在Web / App專案中,有一些請求或操作會對資料產生影響(比如新增、刪除、修改),針對這類請求一般都需要做一些保護,以防止使用者有意或無意的重複發起這樣的請求導致的資料錯亂。

常見處理方案

1.客戶端

  例如表單提交後將提交按鈕設為disable 等等方法...

2.服務端

  前端的限制僅能解決少部分問題,且不夠徹底,後端自有的防重複處理措施必不可少,義不容辭。

  在此提供一個我在專案中用到的方案。簡單來說就是判斷請求url和資料是否和上一次相同。

方法步驟

1.主要邏輯:

  給所有的url加一個攔截器,每次請求將url存入session,下次請求驗證url資料是否相同,相同則拒絕訪問。

  當然,我在此基礎上做了一些優化,比如:

    使用session有侷限性,使用者量大了以後伺服器會撐不住,在此我使用了redis來替換。

    加入了token令牌機制。

2.實現步驟:

2.1自定義一個註解

/**
 * @Title: SameUrlData
 * @Description: 自定義註解防止表單重複提交
 * @Auther: xhq
 * @Version: 1.0
 * @create 2019/3/26 10:43
 */
@Inherited
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface SameUrlData {

}

2.2自定義攔截器類

  • 檢查此介面呼叫的方法是否使用了SameUrlData註解,若沒有使用,表示此介面不需要校驗;
  • 若使用了註解,獲取請求url+引數,並去除一直在變化的引數(比如時間戳timeStamp和簽名sign)
  • 檢查引數中是否有token引數(token代表不同的使用者的唯一標識),沒有直接放行
  • 有token引數,將token+url作為redis的key,url+引數作為value存入redis,並設定自動銷燬時間
  • 再次訪問進行驗證是否重複請求  
import com.alibaba.fastjson.JSONObject;
import com.tuohang.hydra.framework.common.spring.SpringKit;
import com.tuohang.hydra.toolkit.basis.string.StringKit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
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.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @Title: 防止使用者重複提交資料攔截器
 * @Description: 將使用者訪問的url和引數結合token存入redis,每次訪問進行驗證是否重複請求介面
 * @Auther: xhq
 * @Version: 1.0
 * @create 2019/3/26 10:35
 */
@Component
public class SameUrlDataInterceptor extends HandlerInterceptorAdapter {

  private static Logger LOG = LoggerFactory.getLogger(SameUrlDataInterceptor.class);

  /**
   * 是否阻止提交,fasle阻止,true放行
   * @return
   */
  @Override
  public boolean preHandle(HttpServletRequest request,HttpServletResponse response,Object handler) throws Exception {
    if (handler instanceof HandlerMethod) {
      HandlerMethod handlerMethod = (HandlerMethod) handler;
      Method method = handlerMethod.getMethod();
      SameUrlData annotation = method.getAnnotation(SameUrlData.class);
      if (annotation != null) {
        if(repeatDataValidator(request)){
          //請求資料相同
          LOG.warn("please don't repeat submit,url:"+ request.getServletPath());
          JSONObject result = new JSONObject();
          result.put("statusCode","500");
          result.put("message","請勿重複請求");
          response.setCharacterEncoding("UTF-8");
          response.setContentType("application/json; charset=utf-8");
          response.getWriter().write(result.toString());
          response.getWriter().close();
//          攔截之後跳轉頁面
//          String formRequest = request.getRequestURI();
//          request.setAttribute("myurl",formRequest);
//          request.getRequestDispatcher("/WebRoot/common/error/jsp/error_message.jsp").forward(request,response);
          return false;
        }else {//如果不是重複相同資料
          return true;
        }
      }
      return true;
    } else {
      return super.preHandle(request,response,handler);
    }
  }
  /**
   * 驗證同一個url資料是否相同提交,相同返回true
   * @param httpServletRequest
   * @return
   */
  public boolean repeatDataValidator(HttpServletRequest httpServletRequest){
    //獲取請求引數map
    Map<String,String[]> parameterMap = httpServletRequest.getParameterMap();
    Iterator<Map.Entry<String,String[]>> it = parameterMap.entrySet().iterator();
    String token = "";
    Map<String,String[]> parameterMapNew = new HashMap<>();
    while(it.hasNext()){
      Map.Entry<String,String[]> entry = it.next();
      if(!entry.getKey().equals("timeStamp") && !entry.getKey().equals("sign")){
        //去除sign和timeStamp這兩個引數,因為這兩個引數一直在變化
        parameterMapNew.put(entry.getKey(),entry.getValue());
        if(entry.getKey().equals("token")) {
          token = entry.getValue()[0];
        }
      }
    }
    if (StringKit.isBlank(token)){
      //如果沒有token,直接放行
      return false;
    }
    //過濾過後的請求內容
    String params = JSONObject.toJSONString(parameterMapNew);

    System.out.println("params==========="+params);

    String url = httpServletRequest.getRequestURI();
    Map<String,String> map = new HashMap<>();
    //key為介面,value為引數
    map.put(url,params);
    String nowUrlParams = map.toString();

    StringRedisTemplate smsRedisTemplate = SpringKit.getBean(StringRedisTemplate.class);
    String redisKey = token + url;
    String preUrlParams = smsRedisTemplate.opsForValue().get(redisKey);
    if(preUrlParams == null){
      //如果上一個資料為null,表示還沒有訪問頁面
      //存放並且設定有效期,2秒
      smsRedisTemplate.opsForValue().set(redisKey,nowUrlParams,2,TimeUnit.SECONDS);
      return false;
    }else{//否則,已經訪問過頁面
      if(preUrlParams.equals(nowUrlParams)){
        //如果上次url+資料和本次url+資料相同,則表示重複新增資料
        return true;
      }else{//如果上次 url+資料 和本次url加資料不同,則不是重複提交
        smsRedisTemplate.opsForValue().set(redisKey,1,TimeUnit.SECONDS);
        return false;
      }
    }
  }
}

2.3註冊攔截器

@Configuration
public class WebMvcConfigExt extends WebMvcConfig {

  /**
   * 防止重複提交攔截器
   */
  @Autowired
  private SameUrlDataInterceptor sameUrlDataInterceptor;

  @Override
  public void addInterceptors(InterceptorRegistry registry) {
    // 避開靜態資源
    List<String> resourcePaths = defineResourcePaths();
    registry.addInterceptor(sameUrlDataInterceptor).addPathPatterns("/**").excludePathPatterns(resourcePaths);// 重複請求
  }

  /**
   * 自定義靜態資源路徑
   * 
   * @return
   */
  @Override
  public List<String> defineResourcePaths() {
    List<String> patterns = new ArrayList<>();
    patterns.add("/assets/**");
    patterns.add("/upload/**");
    patterns.add("/static/**");
    patterns.add("/common/**");
    patterns.add("/error");
    return patterns;
  }
}

在相應方法上加@SameUrlData註解

@SameUrlData
@ResponseBody
@RequestMapping(value = "/saveOrUpdate")
public String saveOrUpdate(){
}

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。