java 实现ip访问量控制_java 实现IP访问量控制

网站中发现有人恶意频繁的注册账号，打算在注册接口处增加效验工具，思路如下：

记录访问IP 和 时间

如果在五分钟内连续访问更新计时器更新计数器

10分钟内不访问计时器清零

思路很简单源码如下

package com.zcjy.mft.util;

import java.util.Date;

import org.apache.commons.collections.map.LRUMap;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

public class Forbidener {

final Log log = LogFactory

.getLog(Forbidener.class);

private Forbidener(){

ipmap = new LRUMap(100000);

}

public static Forbidener forbidener;

public static Forbidener getInstance(){

if(forbidener == null){

forbidener = new Forbidener();

}

return forbidener;

}

LRUMap ipmap = null ;

/**

* 超过次数封禁

*/

long maxTime = 10;

/**

* 超过访问时间重新计时

*/

long minInteval = 60*1000*60*4;

/**

* 封禁的时间

*/

long forbidenTime = 60*1000*60*4;

public int getMapSize() {

return ipmap.size();

}

public void setMaxTime(int i) {

maxTime = i;

}

public void setMinInteval(int i) {

minInteval = i;

}

public void setForbidenTime(int i) {

forbidenTime = i;

}

public void setMaxMapSize(int i) {

ipmap = new LRUMap(i);

}

public boolean check(String IP){

IPer iper = (IPer)ipmap.get(IP);

if(iper == null){

iper = new IPer();

iper.lastVisiterTime = new Date();

iper.visitTimes = 1;

ipmap.put(IP, iper);

return true;

}

Date now = new Date();

if(iper.isForbiden){

if(now.getTime() - iper.lastVisiterTime.getTime() > forbidenTime){

//解封

iper.isForbiden = false;

iper.visitTimes = 1;

iper.lastVisiterTime = now;

log.debug("unforbiden ip:" + IP);

return true;

}else{

return false;

}

}

if(now.getTime() - iper.lastVisiterTime.getTime() > minInteval){

//太久没有访问了,重新计时

iper.lastVisiterTime = now;

iper.visitTimes = 1;

iper.isForbiden = false;

log.debug("reset ip:" + IP);

return true;

}else{

iper.lastVisiterTime = now;

iper.visitTimes = iper.visitTimes +1;

if(iper.visitTimes > maxTime){

//封禁

iper.isForbiden = true;

log.debug("forbiden ip:" + IP);

return false;

}else{

return true;

}

}

}

public static void main(String[] args) throws InterruptedException {

Date d1 = new Date();

Thread.sleep(1000);

Date d2 = new Date();

System.out.println(d1.getTime() - d2.getTime());

}

}

class IPer {

Date lastVisiterTime ;

int visitTimes;

boolean isForbiden = false;

}测试代码如下

package com.zcjy.mft.util;

import org.junit.AfterClass;

import org.junit.Assert;

import org.junit.Test;

import com.zcjy.mft.util.Forbidener;

public class TestForbidener {

public static void setup(){

//构造初始化数据

}

@AfterClass

public static void clear(){

}

/**

* @throws InterruptedException

*/

@Test

public void check() throws InterruptedException{

//测试十次封禁

Forbidener fb = Forbidener.getInstance();

fb.setMaxTime(10);

fb.setMinInteval(10*1000);

fb.setForbidenTime(60*1000);

fb.setMaxMapSize(3);

String testIP = "127.0.0.1";

for(int i = 1 ; i <= 10;i++){

System.out.println("check");

Assert.assertTrue(fb.check(testIP));

Thread.sleep(1000);

}

Assert.assertFalse(fb.check(testIP));

//测试解封

System.out.println("sleep more than 1 min");

Thread.sleep(65*1000);

Assert.assertTrue(fb.check(testIP));

//测试最短周期

for(int i = 1;i<= 9 ;i++){

Assert.assertTrue(fb.check(testIP));

System.out.println("check");

Thread.sleep(1000);

}

System.out.println("sleep more than 10 sec");

Thread.sleep(15*1000);

Assert.assertTrue(fb.check(testIP));

//测试淘汰

fb.check("1");

fb.check("2");

Assert.assertTrue(fb.getMapSize() == 3);

fb.check("3");

Assert.assertTrue(fb.getMapSize() == 3);

//

}

} 不写测试用例的程序员不是好程序员；