package project.ddos.internal;
|
|
import java.util.Date;
|
import java.util.List;
|
|
import org.springframework.jdbc.core.JdbcTemplate;
|
|
import kernel.bo.RecordObjectMapper;
|
import kernel.util.DateUtils;
|
import kernel.web.ApplicationUtil;
|
import project.ddos.DdosRedisKeys;
|
import project.ddos.IpMenuService;
|
import project.ddos.model.IpMenu;
|
import project.ddos.utils.BlacklistIpSerivceTimeWindow;
|
import project.ddos.utils.LocklistIpSerivceTimeWindow;
|
import project.redis.RedisHandler;
|
|
/**
|
* @author JORGE
|
* @description IP黑白名单服务
|
*/
|
public class IpMenuServiceImpl implements IpMenuService {
|
|
private RedisHandler redisHandler;
|
|
private JdbcTemplate jdbcTemplate;
|
|
private LocklistIpSerivceTimeWindow locklistIpSerivceTimeWindow;
|
|
private BlacklistIpSerivceTimeWindow blacklistIpSerivceTimeWindow;
|
|
public void init() {
|
List<IpMenu> list = jdbcTemplate.query("SELECT * FROM T_IP_MENU WHERE DELETE_STATUS=0",RecordObjectMapper.newInstance(IpMenu.class));
|
for (IpMenu ipMenu : list) {
|
if (DateUtils.addHour(ipMenu.getLast_opera_time(), 24).after(new Date())) { // 黑名单 时间+1天>现在
|
switch (ipMenu.getType()) {
|
case IpMenu.IP_BLACK:
|
blacklistIpSerivceTimeWindow.putBlackIp(ipMenu.getIp(), ipMenu.getIp());
|
break;
|
case IpMenu.IP_LOCK:
|
locklistIpSerivceTimeWindow.putLockIp(ipMenu.getIp(), ipMenu.getIp());
|
break;
|
}
|
} else {
|
ipMenu.setDelete_status(-1);
|
checkTimeWindows(ipMenu);
|
jdbcTemplate.update("UPDATE T_IP_MENU SET DELETE_STATUS=-1 WHERE UUID='"+ipMenu.getId().toString()+"'");
|
redisHandler.setSync(DdosRedisKeys.IP_MENU_IP + ipMenu.getIp(), ipMenu);
|
}
|
}
|
}
|
|
@Override
|
public void save(IpMenu entity) {
|
if(null==entity) return;
|
checkTimeWindows(entity);
|
if(null==entity.getId()) entity.setId(ApplicationUtil.getCurrentTimeUUID());
|
|
ApplicationUtil.executeInsert(entity);
|
redisHandler.setSync(DdosRedisKeys.IP_MENU_IP + entity.getIp(), entity);
|
}
|
|
@Override
|
public void update(IpMenu entity) {
|
if(null==entity) return;
|
checkTimeWindows(entity);
|
|
ApplicationUtil.executeUpdate(entity, "WHERE UUID=?", new Object[] {entity.getId()});
|
redisHandler.setSync(DdosRedisKeys.IP_MENU_IP + entity.getIp(), entity);
|
}
|
|
@Override
|
public void delete(IpMenu entity) {
|
if(null==entity) return;
|
|
ApplicationUtil.executeDelete(IpMenu.class, "WHERE UUID=?", new Object[] {entity.getId()});
|
redisHandler.remove(entity.getIp());
|
}
|
|
@Override
|
public IpMenu cacheByIp(String ip) {
|
return (IpMenu) redisHandler.get(DdosRedisKeys.IP_MENU_IP + ip);
|
}
|
|
/**
|
* 新增ip到白名单
|
* @param ip
|
*/
|
public void saveIpMenuWhite(String ip) {
|
IpMenu ipMenu = this.cacheByIp(ip);
|
if (null == ipMenu) {
|
ipMenu = new IpMenu();
|
ipMenu.setCreate_time(new Date());
|
ipMenu.setDelete_status(0);
|
ipMenu.setLast_opera_time(new Date());
|
ipMenu.setType(IpMenu.IP_WHITE);
|
ipMenu.setIp(ip);
|
this.save(ipMenu);
|
} else if (ipMenu.getDelete_status() == -1 || !IpMenu.IP_WHITE.equals(ipMenu.getType())) {// 名单被删除或者不是白名单
|
ipMenu.setDelete_status(0);
|
ipMenu.setLast_opera_time(new Date());
|
ipMenu.setType(IpMenu.IP_WHITE);
|
this.update(ipMenu);
|
}
|
}
|
|
public void checkTimeWindows(IpMenu entity) {
|
if (entity.getDelete_status() == -1) {
|
blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
|
locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
|
return;
|
}
|
|
switch (entity.getType()) {
|
case IpMenu.IP_WHITE:
|
if (blacklistIpSerivceTimeWindow.getBlackIp(entity.getIp()) != null) {// 白名单直接删除黑名单缓存
|
blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
|
}
|
if (locklistIpSerivceTimeWindow.getLockIp(entity.getIp()) != null) {// 白名单直接删除锁定名单缓存
|
locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
|
}
|
break;
|
case IpMenu.IP_BLACK:
|
if (locklistIpSerivceTimeWindow.getLockIp(entity.getIp()) != null) {// 删除锁定名单缓存
|
locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
|
}
|
blacklistIpSerivceTimeWindow.putBlackIp(entity.getIp(), entity.getIp());
|
break;
|
case IpMenu.IP_LOCK:
|
if (blacklistIpSerivceTimeWindow.getBlackIp(entity.getIp()) != null) {// 删除黑名单缓存
|
blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
|
}
|
locklistIpSerivceTimeWindow.putLockIp(entity.getIp(), entity.getIp());
|
break;
|
}
|
}
|
|
public void setRedisHandler(RedisHandler redisHandler) {
|
this.redisHandler = redisHandler;
|
}
|
|
public void setBlacklistIpSerivceTimeWindow(BlacklistIpSerivceTimeWindow blacklistIpSerivceTimeWindow) {
|
this.blacklistIpSerivceTimeWindow = blacklistIpSerivceTimeWindow;
|
}
|
|
public void setLocklistIpSerivceTimeWindow(LocklistIpSerivceTimeWindow locklistIpSerivceTimeWindow) {
|
this.locklistIpSerivceTimeWindow = locklistIpSerivceTimeWindow;
|
}
|
|
public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
|
this.jdbcTemplate = jdbcTemplate;
|
}
|
}
|
