Merge remote-tracking branch 'origin/v2.0.9_dev' into v2.0.9_dev

This commit is contained in:
HuangXin 2021-09-15 16:24:49 +08:00
commit f8778ad438
23 changed files with 841 additions and 198 deletions

View File

@ -6,32 +6,27 @@ dispose.split_char=,
dispose.request-timeout-second=5 dispose.request-timeout-second=5
# 是否开启隐私保护 # 是否开启隐私保护
dispose.used-privacy-protect=false dispose.used-privacy-protect=false
dispose.call-error-retry-times=3 dispose.call-error-retry-times=3
# 分页配置项 # 分页配置项
# 最大每页数据条数 # 最大每页数据条数
dispose.max-split-page-size=100 dispose.max-split-page-size=100
# 最小每页数据条数 # 最小每页数据条数
dispose.min-split-page-size=10 dispose.min-split-page-size=10
# 迪普设备配置 # 迪普设备配置
# 发送超时时间(s) # 发送超时时间(s)
dptech.soap-conn-timeout-second=60 dptech.soap-conn-timeout-second=60
# 接收超时时间(s) # 接收超时时间(s)
dptech.soap-recv-timeout-second=60 dptech.soap-recv-timeout-second=60
# 用户权限配置 # 用户权限配置
# 是否对设备管理进行用户验证 # 是否对设备管理进行用户验证
permission.admin-check=true permission.admin-check=true
# 运行管理设备的操作员用户名 # 运行管理设备的操作员用户名
permission.admin-users=admin permission.admin-users=admin
# 认证配置 # 认证配置
# 是否对接口访问进行认证 # 是否对接口访问进行认证
auth.verify-request-token=true auth.verify-request-token=true
# token访问超时时间 # token访问超时时间
auth.token-timeout-minute=30 auth.token-timeout-minute=30
# 安全配置 # 安全配置
#加密类型: 0 不加密 #加密类型: 0 不加密
# 1 Base64编码 # 1 Base64编码
@ -41,7 +36,6 @@ crypto.security-protocol-type=0
crypto.aes-key=hkoUV5ZWh0q1jSxMnpjovVn19Qg99HY6DD40 crypto.aes-key=hkoUV5ZWh0q1jSxMnpjovVn19Qg99HY6DD40
# 3DES秘钥 # 3DES秘钥
crypto.des-key=P3mq9iSIvQcvfyfdWR8sAnfAadO crypto.des-key=P3mq9iSIvQcvfyfdWR8sAnfAadO
# Kafka 服务器配置 # Kafka 服务器配置
#重试次数 #重试次数
kafka.producer.retries=3 kafka.producer.retries=3
@ -53,12 +47,6 @@ kafka.producer.linger=1
kafka.producer.buffer.memory=33554432 kafka.producer.buffer.memory=33554432
kafka.producer.servers=172.21.44.189:9092,172.21.44.9:9092,172.21.44.244:9092,172.21.44.236:9092,172.21.44.80:9092 kafka.producer.servers=172.21.44.189:9092,172.21.44.9:9092,172.21.44.244:9092,172.21.44.236:9092,172.21.44.80:9092
kafka.dispose.topic=ddos-vip-customer-ck kafka.dispose.topic=ddos-vip-customer-ck
kafka.consumer.group-id = testGroup
kafka.consumer.auto-offset-reset = earliest
kafka.consumer.enable-auto-commit = true
kafka.consumer.auto-commit-interval = 100
kafka.consumer.max-poll-records =5
#信任主机配置 #信任主机配置
# 白名单开关 # 白名单开关
trust.auth-white-list-check=true trust.auth-white-list-check=true

View File

@ -4,17 +4,16 @@ server.tomcat.basedir=./basedir
# 多个项目放在nginx下同个端口通过该配置区分 # 多个项目放在nginx下同个端口通过该配置区分
server.servlet.context-path=/dispose server.servlet.context-path=/dispose
# 配置数据源 # 配置数据源
#spring.datasource.url=jdbc:mysql://10.88.77.65:33061/ci_dispose_v2?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\ spring.datasource.url=jdbc:mysql://10.88.77.65:33061/ci_dispose_v2?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\
# =convertToNull&useUnicode=true&characterEncoding=utf8&allowMultiQueries=true =convertToNull&useUnicode=true&characterEncoding=utf8&allowMultiQueries=true
#spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
#spring.datasource.username=root
#spring.datasource.password=h0K0_8u
spring.datasource.url=jdbc:mysql://172.21.48.75:3306/ci_dispose_v1?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\
=convertToNull&useUnicode=true
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
spring.datasource.username=root spring.datasource.username=root
spring.datasource.password=BCcf6Dd7&8 spring.datasource.password=h0K0_8u
#spring.datasource.url=jdbc:mysql://172.21.48.75:3306/ci_dispose_v1?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\
# =convertToNull&useUnicode=true
#spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
#spring.datasource.username=root
#spring.datasource.password=BCcf6Dd7&8
# 配置连接池 # 配置连接池
spring.datasource.schema=classpath:test_db/unit_test.sql spring.datasource.schema=classpath:test_db/unit_test.sql
spring.datasource.initialization-mode=always spring.datasource.initialization-mode=always

View File

@ -0,0 +1,111 @@
package com.dispose.common;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
/**
* The emos constants.
*
* @author <chenling@cmhi.chinamoblie.com>
*/
public class Constants {
/**
* DDoS具体攻击类型
*/
public static Map<String, String> ATTACK_TYPE;
static {
Map<String, String> attTypeMap = new HashMap<>();
attTypeMap.put("hosttotaltraffic", "Host total traffic|1|2");
attTypeMap.put("RSTFlood", "RST Flood|3|4");
attTypeMap.put("SYNFlood", "SYN Flood|5|6");
attTypeMap.put("ACKFlood", "ACK Flood|7|8");
attTypeMap.put("TCPnullFlood", "TCP null|9|10");
attTypeMap.put("SYNACKAmplification", "SYN/ACK Amplification|11|12");
attTypeMap.put("tcpmisuse", "TCP Misuse|13|14");
attTypeMap.put("FINFlood", "FIN Flood|15|16");
attTypeMap.put("TCPFragment", "TCP Fragment|17|18");
attTypeMap.put("HTTPFlood", "HTTP Flood|19|20");
attTypeMap.put("HTTPSFlood", "HTTPS Flood|21|22");
attTypeMap.put("SIPFlood", "SIP Flood|23|24");
attTypeMap.put("DNS", "DNS|25|26");
attTypeMap.put("UDPFragment", "UDP Fragment|27|28");
attTypeMap.put("chargenAmplification", "chargen Amplification|29|30");
attTypeMap.put("L2TPAmplification", "L2TP Amplification|31|32");
attTypeMap.put("mDNSAmplification", "mDNS Amplification|33|34");
attTypeMap.put("MSSQLRSAmplification", "MS SQL RS Amplification|35|36");
attTypeMap.put("NetBIOSAmplification", "NetBIOS Amplification|37|38");
attTypeMap.put("NTPAmplification", "NTP Amplification|39|40");
attTypeMap.put("RIPv1Amplification", "RIPv1 Amplification|41|42");
attTypeMap.put("rpcbindAmplification", "rpcbind Amplification|43|44");
attTypeMap.put("SNMPAmplification", "SNMP Amplification|45|46");
attTypeMap.put("SSDPAmplification", "SSDP Amplification|47|48");
attTypeMap.put("DNSAmplification", "DNS Amplification|49|50");
attTypeMap.put("QOTDAmplification", "QOTD Amplification|51|52");
attTypeMap.put("Quake3Amplification", "Quake3 Amplification|53|54");
attTypeMap.put("SteamAmplification", "Steam Amplification|55|56");
attTypeMap.put("CLADPAmplification", "CLADP Amplification|57|58");
attTypeMap.put("MemcacheAmplification", "Memcache Amplification|59|60");
attTypeMap.put("UDPFlood", "UDP Flood|61|62");
attTypeMap.put("smurf", "Smurf|63|64");
attTypeMap.put("icmpfrgment", "ICMP Fragment|65|66");
attTypeMap.put("ICMPFlood", "ICMP Flood|67|68");
attTypeMap.put("IPv4Protocol0", "IPv4 Protocol 0|69|70");
attTypeMap.put("IPPrivate", "IP Private|71|72");
attTypeMap.put("landflood", "Land flood|73|74");
attTypeMap.put("IGMPFlood", "IGMP Flood|75|76");
ATTACK_TYPE = Collections.unmodifiableMap(attTypeMap);
}
/**
* 各省份城市
*/
public static final String REGION_BEIJING = "北京";
public static final String REGION_SHANGHAI = "上海";
public static final String REGION_TIANJIN = "天津";
public static final String REGION_CHONGQING = "重庆";
/**
* 处置类型(1:清洗,2:黑洞,3:高防)
*/
public static final int CLEANUP = 1;
public static final int BLACKHOOL = 2;
public static final int HIDEPEND = 3;
/**
* /**
* 派单eoms模板
*/
public static final String DISPATCH_TEMPLATE = "<AlarmStart>\nMsgSerial:{0}\nSDN:{1}\nNeName:{2}\nEquipmentClass:99236\n" +
"AlarmUniqueId:{3}\nAlarmUniqueClearId:{4}\nLocateNeName:{5}\nLocateNeType:99236\nLocateNeSDN:{6}\nLocateInfo:{7}\n" +
"EventTime:{8}\nCancelTime:{9}\nVendorAlarmType:{10}\nVendorSeverity:{11}\nVendorAlarmId:{12}\nAlarmTitle:{13}\n" +
"ProbableCauseTxt:{14}\nRLocateSDN: \nRLocateNeName: \nRLocateNeType: \nRate: \nAlarmLocation: \nAlarmCheck: \n" +
"HolderType: \nAlarmStatus:{15}\nCorrelateAlarmFlag: \nAlarmActCount: \nNeIp:\nEmsId: \nVendor:99083\nAlarmText:{16}\n" +
"NeAlias: \nVersion: \nRemoteNe: \nAlarmProvince:{17}\nAlarmRegion:{18}\nAlarmCounty: \nSite: \nSiteType: \nSiteProperty: \n" +
"MachineroomIDofZGTT: \nBusinessSystem:{19}\nCircuitNo: \nMac: \nSpecialty:9\nNetworkType:903\nNeSubType: \nEffectCircuitNum: \n" +
"CircuitLevel: \nAlarmSeverity:3\nNmsAlarmId:0903-083-056-10-900001\nStandardAlarmName:{20}\nAlarmLogicClass:{21}\n" +
"AlarmLogicSubClass:{22}\nEffectOnEquipment:5\nEffectOnBusiness:4\nNmsAlarmType:1\nSendGroupFlag: \nStandardFlag:2\n" +
"AlarmExplanation:{23}\nBusinessType: \nBusinessInfo:{24}\nIsRelatedRemote: \nLocateNeStatus:1300\nProjectNo: \n" +
"ProjectName: \nProjectStartTime: \nProjectEndTime: \nGroupCustomer: \nCustomerLevel: \nServiceType: \nServiceLevel: \n" +
"ServiceName: \nServiceCrossDomainType: \nInterruptCircuitState: \nCircuitLocateInfo: \nHomeClientNum: \nHomeCellNum: \n" +
"LinkOnuNum: \n<AlarmEnd>";
/* *
0网元自动清除 --收到从采集源发送的清除告警
*/
public static final int NE_AUTO_CLEARED_STATUS = 0;
/* *
1活动告警---告警当前为活动状态
*/
public static final int ACTIVE_ALARM_STATUS = 1;
/* *
2同步清除---已采集活动告警但采集平台从告警源同步时发现已经没有对应的活动告警由采集平台产生的清除告警
*/
public static final int SYNCHRONIZATION_CLEAR_STATUS = 2;
}

View File

@ -293,6 +293,10 @@ public enum ErrorCode {
* The Err huawei firewall error. * The Err huawei firewall error.
*/ */
ERR_HUAWEIFIREWALL_ERROR(305, "华为防火墙返回错误"), ERR_HUAWEIFIREWALL_ERROR(305, "华为防火墙返回错误"),
/**
* The Err emos create message error.
*/
EMOS_CREATEMESSAGE_ERROR(306, "EMOS发送信息错误"),
; ;
/** /**

View File

@ -1,21 +1,26 @@
package com.dispose.controller; package com.dispose.controller;
import com.alibaba.fastjson.JSONObject;
import com.dispose.common.Constants;
import com.dispose.common.ErrorCode; import com.dispose.common.ErrorCode;
import com.dispose.config.KafkaConfiguration; import com.dispose.config.KafkaConfiguration;
import com.dispose.manager.AlarmInfoManager;
import com.dispose.pojo.dto.protocol.base.BaseRespStatus; import com.dispose.pojo.dto.protocol.base.BaseRespStatus;
import com.dispose.pojo.dto.protocol.base.ProtocolReqDTO; import com.dispose.pojo.dto.protocol.base.ProtocolReqDTO;
import com.dispose.pojo.dto.protocol.base.ProtocolRespDTO; import com.dispose.pojo.dto.protocol.base.ProtocolRespDTO;
import com.dispose.pojo.dto.protocol.kafka.AlarmInfo; import com.dispose.pojo.dto.protocol.kafka.EmosAlarmInfo;
import com.dispose.pojo.dto.protocol.kafka.AlarmInfoReq; import com.dispose.pojo.dto.protocol.kafka.AlarmInfoReq;
import com.dispose.pojo.entity.AlarmInformation;
import com.dispose.pojo.entity.MsgSerial;
import com.dispose.pojo.po.MulReturnType;
import com.dispose.security.annotation.Decryption; import com.dispose.security.annotation.Decryption;
import com.dispose.security.annotation.Encryption; import com.dispose.security.annotation.Encryption;
import com.dispose.service.MsgSerialService;
import com.dispose.validation.group.ValidGroups; import com.dispose.validation.group.ValidGroups;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import io.swagger.annotations.Api; import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation; import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.exception.ExceptionUtils;
import org.springframework.kafka.support.SendResult; import org.springframework.kafka.support.SendResult;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.springframework.stereotype.Controller; import org.springframework.stereotype.Controller;
@ -27,12 +32,13 @@ import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.bind.annotation.ResponseBody;
import javax.annotation.Resource; import javax.annotation.Resource;
import java.text.MessageFormat;
import java.util.Objects; import java.util.Objects;
/** /**
* The type Auth controller. * The type Auth controller.
* *
* @author <huangxin@cmhi.chinamoblie.com> * @author <chenling@cmhi.chinamoblie.com>
*/ */
@Controller @Controller
@RequestMapping(value = "/kafka") @RequestMapping(value = "/kafka")
@ -43,20 +49,24 @@ import java.util.Objects;
@Encryption @Encryption
@Decryption @Decryption
public class kafkaController { public class kafkaController {
/**
* The Object mapper.
*/
@Resource
private ObjectMapper objectMapper;
/** /**
* The Kafka configuration. * The Kafka configuration.
*/ */
// final private KafkaConfiguration kafkaConfiguration =
// SpringBootBeanUtil.getBean(com.dispose.config.KafkaConfiguration.class);
@Resource @Resource
private KafkaConfiguration kafkaConfiguration; private KafkaConfiguration kafkaConfiguration;
/**
* The message serial service.
*/
@Resource
private MsgSerialService msgSerialService;
/**
* The alarm information manager.
*/
@Resource
private AlarmInfoManager alarmInfoManager;
/** /**
* Dispatch command sent to kafka. * Dispatch command sent to kafka.
* *
@ -68,41 +78,177 @@ public class kafkaController {
@ApiOperation("发送消息") @ApiOperation("发送消息")
public ProtocolRespDTO<BaseRespStatus> dispatchCommand( public ProtocolRespDTO<BaseRespStatus> dispatchCommand(
@Validated(ValidGroups.ProtocolCommonValid.class) @Validated(ValidGroups.ProtocolCommonValid.class)
@RequestBody ProtocolReqDTO<AlarmInfoReq> mr) throws JsonProcessingException { @RequestBody ProtocolReqDTO<AlarmInfoReq> mr) {
//获取入参信息进行所需数据格式拼接 //获取入参信息
log.info("alarm message information :{}", mr.getMsgContent().getAlarmInfo()); BaseRespStatus rspInfo = new BaseRespStatus();
AlarmInfo alarmInfo = objectMapper.readValue(mr.getMsgContent().getAlarmInfo(), new TypeReference<AlarmInfo>() { log.info("emos alarm is:{}", mr.getMsgContent().getAlarmInfo());
}); EmosAlarmInfo alarmInfo = JSONObject.parseObject(mr.getMsgContent().getAlarmInfo(), EmosAlarmInfo.class);
//拼接emos据格式
String content = createSendContent(alarmInfo);
if (content == null) {
rspInfo.setStatus(ErrorCode.EMOS_CREATEMESSAGE_ERROR.getCode());
rspInfo.setMessage(new String[]{ErrorCode.EMOS_CREATEMESSAGE_ERROR.getMsg()});
return ProtocolRespDTO.result(ErrorCode.EMOS_CREATEMESSAGE_ERROR, rspInfo);
}
//保存数据格式到数据库
log.info("send alarm :{}", content);
AlarmInformation alarmInformation = AlarmInformation.builder().alarmInfo(content).build();
alarmInfoManager.addAlarmInfo(alarmInformation);
//推动数据格式到kafka //推动数据格式到kafka
log.info("send alarm message :{}", alarmInfo.toString());
String sendMessage = mr.getMsgContent().getAlarmInfo();
ListenableFuture<SendResult<String, String>> sendResult = kafkaConfiguration ListenableFuture<SendResult<String, String>> sendResult = kafkaConfiguration
.kafkaTemplate() .kafkaTemplate()
.sendDefault(0, System.currentTimeMillis(), "dispose", sendMessage); .sendDefault(0, System.currentTimeMillis(), "dispose", content);
sendResult.addCallback(v -> log.info("Kafka send {} to {} at {}", sendMessage, sendResult.addCallback(v -> log.info("Kafka send {} to {} at {}", content,
Objects.requireNonNull(v) Objects.requireNonNull(v)
.getRecordMetadata() .getRecordMetadata()
.topic(), v.getRecordMetadata().partition()), .topic(), v.getRecordMetadata().partition()),
ex -> log.error("Kafka send error: {}", ex.getMessage())); ex -> log.error("Kafka send error: {}", ex.getMessage()));
BaseRespStatus rspInfo = new BaseRespStatus();
rspInfo.setStatus(ErrorCode.ERR_OK.getCode()); rspInfo.setStatus(ErrorCode.ERR_OK.getCode());
rspInfo.setMessage(new String[]{ErrorCode.ERR_OK.getMsg()}); rspInfo.setMessage(new String[]{ErrorCode.ERR_OK.getMsg()});
return ProtocolRespDTO.result(ErrorCode.ERR_OK, rspInfo); return ProtocolRespDTO.result(ErrorCode.ERR_OK, rspInfo);
} }
/**
* @param alarmInfo emos告警信息
* @return 发送的消息
*/
private String createSendContent(EmosAlarmInfo alarmInfo) {
try {
long dbIncrement = msgSerialService.getMaxMessageSerial();
long increment = dbIncrement + 1;
MsgSerial msgSerial = MsgSerial.builder().msgSerial(increment).build();
MulReturnType<ErrorCode, Long> returnType = msgSerialService.addMessageSerial(msgSerial);
if (returnType.getFirstParam() == ErrorCode.ERR_OK) {
increment = returnType.getSecondParam();
} else {
increment = dbIncrement;
}
// @KafkaListener(topics = {"ddos-vip-customer-ck"}) String dstIp = alarmInfo.getDstIp();
// public void kafkaListen(ConsumerRecord<String, String> consumerRecord) { String alarmId = alarmInfo.getAlarmId();
// //判断消息是否为null //LocateInfo: 192.168.11.6 ddos attack alarmMemcache , 2019-08-01 12:31, 2.1Gbps, 1.01Mpps
// Optional<String> kafkaMessage = Optional.ofNullable(consumerRecord.value()); String locateInfo = getAlarmEvent(alarmInfo);
// log.info(">>>>>>>>>>> record = " + kafkaMessage); String eventTime = alarmInfo.getStartTime();
// if (kafkaMessage.isPresent()) { String cancelTime = alarmInfo.getEndTime();
// String consumerMsg = kafkaMessage.get();
// log.info("消费消息:" + consumerMsg);
// }
// }
//告警类型vendorAlarmType告警级别vendorSeverity告警号vendorAlarmId告警标题AlarmTitle告警可能原因ProbableCauseTxt
String type = Constants.ATTACK_TYPE.get(alarmInfo.getAttackType());
String vendorSeverity = characterEncode("一级");
String vendorAlarmType;
String vendorAlarmId;
if (type != null) {
String[] arr = type.split("\\|");
vendorAlarmType = arr[0];
if (alarmInfo.getBpspps().compareTo("bps") == 0) {
vendorAlarmId = arr[1];
} else {
vendorAlarmId = arr[2];
}
} else {
vendorAlarmType = alarmInfo.getAttackType();
vendorAlarmId = "0";
log.info("unKnown attack type:" + vendorAlarmType);
}
String alarmStatus = String.valueOf(Constants.ACTIVE_ALARM_STATUS);
String alarmText = characterEncode(getAlarmText(alarmInfo));
String alarmExplanation = characterEncode(getAlarmExplanation(alarmInfo));
String content = MessageFormat.format(Constants.DISPATCH_TEMPLATE, increment, dstIp,
dstIp, alarmId, alarmId, dstIp, dstIp, locateInfo, eventTime, cancelTime,
vendorAlarmType, vendorSeverity, vendorAlarmId, characterEncode("重保攻击事件告警"),
characterEncode("DDos攻击事件"), alarmStatus, alarmText,
characterEncode(alarmInfo.getDstProvince()), characterEncode(alarmInfo.getDstCity()),
characterEncode("网络部集中抗D系统"), characterEncode("DDOS攻击事件告警"),
characterEncode("安全告警"), characterEncode("DDOS告警"),
alarmExplanation, characterEncode("集中抗D"));
return content;
} catch (Exception e) {
log.error("createSendContent告警消息异常详细信息{}", ExceptionUtils.getStackTrace(e));
return null;
}
}
/**
* 获取告警事件AlarmEvent
*/
private String getAlarmEvent(EmosAlarmInfo a) {
return a.getDstIp() + " ddos attack alarm, " + a.getAttackType() + ", " + a.getStartTime() +
", " + a.getMaxBps() + ", " + a.getMaxPps();
}
/**
* 数据编码都采用GBK编码方式
*/
private String characterEncode(String character) {
try {
return new String(character.getBytes("GBK"), "GBK");
} catch (Exception e) {
log.info("character encoding failed:" + e.getMessage());
return character;
}
}
/**
* 获取告警区域
*/
private String getAreaDes(String province, String city) {
String areaDes;
if (Constants.REGION_BEIJING.equals(province)) {
areaDes = "北京市";
} else if (Constants.REGION_SHANGHAI.equals(province)) {
areaDes = "上海市";
} else if (Constants.REGION_TIANJIN.equals(province)) {
areaDes = "天津市";
} else if (Constants.REGION_CHONGQING.equals(province)) {
areaDes = "重庆市";
} else {
areaDes = province + "" + city + "";
}
return areaDes;
}
/**
* 获取处置类型(1:清洗,2:黑洞,3:高防)
*/
private String getOperateType(Integer disposeType) {
String operateType = null;
if (Constants.CLEANUP == disposeType) {
operateType = "清洗";
} else if (Constants.BLACKHOOL == disposeType) {
operateType = "流控";
} else if (Constants.HIDEPEND == disposeType) {
operateType = "黑洞";
}
return operateType;
}
/**
* 获取告警正文AlarmText
*/
private String getAlarmText(EmosAlarmInfo a) {
String area = getAreaDes(a.getDstProvince(), a.getDstCity());
String operateType = getOperateType(a.getDisposeType());
return "攻击目的IP" + a.getDstIp() + "," + area + "," + "处置操作:" + operateType + "," + "处置时长:" + a.getDisposeTime() + "分钟";
}
/**
* 获取告警解释AlarmExplanation
*/
private String getAlarmExplanation(EmosAlarmInfo a) {
String operateType = getOperateType(a.getDisposeType());
StringBuilder srcIp = new StringBuilder();
for (String ip : a.getSrcIpLs()) {
srcIp.append(ip).append(",");
}
if (srcIp.length() > 0) {
srcIp = new StringBuilder(srcIp.substring(0, srcIp.length() - 1));
}
return "攻击目的IP" + a.getDstIp() + "," + "攻击源地址:(" + srcIp + ")," + "处置操作:" + operateType + "," + "处置时长:" + a.getDisposeTime() + "分钟";
}
} }

View File

@ -0,0 +1,17 @@
package com.dispose.manager;
import com.dispose.pojo.entity.AlarmInformation;
/**
* The interface alarm information manager.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
public interface AlarmInfoManager {
/**
* Add alarm information.
*
* @param alarmInformation the alarm information
*/
void addAlarmInfo(AlarmInformation alarmInformation);
}

View File

@ -0,0 +1,21 @@
package com.dispose.manager;
import com.dispose.common.ErrorCode;
import com.dispose.pojo.entity.MsgSerial;
public interface MsgSerialManager {
/**
* Add user business error code.
*
* @param msgSerial the message serial
* @return the error code
*/
ErrorCode addMsgSerialNumber(MsgSerial msgSerial);
/**
* get new max message serial.
*
* @return the long
*/
Long getMaxMsgSerial();
}

View File

@ -0,0 +1,34 @@
package com.dispose.manager.impl;
import com.dispose.manager.AlarmInfoManager;
import com.dispose.mapper.AlarmInformationMapper;
import com.dispose.pojo.entity.AlarmInformation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
/**
* The interface alarm information manager.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
@Component
@Slf4j
public class AlarmInfoManagerImpl implements AlarmInfoManager {
/**
* The alarm information mapper.
*/
@Resource
private AlarmInformationMapper alarmInformationMapper;
/**
* Add alarm information.
*
* @param alarmInformation the alarm information
*/
@Override
public void addAlarmInfo(AlarmInformation alarmInformation) {
alarmInformationMapper.addAlarmInfo(alarmInformation);
}
}

View File

@ -0,0 +1,51 @@
package com.dispose.manager.impl;
import com.dispose.common.ErrorCode;
import com.dispose.manager.MsgSerialManager;
import com.dispose.mapper.MsgSerialMapper;
import com.dispose.pojo.entity.MsgSerial;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
/**
* The interface message serial manager.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
@Component
@Slf4j
public class MsgSerialManagerImpl implements MsgSerialManager {
/**
* The message serial mapper.
*/
@Resource
private MsgSerialMapper msgSerialMapper;
/**
* Add user business error code.
*
* @param msgSerial the message serial
* @return the error code
*/
@Override
public ErrorCode addMsgSerialNumber(MsgSerial msgSerial) {
if (msgSerialMapper.addMsgSerial(msgSerial) == 1) {
return ErrorCode.ERR_OK;
} else {
return ErrorCode.ERR_DATABASE;
}
}
/**
* get new max message serial.
*
* @return the long
*/
@Override
public Long getMaxMsgSerial() {
return msgSerialMapper.getMaxMsgSerial();
}
}

View File

@ -0,0 +1,18 @@
package com.dispose.mapper;
import com.dispose.pojo.entity.AlarmInformation;
/**
* The interface alarm information mapper.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
public interface AlarmInformationMapper {
/**
* Add alarm information.
*
* @param alarmInformation the alarm information
* @return the int
*/
int addAlarmInfo(AlarmInformation alarmInformation);
}

View File

@ -0,0 +1,26 @@
package com.dispose.mapper;
import com.dispose.pojo.entity.MsgSerial;
/**
* The interface message serial mapper.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
public interface MsgSerialMapper {
/**
* Add new task int.
*
* @param msgSerial the msgSerial number
* @return the int
*/
int addMsgSerial(MsgSerial msgSerial);
/**
* get new max message serial.
*
* @return the long
*/
long getMaxMsgSerial();
}

View File

@ -16,7 +16,7 @@ import java.util.List;
@JsonPropertyOrder({"alarmId", "dstIp", "attackType", "bpspps", "dstProvince", "dstCity", "srcIpLs", "startTime", @JsonPropertyOrder({"alarmId", "dstIp", "attackType", "bpspps", "dstProvince", "dstCity", "srcIpLs", "startTime",
"endTime", "disposeType", "disposeTime", "maxBps", "maxPps"}) "endTime", "disposeType", "disposeTime", "maxBps", "maxPps"})
@JsonInclude(JsonInclude.Include.NON_NULL) @JsonInclude(JsonInclude.Include.NON_NULL)
public class AlarmInfo { public class EmosAlarmInfo {
/** /**
* 告警id. * 告警id.
*/ */

View File

@ -0,0 +1,51 @@
package com.dispose.pojo.entity;
import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import tk.mybatis.mapper.annotation.KeySql;
import tk.mybatis.mapper.annotation.NameStyle;
import tk.mybatis.mapper.code.Style;
import javax.persistence.Id;
import javax.persistence.Table;
import java.io.Serializable;
/**
* The emos alarm information.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
@JsonInclude(JsonInclude.Include.NON_NULL)
@Table(name = "alarm_information")
@NameStyle(Style.normal)
public class AlarmInformation implements Serializable {
/**
* The constant serialVersionUID.
*/
private static final long serialVersionUID = 1L;
/**
* The id.
*/
@Id
@KeySql(useGeneratedKeys = true)
private Long id;
/**
* The alarm information.
*/
private String alarmInfo;
/**
* The creating time.
*/
private String createTime;
}

View File

@ -0,0 +1,46 @@
package com.dispose.pojo.entity;
import com.fasterxml.jackson.annotation.JsonInclude;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import tk.mybatis.mapper.annotation.KeySql;
import tk.mybatis.mapper.annotation.NameStyle;
import tk.mybatis.mapper.code.Style;
import javax.persistence.Id;
import javax.persistence.Table;
import java.io.Serializable;
/**
* The message serial.
*
* @author <chenlinghy@cmhi.chinamoblie.com>
*/
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
@JsonInclude(JsonInclude.Include.NON_NULL)
@Table(name = "msg_serial")
@NameStyle(Style.normal)
public class MsgSerial implements Serializable {
/**
* The constant serialVersionUID.
*/
private static final long serialVersionUID = 1L;
/**
* The id.
*/
@Id
@KeySql(useGeneratedKeys = true)
private Long id;
/**
* The message serial.
*/
private Long msgSerial;
}

View File

@ -0,0 +1,23 @@
package com.dispose.service;
import com.dispose.common.ErrorCode;
import com.dispose.pojo.entity.MsgSerial;
import com.dispose.pojo.po.MulReturnType;
public interface MsgSerialService {
/**
* add message serial.
*
* @param msgSerial the message serial
* @return the mul return type
*/
MulReturnType<ErrorCode, Long> addMessageSerial(MsgSerial msgSerial);
/**
* get new max message serial.
*
* @return the long
*/
Long getMaxMessageSerial();
}

View File

@ -0,0 +1,50 @@
package com.dispose.service.impl;
import com.dispose.common.ErrorCode;
import com.dispose.manager.MsgSerialManager;
import com.dispose.pojo.entity.MsgSerial;
import com.dispose.pojo.po.MulReturnType;
import com.dispose.service.MsgSerialService;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
@Service
public class MsgSerialServiceImpl implements MsgSerialService {
@Resource
private MsgSerialManager msgSerialManager;
/**
* add message serial.
*
* @param msgSerial the message serial
* @return the mul return type
*/
@Override
public MulReturnType<ErrorCode, Long> addMessageSerial(MsgSerial msgSerial) {
//告警序号的最大值2^32-1
long indexEnd = 4294967295L;
long indexStart = 1L;
long currentSerial = msgSerial.getMsgSerial();
//编号从1开始以实时消息发布通道为单位进行编号如果编号超过最大正整数(2^32-1)重新从1开始编号
if (currentSerial > indexEnd) {
currentSerial = indexStart;
}
if (msgSerialManager.addMsgSerialNumber(msgSerial) == ErrorCode.ERR_OK) {
return new MulReturnType<>(ErrorCode.ERR_OK, currentSerial);
} else {
return new MulReturnType<>(ErrorCode.ERR_DATABASE, null);
}
}
/**
* get new max message serial.
*
* @return the long
*/
@Override
public Long getMaxMessageSerial() {
return msgSerialManager.getMaxMsgSerial();
}
}

View File

@ -1,79 +0,0 @@
package com.dispose.setup;
import lombok.extern.slf4j.Slf4j;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
/**
* The type Spring boot bean util.
*/
@Component
@Slf4j
public class SpringBootBeanUtil implements ApplicationContextAware {
/**
* The constant applicationContext.
*/
private static ApplicationContext applicationContext;
/**
* Sets application context.
*
* @param applicationContext the application context
* @throws BeansException the beans exception
*/
@Override
public void setApplicationContext(@NotNull ApplicationContext applicationContext) throws BeansException {
if (SpringBootBeanUtil.applicationContext == null) {
SpringBootBeanUtil.applicationContext = applicationContext;
}
log.debug("========ApplicationContext配置成功========");
log.debug("========在普通类可以通过调用SpringUtils.getAppContext()获取applicationContext对象========");
log.debug("========applicationContext=" + SpringBootBeanUtil.applicationContext + "========");
}
/**
* Gets application context.
*
* @return the application context
*/
public static ApplicationContext getApplicationContext() {
return applicationContext;
}
/**
* Gets bean.
*
* @param name the name
* @return the bean
*/
public static Object getBean(String name) {
return getApplicationContext().getBean(name);
}
/**
* Gets bean.
*
* @param <T> the type parameter
* @param clazz the clazz
* @return the bean
*/
public static <T> T getBean(Class<T> clazz) {
return getApplicationContext().getBean(clazz);
}
/**
* Gets bean.
*
* @param <T> the type parameter
* @param name the name
* @param clazz the clazz
* @return the bean
*/
public static <T> T getBean(String name, Class<T> clazz) {
return getApplicationContext().getBean(name, clazz);
}
}

View File

@ -9,3 +9,11 @@ INSERT INTO `user_account`
VALUES (1, 'admin', 'c3855e6b6bb120450f160ba91134522868f89d36062f2061ebeefd80817e1d58', '2020-11-13 09:25:19', '', VALUES (1, 'admin', 'c3855e6b6bb120450f160ba91134522868f89d36062f2061ebeefd80817e1d58', '2020-11-13 09:25:19', '',
'2021-01-20 10:18:56', '1db9ddc47de514eb16b7ec07d7f7f96f7a714ae00e1209755bab30d543a0a2c3', '2021-01-20 10:18:56', '1db9ddc47de514eb16b7ec07d7f7f96f7a714ae00e1209755bab30d543a0a2c3',
'2021-01-20 10:20:57', '1970-01-02 00:00:00', 0, 0); '2021-01-20 10:20:57', '1970-01-02 00:00:00', 0, 0);
-- ----------------------------
-- Records of msg_serial
-- ----------------------------
INSERT INTO `msg_serial` VALUES ('1', '1');

View File

@ -15,7 +15,8 @@
*/ */
SET NAMES utf8mb4; SET NAMES utf8mb4;
SET FOREIGN_KEY_CHECKS = 0; SET
FOREIGN_KEY_CHECKS = 0;
-- ---------------------------- -- ----------------------------
-- Table structure for device_task -- Table structure for device_task
@ -23,23 +24,23 @@ SET FOREIGN_KEY_CHECKS = 0;
DROP TABLE IF EXISTS `device_task`; DROP TABLE IF EXISTS `device_task`;
CREATE TABLE `device_task` CREATE TABLE `device_task`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '任务信息唯一标识符', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '任务信息唯一标识符',
`taskId` bigint(11) NOT NULL COMMENT '处置任务唯一标识符', `taskId` bigint(11) NOT NULL COMMENT '处置任务唯一标识符',
`deviceId` bigint(11) NOT NULL COMMENT '处置设备唯一标识符', `deviceId` bigint(11) NOT NULL COMMENT '处置设备唯一标识符',
`beginTime` timestamp(0) NULL DEFAULT NULL COMMENT '任务开始时间', `beginTime` timestamp(0) NULL DEFAULT NULL COMMENT '任务开始时间',
`endTime` timestamp(0) NULL DEFAULT NULL COMMENT '任务结束时间', `endTime` timestamp(0) NULL DEFAULT NULL COMMENT '任务结束时间',
`taskAttackType` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '需要处置的攻击类型', `taskAttackType` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '需要处置的攻击类型',
`execAttackTypeIn` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '已经执行处置的攻击类型(Input)', `execAttackTypeIn` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '已经执行处置的攻击类型(Input)',
`attackTypeStatusIn` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '执行的攻击类型状态(Input)', `attackTypeStatusIn` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '执行的攻击类型状态(Input)',
`execAttackTypeOut` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '已经执行处置的攻击类型(Output)', `execAttackTypeOut` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '已经执行处置的攻击类型(Output)',
`attackTypeStatusOut` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '执行的攻击类型状态(Input)', `attackTypeStatusOut` bigint(255) UNSIGNED NOT NULL DEFAULT 0 COMMENT '执行的攻击类型状态(Input)',
`externId` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '扩展任务ID', `externId` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '扩展任务ID',
`errRetry` int(11) UNSIGNED NOT NULL DEFAULT 0 COMMENT '调用失败重试次数', `errRetry` int(11) UNSIGNED NOT NULL DEFAULT 0 COMMENT '调用失败重试次数',
`status` int(11) NOT NULL DEFAULT 0 COMMENT '任务状态', `status` int(11) NOT NULL DEFAULT 0 COMMENT '任务状态',
`devStatus` int(11) NOT NULL DEFAULT 0 COMMENT '设备任务状态', `devStatus` int(11) NOT NULL DEFAULT 0 COMMENT '设备任务状态',
PRIMARY KEY (`id`) USING BTREE, PRIMARY KEY (`id`) USING BTREE,
INDEX `task_info_ibfk_1` (`taskId`) USING BTREE, INDEX `task_info_ibfk_1` (`taskId`) USING BTREE,
INDEX `task_info_ibfk_2` (`deviceId`) USING BTREE, INDEX `task_info_ibfk_2` (`deviceId`) USING BTREE,
CONSTRAINT `device_task_ibfk_1` FOREIGN KEY (`taskId`) REFERENCES `dispose_task` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION, CONSTRAINT `device_task_ibfk_1` FOREIGN KEY (`taskId`) REFERENCES `dispose_task` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION,
CONSTRAINT `device_task_ibfk_2` FOREIGN KEY (`deviceId`) REFERENCES `dispose_device` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION CONSTRAINT `device_task_ibfk_2` FOREIGN KEY (`deviceId`) REFERENCES `dispose_device` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION
) ENGINE = InnoDB ) ENGINE = InnoDB
@ -54,15 +55,15 @@ CREATE TABLE `device_task`
DROP TABLE IF EXISTS `dispose_capacity`; DROP TABLE IF EXISTS `dispose_capacity`;
CREATE TABLE `dispose_capacity` CREATE TABLE `dispose_capacity`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '设备处置能力唯一标识符', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '设备处置能力唯一标识符',
`deviceId` bigint(11) NOT NULL COMMENT '设备ID', `deviceId` bigint(11) NOT NULL COMMENT '设备ID',
`capacityType` int(11) NOT NULL COMMENT '处置能力:\r\n0清洗\r\n1高防 \r\n2路由黑洞 \r\n3检测\r\n4WAF封堵\r\n', `capacityType` int(11) NOT NULL COMMENT '处置能力:\r\n0清洗\r\n1高防 \r\n2路由黑洞 \r\n3检测\r\n4WAF封堵\r\n',
`objectType` int(11) NOT NULL COMMENT '处置对象类型:\r\n0IP\r\n1域名 \r\n2URL', `objectType` int(11) NOT NULL COMMENT '处置对象类型:\r\n0IP\r\n1域名 \r\n2URL',
`ipType` int(11) NOT NULL DEFAULT 3 COMMENT 'IP 地址类型:\r\n1 << 0支持IPV4\r\n1 << 1支持IPV6', `ipType` int(11) NOT NULL DEFAULT 3 COMMENT 'IP 地址类型:\r\n1 << 0支持IPV4\r\n1 << 1支持IPV6',
`protectIp` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '该处置能力能够处置的IP', `protectIp` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '该处置能力能够处置的IP',
`reserveNetflow` int(11) NULL DEFAULT NULL COMMENT '清洗能力储备流量值,单位(G)', `reserveNetflow` int(11) NULL DEFAULT NULL COMMENT '清洗能力储备流量值,单位(G)',
PRIMARY KEY (`id`) USING BTREE, PRIMARY KEY (`id`) USING BTREE,
INDEX `id` (`deviceId`) USING BTREE, INDEX `id` (`deviceId`) USING BTREE,
CONSTRAINT `dispose_capacity_ibfk_1` FOREIGN KEY (`deviceId`) REFERENCES `dispose_device` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION CONSTRAINT `dispose_capacity_ibfk_1` FOREIGN KEY (`deviceId`) REFERENCES `dispose_device` (`id`) ON DELETE NO ACTION ON UPDATE NO ACTION
) ENGINE = InnoDB ) ENGINE = InnoDB
AUTO_INCREMENT = 1 AUTO_INCREMENT = 1
@ -76,23 +77,23 @@ CREATE TABLE `dispose_capacity`
DROP TABLE IF EXISTS `dispose_device`; DROP TABLE IF EXISTS `dispose_device`;
CREATE TABLE `dispose_device` CREATE TABLE `dispose_device`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '设备唯一标识符', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '设备唯一标识符',
`ipAddr` varchar(64) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '设备IP地址, IPv4/IPv6', `ipAddr` varchar(64) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '设备IP地址, IPv4/IPv6',
`ipPort` varchar(64) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '端口', `ipPort` varchar(64) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '端口',
`deviceType` int(11) NOT NULL DEFAULT 0 COMMENT '能力节点类型0迪普UMC平台1浩瀚设备', `deviceType` int(11) NOT NULL DEFAULT 0 COMMENT '能力节点类型0迪普UMC平台1浩瀚设备',
`areaCode` int(11) NOT NULL COMMENT '设备物理位置区域码', `areaCode` int(11) NOT NULL COMMENT '设备物理位置区域码',
`deviceName` varchar(256) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '设备名称', `deviceName` varchar(256) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '设备名称',
`manufacturer` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '生产厂商', `manufacturer` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '生产厂商',
`model` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '型号', `model` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '型号',
`version` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '软件版本', `version` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT NULL COMMENT '软件版本',
`userName` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '系统管理用户名', `userName` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '系统管理用户名',
`password` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '系统管理密码', `password` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT '系统管理密码',
`urlPath` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT 'URL路径', `urlPath` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL DEFAULT '' COMMENT 'URL路径',
`urlType` int(11) NOT NULL DEFAULT 0 COMMENT 'URL类型: 0: HTTP, 1: HTTPS', `urlType` int(11) NOT NULL DEFAULT 0 COMMENT 'URL类型: 0: HTTP, 1: HTTPS',
`readme` varchar(1024) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '备注', `readme` varchar(1024) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '备注',
`status` int(11) NOT NULL DEFAULT 0 COMMENT '状态0正常1锁定2禁用 3删除', `status` int(11) NOT NULL DEFAULT 0 COMMENT '状态0正常1锁定2禁用 3删除',
PRIMARY KEY (`id`) USING BTREE, PRIMARY KEY (`id`) USING BTREE,
INDEX `ipAddr` (`ipAddr`) USING BTREE INDEX `ipAddr` (`ipAddr`) USING BTREE
) ENGINE = InnoDB ) ENGINE = InnoDB
AUTO_INCREMENT = 1 AUTO_INCREMENT = 1
CHARACTER SET = utf8 CHARACTER SET = utf8
@ -105,23 +106,23 @@ CREATE TABLE `dispose_device`
DROP TABLE IF EXISTS `dispose_task`; DROP TABLE IF EXISTS `dispose_task`;
CREATE TABLE `dispose_task` CREATE TABLE `dispose_task`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '处置任务唯一标识符', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '处置任务唯一标识符',
`deviceId` bigint(11) NOT NULL COMMENT '处置设备唯一标识符', `deviceId` bigint(11) NOT NULL COMMENT '处置设备唯一标识符',
`accountId` bigint(11) NOT NULL COMMENT '用户唯一标识符', `accountId` bigint(11) NOT NULL COMMENT '用户唯一标识符',
`areaCode` int(11) NOT NULL DEFAULT -1 COMMENT '设备物理位置区域码', `areaCode` int(11) NOT NULL DEFAULT -1 COMMENT '设备物理位置区域码',
`disposeCapacity` int(11) NOT NULL COMMENT '处置能力类型0清洗 1高防 2黑洞 3检测', `disposeCapacity` int(11) NOT NULL COMMENT '处置能力类型0清洗 1高防 2黑洞 3检测',
`disposeObject` varchar(4096) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '处置对象', `disposeObject` varchar(4096) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '处置对象',
`objectType` int(8) NOT NULL COMMENT '处置对象类型:\r\n0IP\r\n1域名 \r\n2URL', `objectType` int(8) NOT NULL COMMENT '处置对象类型:\r\n0IP\r\n1域名 \r\n2URL',
`createTime` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '开始时间', `createTime` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '开始时间',
`planEndTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '计划结束时间', `planEndTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '计划结束时间',
`endTime` timestamp(0) NULL DEFAULT '1970-01-02 00:00:00' COMMENT '实际结束时间', `endTime` timestamp(0) NULL DEFAULT '1970-01-02 00:00:00' COMMENT '实际结束时间',
`flowDirection` int(11) NOT NULL DEFAULT 2 COMMENT '流量方向, 0流入1流出2双向', `flowDirection` int(11) NOT NULL DEFAULT 2 COMMENT '流量方向, 0流入1流出2双向',
`attackType` bigint(20) NOT NULL DEFAULT 0 COMMENT '攻击类型默认0 全部攻击', `attackType` bigint(20) NOT NULL DEFAULT 0 COMMENT '攻击类型默认0 全部攻击',
`flowBandWidth` int(11) UNSIGNED NULL DEFAULT 1024 COMMENT '攻击流量占用带宽(MB)', `flowBandWidth` int(11) UNSIGNED NULL DEFAULT 1024 COMMENT '攻击流量占用带宽(MB)',
`currentStatus` int(11) NOT NULL DEFAULT 0 COMMENT '状态0停止。1启动', `currentStatus` int(11) NOT NULL DEFAULT 0 COMMENT '状态0停止。1启动',
PRIMARY KEY (`id`) USING BTREE, PRIMARY KEY (`id`) USING BTREE,
INDEX `dispose_task_device_capacity_id_fk` (`disposeCapacity`) USING BTREE, INDEX `dispose_task_device_capacity_id_fk` (`disposeCapacity`) USING BTREE,
INDEX `dispose_task_user_account_id_fk` (`accountId`) USING BTREE INDEX `dispose_task_user_account_id_fk` (`accountId`) USING BTREE
) ENGINE = InnoDB ) ENGINE = InnoDB
AUTO_INCREMENT = 1 AUTO_INCREMENT = 1
CHARACTER SET = utf8 CHARACTER SET = utf8
@ -134,10 +135,10 @@ CREATE TABLE `dispose_task`
DROP TABLE IF EXISTS `service_group`; DROP TABLE IF EXISTS `service_group`;
CREATE TABLE `service_group` CREATE TABLE `service_group`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '业务ID唯一标识符', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '业务ID唯一标识符',
`serviceId` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务ID', `serviceId` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务ID',
`serviceType` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务类型SERVER,WEB,DNS,GAME', `serviceType` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务类型SERVER,WEB,DNS,GAME',
`serviceBandwidth` int(11) NOT NULL DEFAULT 1 COMMENT '业务带宽单位M', `serviceBandwidth` int(11) NOT NULL DEFAULT 1 COMMENT '业务带宽单位M',
`serviceIp` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务IP地址逗号分割', `serviceIp` varchar(255) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '业务IP地址逗号分割',
PRIMARY KEY (`id`) USING BTREE PRIMARY KEY (`id`) USING BTREE
) ENGINE = InnoDB ) ENGINE = InnoDB
@ -152,17 +153,17 @@ CREATE TABLE `service_group`
DROP TABLE IF EXISTS `user_account`; DROP TABLE IF EXISTS `user_account`;
CREATE TABLE `user_account` CREATE TABLE `user_account`
( (
`id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '账户唯一编号', `id` bigint(11) NOT NULL AUTO_INCREMENT COMMENT '账户唯一编号',
`username` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '用户名', `username` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '用户名',
`password` varchar(512) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '密码', `password` varchar(512) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '密码',
`createTime` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '创建时间', `createTime` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '创建时间',
`operators` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '操作员', `operators` varchar(128) CHARACTER SET utf8 COLLATE utf8_general_ci NOT NULL COMMENT '操作员',
`lastLoginTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '最后一次成功登录时间', `lastLoginTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '最后一次成功登录时间',
`token` varchar(512) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '用户token', `token` varchar(512) CHARACTER SET utf8 COLLATE utf8_general_ci NULL DEFAULT '' COMMENT '用户token',
`lastAccess` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '最后一次访问时间戳', `lastAccess` timestamp(0) NOT NULL DEFAULT CURRENT_TIMESTAMP(0) COMMENT '最后一次访问时间戳',
`lockTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '账户锁定时间', `lockTime` timestamp(0) NOT NULL DEFAULT '1970-01-02 00:00:00' COMMENT '账户锁定时间',
`pwdErrTimes` int(11) NOT NULL DEFAULT 0 COMMENT '密码错误次数', `pwdErrTimes` int(11) NOT NULL DEFAULT 0 COMMENT '密码错误次数',
`status` int(11) NULL DEFAULT 0 COMMENT '账户状态', `status` int(11) NULL DEFAULT 0 COMMENT '账户状态',
PRIMARY KEY (`id`) USING BTREE, PRIMARY KEY (`id`) USING BTREE,
UNIQUE INDEX `USERNAME` (`username`) USING BTREE UNIQUE INDEX `USERNAME` (`username`) USING BTREE
) ENGINE = InnoDB ) ENGINE = InnoDB
@ -171,4 +172,30 @@ CREATE TABLE `user_account`
COLLATE = utf8_general_ci COLLATE = utf8_general_ci
ROW_FORMAT = DYNAMIC; ROW_FORMAT = DYNAMIC;
SET FOREIGN_KEY_CHECKS = 1; SET
FOREIGN_KEY_CHECKS = 1;
-- ----------------------------
-- Table structure for msg_serial
-- ----------------------------
DROP TABLE IF EXISTS `msg_serial`;
CREATE TABLE `msg_serial`
(
`id` bigint(20) NOT NULL AUTO_INCREMENT,
`msgSerial` bigint(20) NOT NULL DEFAULT '1' COMMENT '连续消息序号',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=50 DEFAULT CHARSET=utf8;
-- ----------------------------
-- Table structure for alarm_information
-- ----------------------------
DROP TABLE IF EXISTS `alarm_information`;
CREATE TABLE `alarm_information`
(
`id` bigint(11) NOT NULL AUTO_INCREMENT,
`alarmInfo` varchar(255) NOT NULL COMMENT 'emos告警内容',
`createTime` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '创建时间',
PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=8 DEFAULT CHARSET=utf8;

View File

@ -0,0 +1,19 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.dispose.mapper.AlarmInformationMapper">
<resultMap id="msg_serial" type="com.dispose.pojo.entity.AlarmInformation">
<id column="id" property="id"/>
<result column="alarmInfo" property="alarmInfo"/>
<result column="createTime" property="createTime"/>
</resultMap>
<insert id="addAlarmInfo" useGeneratedKeys="true" keyProperty="id"
parameterType="com.dispose.pojo.entity.AlarmInformation">
INSERT
IGNORE INTO alarm_information(alarmInfo)
VALUES (
#{alarmInfo}
)
</insert>
</mapper>

View File

@ -0,0 +1,23 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.dispose.mapper.MsgSerialMapper">
<resultMap id="msg_serial" type="com.dispose.pojo.entity.MsgSerial">
<id column="id" property="id"/>
<result column="msgSerial" property="msgSerial"/>
</resultMap>
<insert id="addMsgSerial" useGeneratedKeys="true" keyProperty="id"
parameterType="com.dispose.pojo.entity.MsgSerial">
INSERT
IGNORE INTO msg_serial(msgSerial)
VALUES (
#{msgSerial}
)
</insert>
<select id="getMaxMsgSerial" resultType="java.lang.Long">
SELECT msgSerial
FROM msg_serial
ORDER BY ID DESC LIMIT 1
</select>
</mapper>

View File

@ -75,15 +75,15 @@ public class KafkaControllerTest extends InitTestEnvironment {
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
disposeParam.set("alarmId", "1"); disposeParam.set("alarmId", "1");
disposeParam.set("dstIp", "192.168.1.1"); disposeParam.set("dstIp", "192.168.1.1");
disposeParam.set("attackType", "1"); disposeParam.set("attackType", "RSTFlood");
disposeParam.set("bpspps", "bps"); disposeParam.set("bpspps", "bps");
disposeParam.set("dstProvince", "ZHEJIANG"); disposeParam.set("dstProvince", "浙江");
disposeParam.set("dstCity", "HANGZHOU"); disposeParam.set("dstCity", "杭州");
disposeParam.set("startTime", sdf.format(new Date())); disposeParam.set("startTime", sdf.format(new Date()));
// 1清洗2流控3黑洞 // 1清洗2流控3黑洞
disposeParam.set("disposeType", 1); disposeParam.set("disposeType", 1);
disposeParam.set("disposeTime", 30); disposeParam.set("disposeTime", 30);
disposeParam.set("endTime", new Date()); disposeParam.set("endTime", sdf.format(new Date()));
List<String> srcIp = new ArrayList<>(); List<String> srcIp = new ArrayList<>();
srcIp.add("192.168.10.1"); srcIp.add("192.168.10.1");
srcIp.add("192.168.10.2"); srcIp.add("192.168.10.2");

View File

@ -0,0 +1,60 @@
package com.dispose.test.dev.mapper;
import com.dispose.mapper.MsgSerialMapper;
import com.dispose.pojo.entity.MsgSerial;
import com.dispose.test.dev.Global.InitTestEnvironment;
import lombok.extern.slf4j.Slf4j;
import org.junit.Assert;
import org.junit.FixMethodOrder;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.MethodSorters;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
/**
* The message serial test.
*/
@RunWith(SpringRunner.class)
@SpringBootTest
@Slf4j
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
@Transactional
public class MsgSerialMapperTest extends InitTestEnvironment {
/**
* The message serial mapper.
*/
@Resource
MsgSerialMapper msgSerialMapper;
/**
* A 1 add new message serial.
*/
@Test
public void a1_addMsgSerial() {
for (long i = 1L; i <= 10L; i++) {
MsgSerial msgSerial = MsgSerial.builder()
.msgSerial(i)
.build();
log.info("++++++++++++++++++MsgSerial {}", msgSerial.toString());
msgSerialMapper.addMsgSerial(msgSerial);
}
}
@Test
public void a2_getMaxMsgSerial() {
for (long i = 1L; i <= 16L; i++) {
MsgSerial msgSerial = MsgSerial.builder()
.msgSerial(i)
.build();
msgSerialMapper.addMsgSerial(msgSerial);
}
long maxMsgSerial = msgSerialMapper.getMaxMsgSerial();
log.info("+++++++++++++++++++ max MsgSerial {}", maxMsgSerial);
Assert.assertEquals(maxMsgSerial, 16);
}
}