parent
9f354c1cb8
commit
188a49ae99
|
@ -24,6 +24,14 @@ public class ConstValue {
|
||||||
public class ProtocolCmdId {
|
public class ProtocolCmdId {
|
||||||
public static final int AUTH_LOGIN = 10;
|
public static final int AUTH_LOGIN = 10;
|
||||||
public static final int AUTH_LOGOUT = 11;
|
public static final int AUTH_LOGOUT = 11;
|
||||||
|
|
||||||
|
public static final int START_DISPOSE_CMD = 400;
|
||||||
|
public static final int STOP_DISPOSE_CMD = 401;
|
||||||
|
public static final int STOP_DISPOSE_CMD_BY_IP = 402;
|
||||||
|
public static final int STOP_DISPOSE_CMD_BY_POINT = 403;
|
||||||
|
public static final int STOP_DISPOSE_CMD_ALL = 404;
|
||||||
|
public static final int GET_DISPOSE_TASK_LIST = 405;
|
||||||
|
public static final int GET_DISPOSE_TASK_DETAIL = 406;
|
||||||
}
|
}
|
||||||
|
|
||||||
public enum DisposeDeviceType {
|
public enum DisposeDeviceType {
|
||||||
|
|
|
@ -0,0 +1,91 @@
|
||||||
|
package com.cmcc.hy.phoenix.controller;
|
||||||
|
|
||||||
|
import com.cmcc.hy.phoenix.annotation.bodyencdec.ReqDec;
|
||||||
|
import com.cmcc.hy.phoenix.common.ConstValue;
|
||||||
|
import com.cmcc.hy.phoenix.common.ErrorCode;
|
||||||
|
import com.cmcc.hy.phoenix.pojo.dto.ProtocolDTO;
|
||||||
|
import com.cmcc.hy.phoenix.pojo.vo.ProtocolResp;
|
||||||
|
import com.cmcc.hy.phoenix.service.LoginService;
|
||||||
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import io.swagger.annotations.Api;
|
||||||
|
import io.swagger.annotations.ApiOperation;
|
||||||
|
import lombok.Builder;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
import org.springframework.stereotype.Controller;
|
||||||
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
|
||||||
|
import java.security.NoSuchAlgorithmException;
|
||||||
|
import java.util.Objects;
|
||||||
|
|
||||||
|
import static com.cmcc.hy.phoenix.common.ErrorCode.ERR_OK;
|
||||||
|
import static com.cmcc.hy.phoenix.common.ErrorCode.ERR_UNKNOWNCMD;
|
||||||
|
|
||||||
|
@Controller
|
||||||
|
@RequestMapping(value = "/handle")
|
||||||
|
@Slf4j
|
||||||
|
@Api(value = "抗DDoS处置平台认证接口", tags = "抗DDoS处置平台认证接口")
|
||||||
|
@Component
|
||||||
|
public class taskController {
|
||||||
|
|
||||||
|
@Resource
|
||||||
|
private ObjectMapper objectMapper;
|
||||||
|
|
||||||
|
@Resource
|
||||||
|
private LoginService loginService;
|
||||||
|
|
||||||
|
@PostMapping("/command")
|
||||||
|
@ResponseBody
|
||||||
|
@ApiOperation("处置任务接口")
|
||||||
|
@ReqDec
|
||||||
|
@Builder
|
||||||
|
public ProtocolResp disposeTaskCommand(@RequestBody(required = true) ProtocolDTO mr,
|
||||||
|
@RequestHeader HttpHeaders headers) {
|
||||||
|
// 存放返回信息Json字符串
|
||||||
|
String msgCtx = "";
|
||||||
|
ErrorCode err = ERR_OK;
|
||||||
|
|
||||||
|
log.info("请求token: {}", headers.get("Authorization"));
|
||||||
|
log.info("请求参数 {}", mr);
|
||||||
|
|
||||||
|
// 参数检测
|
||||||
|
if (mr == null) {
|
||||||
|
err = ErrorCode.ERR_PARAMS;
|
||||||
|
return ProtocolResp.result(err, -1, msgCtx);
|
||||||
|
} else if (mr.IsRequestTimeout()) {
|
||||||
|
err = ErrorCode.ERR_REQTIMEOUT;
|
||||||
|
} else {
|
||||||
|
//try{
|
||||||
|
// 获取 Token
|
||||||
|
String token = Objects.
|
||||||
|
requireNonNull(headers.get("Authorization"))
|
||||||
|
.get(0).replaceFirst("Bearer ", "");
|
||||||
|
|
||||||
|
// 判断Head头是否存在
|
||||||
|
if (headers.get("Authorization") == null) {
|
||||||
|
err = ErrorCode.ERR_MISSAUTHHEAD;
|
||||||
|
} else if ((err = loginService.authTokenCheck(token)) == ERR_OK) {
|
||||||
|
// Token 正确则处理业务
|
||||||
|
switch (mr.getCmdId()) {
|
||||||
|
case ConstValue.ProtocolCmdId.START_DISPOSE_CMD:
|
||||||
|
break;
|
||||||
|
case ConstValue.ProtocolCmdId.STOP_DISPOSE_CMD:
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
err = ErrorCode.ERR_UNKNOWNCMD;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// } catch (JsonProcessingException | NoSuchAlgorithmException ex) {
|
||||||
|
// log.error(ex.getMessage());
|
||||||
|
// err = ErrorCode.ERR_EXCEPTION;
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
|
||||||
|
return ProtocolResp.result(err, mr.getCmdId(), msgCtx);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue