Merge remote-tracking branch 'origin/v2.0.9_dev' into v2.0.9_dev
This commit is contained in:
HuangXin
commit
f8778ad438
|
|
@ -6,32 +6,27 @@ dispose.split_char=,
|
|||
dispose.request-timeout-second=5
|
||||
# 是否开启隐私保护
|
||||
dispose.used-privacy-protect=false
|
||||
|
||||
dispose.call-error-retry-times=3
|
||||
# 分页配置项
|
||||
# 最大每页数据条数
|
||||
dispose.max-split-page-size=100
|
||||
# 最小每页数据条数
|
||||
dispose.min-split-page-size=10
|
||||
|
||||
# 迪普设备配置
|
||||
# 发送超时时间(s)
|
||||
dptech.soap-conn-timeout-second=60
|
||||
# 接收超时时间(s)
|
||||
dptech.soap-recv-timeout-second=60
|
||||
|
||||
# 用户权限配置
|
||||
# 是否对设备管理进行用户验证
|
||||
permission.admin-check=true
|
||||
# 运行管理设备的操作员用户名
|
||||
permission.admin-users=admin
|
||||
|
||||
# 认证配置
|
||||
# 是否对接口访问进行认证
|
||||
auth.verify-request-token=true
|
||||
# token访问超时时间
|
||||
auth.token-timeout-minute=30
|
||||
|
||||
# 安全配置
|
||||
#加密类型： 0 不加密
|
||||
# 1 Base64编码
|
||||
|
|
@ -41,7 +36,6 @@ crypto.security-protocol-type=0
|
|||
crypto.aes-key=hkoUV5ZWh0q1jSxMnpjovVn19Qg99HY6DD40
|
||||
# 3DES秘钥
|
||||
crypto.des-key=P3mq9iSIvQcvfyfdWR8sAnfAadO
|
||||
|
||||
# Kafka 服务器配置
|
||||
#重试次数
|
||||
kafka.producer.retries=3
|
||||
|
|
@ -53,12 +47,6 @@ kafka.producer.linger=1
|
|||
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.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
|
||||
|
|
|
|||
|
|
@ -4,17 +4,16 @@ server.tomcat.basedir=./basedir
|
|||
# 多个项目放在nginx下同个端口，通过该配置区分
|
||||
server.servlet.context-path=/dispose
|
||||
# 配置数据源
|
||||
#spring.datasource.url=jdbc:mysql://10.88.77.65:33061/ci_dispose_v2?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\
|
||||
# =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.url=jdbc:mysql://10.88.77.65:33061/ci_dispose_v2?serverTimezone=Asia/Shanghai&zeroDateTimeBehavior\
|
||||
=convertToNull&useUnicode=true&characterEncoding=utf8&allowMultiQueries=true
|
||||
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
|
||||
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.initialization-mode=always
|
||||
|
|
|
|||
|
|
@ -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;
|
||||
}
|
||||
|
|
@ -293,6 +293,10 @@ public enum ErrorCode {
|
|||
* The Err huawei firewall error.
|
||||
*/
|
||||
ERR_HUAWEIFIREWALL_ERROR(305, "华为防火墙返回错误"),
|
||||
/**
|
||||
* The Err emos create message error.
|
||||
*/
|
||||
EMOS_CREATEMESSAGE_ERROR(306, "EMOS发送信息错误"),
|
||||
;
|
||||
|
||||
/**
|
||||
|
|
|
|||
|
|
@ -1,21 +1,26 @@
|
|||
package com.dispose.controller;
|
||||
|
||||
import com.alibaba.fastjson.JSONObject;
|
||||
import com.dispose.common.Constants;
|
||||
import com.dispose.common.ErrorCode;
|
||||
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.ProtocolReqDTO;
|
||||
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.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.Encryption;
|
||||
import com.dispose.service.MsgSerialService;
|
||||
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.ApiOperation;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang.exception.ExceptionUtils;
|
||||
import org.springframework.kafka.support.SendResult;
|
||||
import org.springframework.stereotype.Component;
|
||||
import org.springframework.stereotype.Controller;
|
||||
|
|
@ -27,12 +32,13 @@ import org.springframework.web.bind.annotation.RequestMapping;
|
|||
import org.springframework.web.bind.annotation.ResponseBody;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.text.MessageFormat;
|
||||
import java.util.Objects;
|
||||
|
||||
/**
|
||||
* The type Auth controller.
|
||||
*
|
||||
* @author <huangxin@cmhi.chinamoblie.com>
|
||||
* @author <chenling@cmhi.chinamoblie.com>
|
||||
*/
|
||||
@Controller
|
||||
@RequestMapping(value = "/kafka")
|
||||
|
|
@ -43,20 +49,24 @@ import java.util.Objects;
|
|||
@Encryption
|
||||
@Decryption
|
||||
public class kafkaController {
|
||||
/**
|
||||
* The Object mapper.
|
||||
*/
|
||||
@Resource
|
||||
private ObjectMapper objectMapper;
|
||||
|
||||
/**
|
||||
* The Kafka configuration.
|
||||
*/
|
||||
// final private KafkaConfiguration kafkaConfiguration =
|
||||
// SpringBootBeanUtil.getBean(com.dispose.config.KafkaConfiguration.class);
|
||||
@Resource
|
||||
private KafkaConfiguration kafkaConfiguration;
|
||||
|
||||
/**
|
||||
* The message serial service.
|
||||
*/
|
||||
@Resource
|
||||
private MsgSerialService msgSerialService;
|
||||
|
||||
/**
|
||||
* The alarm information manager.
|
||||
*/
|
||||
@Resource
|
||||
private AlarmInfoManager alarmInfoManager;
|
||||
|
||||
/**
|
||||
* Dispatch command sent to kafka.
|
||||
*
|
||||
|
|
@ -68,41 +78,177 @@ public class kafkaController {
|
|||
@ApiOperation("发送消息")
|
||||
public ProtocolRespDTO<BaseRespStatus> dispatchCommand(
|
||||
@Validated(ValidGroups.ProtocolCommonValid.class)
|
||||
@RequestBody ProtocolReqDTO<AlarmInfoReq> mr) throws JsonProcessingException {
|
||||
//获取入参信息，进行所需数据格式拼接
|
||||
log.info("alarm message information :{}", mr.getMsgContent().getAlarmInfo());
|
||||
AlarmInfo alarmInfo = objectMapper.readValue(mr.getMsgContent().getAlarmInfo(), new TypeReference<AlarmInfo>() {
|
||||
});
|
||||
@RequestBody ProtocolReqDTO<AlarmInfoReq> mr) {
|
||||
//获取入参信息
|
||||
BaseRespStatus rspInfo = new BaseRespStatus();
|
||||
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
|
||||
log.info("send alarm message :{}", alarmInfo.toString());
|
||||
String sendMessage = mr.getMsgContent().getAlarmInfo();
|
||||
ListenableFuture<SendResult<String, String>> sendResult = kafkaConfiguration
|
||||
.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)
|
||||
.getRecordMetadata()
|
||||
.topic(), v.getRecordMetadata().partition()),
|
||||
ex -> log.error("Kafka send error: {}", ex.getMessage()));
|
||||
|
||||
BaseRespStatus rspInfo = new BaseRespStatus();
|
||||
rspInfo.setStatus(ErrorCode.ERR_OK.getCode());
|
||||
rspInfo.setMessage(new String[]{ErrorCode.ERR_OK.getMsg()});
|
||||
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"})
|
||||
// public void kafkaListen(ConsumerRecord<String, String> consumerRecord) {
|
||||
// //判断消息是否为null
|
||||
// Optional<String> kafkaMessage = Optional.ofNullable(consumerRecord.value());
|
||||
// log.info(">>>>>>>>>>> record = " + kafkaMessage);
|
||||
// if (kafkaMessage.isPresent()) {
|
||||
// String consumerMsg = kafkaMessage.get();
|
||||
// log.info("消费消息：" + consumerMsg);
|
||||
// }
|
||||
// }
|
||||
String dstIp = alarmInfo.getDstIp();
|
||||
String alarmId = alarmInfo.getAlarmId();
|
||||
//LocateInfo: 192.168.11.6 ddos attack alarm，Memcache , 2019-08-01 12:31, 2.1Gbps, 1.01Mpps
|
||||
String locateInfo = getAlarmEvent(alarmInfo);
|
||||
String eventTime = alarmInfo.getStartTime();
|
||||
String cancelTime = alarmInfo.getEndTime();
|
||||
|
||||
//告警类型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() + "分钟";
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -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);
|
||||
}
|
||||
|
|
@ -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();
|
||||
}
|
||||
|
|
@ -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);
|
||||
}
|
||||
}
|
||||
|
|
@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -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);
|
||||
}
|
||||
|
|
@ -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();
|
||||
}
|
||||
|
|
@ -16,7 +16,7 @@ import java.util.List;
|
|||
@JsonPropertyOrder({"alarmId", "dstIp", "attackType", "bpspps", "dstProvince", "dstCity", "srcIpLs", "startTime",
|
||||
"endTime", "disposeType", "disposeTime", "maxBps", "maxPps"})
|
||||
@JsonInclude(JsonInclude.Include.NON_NULL)
|
||||
public class AlarmInfo {
|
||||
public class EmosAlarmInfo {
|
||||
/**
|
||||
* 告警id.
|
||||
*/
|
||||
|
|
@ -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;
|
||||
}
|
||||
|
|
@ -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;
|
||||
}
|
||||
|
|
@ -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();
|
||||
|
||||
}
|
||||
|
|
@ -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();
|
||||
}
|
||||
}
|
||||
|
|
@ -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);
|
||||
}
|
||||
}
|
||||
|
|
@ -9,3 +9,11 @@ INSERT INTO `user_account`
|
|||
VALUES (1, 'admin', 'c3855e6b6bb120450f160ba91134522868f89d36062f2061ebeefd80817e1d58', '2020-11-13 09:25:19', '',
|
||||
'2021-01-20 10:18:56', '1db9ddc47de514eb16b7ec07d7f7f96f7a714ae00e1209755bab30d543a0a2c3',
|
||||
'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');
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -15,7 +15,8 @@
|
|||
*/
|
||||
|
||||
SET NAMES utf8mb4;
|
||||
SET FOREIGN_KEY_CHECKS = 0;
|
||||
SET
|
||||
FOREIGN_KEY_CHECKS = 0;
|
||||
|
||||
-- ----------------------------
|
||||
-- Table structure for device_task
|
||||
|
|
@ -171,4 +172,30 @@ CREATE TABLE `user_account`
|
|||
COLLATE = utf8_general_ci
|
||||
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;
|
||||
|
||||
|
|
|
|||
|
|
@ -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>
|
||||
|
|
@ -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>
|
||||
|
|
@ -75,15 +75,15 @@ public class KafkaControllerTest extends InitTestEnvironment {
|
|||
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
|
||||
disposeParam.set("alarmId", "1");
|
||||
disposeParam.set("dstIp", "192.168.1.1");
|
||||
disposeParam.set("attackType", "1");
|
||||
disposeParam.set("attackType", "RSTFlood");
|
||||
disposeParam.set("bpspps", "bps");
|
||||
disposeParam.set("dstProvince", "ZHEJIANG");
|
||||
disposeParam.set("dstCity", "HANGZHOU");
|
||||
disposeParam.set("dstProvince", "浙江");
|
||||
disposeParam.set("dstCity", "杭州");
|
||||
disposeParam.set("startTime", sdf.format(new Date()));
|
||||
// 1清洗，2流控，3黑洞
|
||||
disposeParam.set("disposeType", 1);
|
||||
disposeParam.set("disposeTime", 30);
|
||||
disposeParam.set("endTime", new Date());
|
||||
disposeParam.set("endTime", sdf.format(new Date()));
|
||||
List<String> srcIp = new ArrayList<>();
|
||||
srcIp.add("192.168.10.1");
|
||||
srcIp.add("192.168.10.2");
|
||||
|
|
|
|||
|
|
@ -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);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue