springboot过滤器禁止ip频繁访问

package ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

import ;

/**

* 自定义过滤器，用来判断IP访问次数是否超限

* 如果前台用户访问网站的频率过快（达到超过50次/秒），则判定该ip恶意刷新操作，

* 限制该IP的访问，1小时后自己解除限制

*

* @author wukaikai

*

*/

@WebFilter(urlPatterns="/*")

public class IpFilter implements Filter{

/**

* 默认限制时间（单位：ms）

*/

private static final long LIMITED_TIME_MILLIS = 60 * 60 * 1000;

/**

* 用户连续访问最高阀值，超过该值则认定为恶意操作的IP，进行限制

*/

private static final int LIMIT_NUMBER = 50;

/**

* 用户访问最小安全时间，在该时间内如果访问次数大于阀值，则记录为恶意IP，否则视为正常访问

*/

private static final int MIN_SAFE_TIME = 5000;

private FilterConfig config;

@Override

public void init(FilterConfig filterConfig) throws ServletException {

this.config = filterConfig; //设置属性filterConfig

}

/* (non-Javadoc)

* @see #doFilter(, , )

*/

@SuppressWarnings("unchecked")

@Override

public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain chain)

throws IOException, ServletException {

HttpServletRequest request = (HttpServletRequest) servletRequest;

HttpServletResponse response = (HttpServletResponse) servletResponse;

ServletContext context = ();

// 获取限制IP存储器：存储被限制的IP信息

Map<String, Long> limitedIpMap = (Map<String, Long>) ("limitedIpMap");

// 过滤受限的IP

filterLimitedIpMap(limitedIpMap);

// 获取用户IP

String ip = (request);

("ip:"+ip);

// 判断是否是被限制的IP，如果是则跳到异常页面

if (isLimitedIP(limitedIpMap, ip)) {

long limitedTime = (ip) - ();

// 剩余限制时间(用为从毫秒到秒转化的一定会存在些许误差，但基本可以忽略不计)

("remainingTime", ((limitedTime / 1000) + (limitedTime % 1000 > 0 ? 1 : 0)));

//("/error/overLimitIP").forward(request, response);

("ip访问过于频繁："+ip);

//("gb2312"); //设置输出内容编码格式

// ();

// ("utf-8"); //设置输出内容编码格式

// PrintWriter out = ();

// ("<b>由于您访问过于频繁，被系统自动认定为机器人。1个小时自动解除</b>");、

return;

}

// 获取IP存储器

Map<String, Long[]> ipMap = (Map<String, Long[]>) ("ipMap");

// 判断存储器中是否存在当前IP，如果没有则为初次访问，初始化该ip

// 如果存在当前ip，则验证当前ip的访问次数

// 如果大于限制阀值，判断达到阀值的时间，如果不大于[用户访问最小安全时间]则视为恶意访问，跳转到异常页面

if ((ip)) {

Long[] ipInfo = (ip);

ipInfo[0] = ipInfo[0] + 1;

("当前第[" + (ipInfo[0]) + "]次访问");

if (ipInfo[0] > LIMIT_NUMBER) {

Long ipAccessTime = ipInfo[1];

Long currentTimeMillis = ();

if (currentTimeMillis - ipAccessTime <= MIN_SAFE_TIME) {

(ip, currentTimeMillis + LIMITED_TIME_MILLIS);

("remainingTime", LIMITED_TIME_MILLIS);

("ip访问过于频繁："+ip);

("/error/overLimitIP").forward(request, response);

return;

} else {

initIpVisitsNumber(ipMap, ip);

}

}

} else {

initIpVisitsNumber(ipMap, ip);

("您首次访问该网站");

}

("ipMap", ipMap);

(request, response);

}

@Override

public void destroy() {

// TODO Auto-generated method stub

}

/**

* @Description 过滤受限的IP，剔除已经到期的限制IP

* @param limitedIpMap

*/

private void filterLimitedIpMap(Map<String, Long> limitedIpMap) {

if (limitedIpMap == null) {

return;

}

Set<String> keys = ();

Iterator<String> keyIt = ();

long currentTimeMillis = ();

while (()) {

long expireTimeMillis = (());

if (expireTimeMillis <= currentTimeMillis) {

();

}

}

}

/**

* @Description 是否是被限制的IP

* @param limitedIpMap

* @param ip

* @return true : 被限制 | false : 正常

*/

private boolean isLimitedIP(Map<String, Long> limitedIpMap, String ip) {

if (limitedIpMap == null || ip == null) {

// 没有被限制

return false;

}

Set<String> keys = ();

Iterator<String> keyIt = ();

while (()) {

String key = ();

if ((ip)) {

// 被限制的IP

return true;

}

}

return false;

}

/**

* 初始化用户访问次数和访问时间

*

* @param ipMap

* @param ip

*/

private void initIpVisitsNumber(Map<String, Long[]> ipMap, String ip) {

Long[] ipInfo = new Long[2];

ipInfo[0] = 0L;// 访问次数

ipInfo[1] = ();// 初次访问时间

(ip, ipInfo);

}

}