You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
214 lines
12 KiB
214 lines
12 KiB
package com.bsmlab.dfx.agent;
|
|
|
|
import com.bsmlab.dfx.agent.config.AgentConfigDto;
|
|
import com.bsmlab.dfx.agent.config.AgentConfigReader;
|
|
import com.bsmlab.dfx.agent.config.datasource.DynamicRoutingDataSource;
|
|
import com.bsmlab.dfx.agent.listener.dto.AckDto;
|
|
import com.bsmlab.dfx.agent.listener.dto.CommandDto;
|
|
import com.fasterxml.jackson.core.JsonProcessingException;
|
|
import com.fasterxml.jackson.core.type.TypeReference;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import lombok.RequiredArgsConstructor;
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springframework.boot.ApplicationArguments;
|
|
import org.springframework.boot.ApplicationRunner;
|
|
import org.springframework.http.HttpEntity;
|
|
import org.springframework.http.HttpHeaders;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.stereotype.Component;
|
|
import org.springframework.web.client.ResourceAccessException;
|
|
import org.springframework.web.client.RestTemplate;
|
|
|
|
import javax.sql.DataSource;
|
|
import java.sql.Connection;
|
|
import java.sql.SQLException;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
|
|
@Slf4j
|
|
@RequiredArgsConstructor
|
|
@Component
|
|
public class DfxAgentRunner implements ApplicationRunner {
|
|
private final AgentConfigReader agentConfigReader;
|
|
private final DynamicRoutingDataSource dynamicRoutingDataSource;
|
|
|
|
@Override
|
|
public void run(ApplicationArguments args) throws Exception {
|
|
// command line 실행 중 --check-config 가 있는 경우 settings.json 을 로드하여 DB 연결, knownAgent 연결 테스트를 진행한다.
|
|
// java -jar $AGENT_HOME/lib/dfxagent.jar --setting.check --setting.file=$AGENT_HOME/conf/settings.json &
|
|
if(args.containsOption("setting.check")) {
|
|
this.print("-----------------------");
|
|
this.print("DfxAgent 설정 및 환경 점검");
|
|
this.print("-----------------------");
|
|
this.print(" (설정 파일: " + args.getOptionValues("setting.file") + ")");
|
|
// settings.json 점검 로직
|
|
boolean isValid = true;
|
|
|
|
this.print("");
|
|
this.print("데이터베이스 설정 점검");
|
|
Map<Object, Object> dataSourceMap = dynamicRoutingDataSource.getDataSourceMap();
|
|
if(dataSourceMap != null && !dataSourceMap.isEmpty()) {
|
|
for(Object key : dataSourceMap.keySet()) {
|
|
DataSource dataSource = (DataSource) dataSourceMap.get(key);
|
|
AgentConfigDto.DataSourceConfig dataSourceConfig = null;
|
|
List<AgentConfigDto.DataSourceConfig> dataSourceConfigList = agentConfigReader.getAgentConfigDto().getDataSourceConfig();
|
|
for(AgentConfigDto.DataSourceConfig thisDataSourceConfig : dataSourceConfigList) {
|
|
if(thisDataSourceConfig.getDataSourceId().equals(key)) {
|
|
dataSourceConfig = thisDataSourceConfig;
|
|
break;
|
|
}
|
|
}
|
|
this.print("DB 점검 [" + dataSourceConfig.toString() + "]");
|
|
try(Connection connection = dataSource.getConnection()) {
|
|
connection.isValid(10);
|
|
this.print("DB 연결에 성공하였습니다.");
|
|
}
|
|
catch(SQLException e) {
|
|
this.print("DB 연결에 실패하였습니다. (" + e.getLocalizedMessage() + ")");
|
|
isValid = false;
|
|
}
|
|
catch(Exception e) {
|
|
this.print("DB 점검에 실패하였습니다. (" + e.getLocalizedMessage() + ")");
|
|
isValid = false;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
this.print("데이터베이스 설정은 포함되어 있지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
|
|
this.print("");
|
|
this.print("상대 agent 설정 점검");
|
|
List<AgentConfigDto.KnownAgent> knownAgentList = agentConfigReader.getAgentConfigDto().getKnownAgentList();
|
|
if(knownAgentList != null && !knownAgentList.isEmpty()) {
|
|
Map<String, AgentConfigDto.KnownAgent> knownAgentMap = new HashMap<>();
|
|
for(AgentConfigDto.KnownAgent knownAgent : knownAgentList) {
|
|
knownAgentMap.put(knownAgent.getHostId(), knownAgent);
|
|
try {
|
|
String messageUuid = UUID.randomUUID().toString();
|
|
CommandDto commandDto = CommandDto.builder().commandType(CommandDto.CommandType.ALIVE).messageUuid(messageUuid).build();
|
|
HttpHeaders httpHeaders = new HttpHeaders();
|
|
httpHeaders.setContentType(MediaType.APPLICATION_JSON);
|
|
httpHeaders.set("User-Agent", agentConfigReader.getApplicationName() + ", version: " + agentConfigReader.getApplicationVersion() + "(" + agentConfigReader.getApplicationCommitId() + ")" + ", host ID: " + agentConfigReader.getAgentConfigDto().getMyHostId());
|
|
HttpEntity<CommandDto> bodyEntity = new HttpEntity<>(commandDto, httpHeaders);
|
|
RestTemplate restTemplate = new RestTemplate();
|
|
String url = "http://" + knownAgent.getHostName() + ":" + knownAgent.getListenPort() + "/command";
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ":" + knownAgent.getListenPort() + ") 연결 점검");
|
|
String response = restTemplate.postForObject(url, bodyEntity, String.class);
|
|
ObjectMapper objectMapper = new ObjectMapper();
|
|
AckDto ackDto = null;
|
|
ackDto = objectMapper.readValue(response, new TypeReference<AckDto>() {});
|
|
if(AckDto.ResultType.PROCESS_SUCCESS == ackDto.getResult()) {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + "가 정상 동작합니다.");
|
|
commandDto.setCommandType(CommandDto.CommandType.INFORMATION);
|
|
bodyEntity = new HttpEntity<>(commandDto, httpHeaders);
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ":" + knownAgent.getListenPort() + ") DropBox 점검");
|
|
response = restTemplate.postForObject(url, bodyEntity, String.class);
|
|
ackDto = objectMapper.readValue(response, new TypeReference<AckDto>() {});
|
|
AgentConfigDto knownAgentConfigDto = objectMapper.readValue(ackDto.getResultText(), new TypeReference<AgentConfigDto>() {});
|
|
List<AgentConfigDto.DropBox> knownAgentDropBoxList = knownAgentConfigDto.getDropBoxConfig().getDropBoxList();
|
|
if(knownAgent.getDropBoxIdList() != null) {
|
|
if(knownAgentDropBoxList != null) {
|
|
for(String dropBoxId : knownAgent.getDropBoxIdList()) {
|
|
boolean isHasDropBoxId = false;
|
|
for(AgentConfigDto.DropBox knwonAgentDropBox : knownAgentDropBoxList) {
|
|
if(dropBoxId.equals(knwonAgentDropBox.getDropBoxId())) {
|
|
isHasDropBoxId = true;
|
|
}
|
|
}
|
|
if(!isHasDropBoxId) {
|
|
this.print("DropBoxId " + dropBoxId + "에 해당하는 상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ")의 DropBoxId가 존재하지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ")의 DropBox 설정이 존재하지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ")가 응답하였으나 정상 동작하지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
}
|
|
catch (JsonProcessingException e) {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ")가 응답메시지가 정상적이지 않습니다. (" + e.getLocalizedMessage() + ")");
|
|
isValid = false;
|
|
}
|
|
catch (ResourceAccessException e) {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ")가 응답하지 않습니다. (" + e.getLocalizedMessage() + ")");
|
|
isValid = false;
|
|
}
|
|
catch (Exception e) {
|
|
this.print("상대 agent " + knownAgent.getHostId() + "(" + knownAgent.getHostName() + ") 상태를 알 수 없습니다. (" + e.getLocalizedMessage() + ")");
|
|
isValid = false;
|
|
}
|
|
}
|
|
|
|
this.print("");
|
|
this.print("Postman 설정 점검");
|
|
if(agentConfigReader.getAgentConfigDto().getPostmanConfig() != null) {
|
|
this.print("Postman thread pool size: " + agentConfigReader.getAgentConfigDto().getPostmanConfig().getThreadPoolSize());
|
|
List<AgentConfigDto.Postman> postManList = agentConfigReader.getAgentConfigDto().getPostmanConfig().getPostmanList();
|
|
if(postManList != null) {
|
|
for(AgentConfigDto.Postman postman : postManList) {
|
|
AgentConfigDto.KnownAgent knownAgent = knownAgentMap.get(postman.getRecipientHostId());
|
|
if(knownAgent != null) {
|
|
boolean isHasDropBoxId = false;
|
|
for(String knownAgentDropBoxId : knownAgent.getDropBoxIdList()) {
|
|
if(postman.getRecipientDropBoxId().equals(knownAgentDropBoxId)) {
|
|
isHasDropBoxId = true;
|
|
}
|
|
}
|
|
if(!isHasDropBoxId) {
|
|
this.print("postman " + postman.getPostmanId() + "에 대한 recipientDropBoxId(" + postman.getRecipientDropBoxId() + ")가 knownAgent(" + knownAgent.getHostId() + ") dropBoxIdList 목록에 존재하지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
}
|
|
else {
|
|
this.print("postman " + postman.getPostmanId() + "에 대한 recipientHostId(" + postman.getRecipientHostId() + ")가 knownAgent 목록에 존재하지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
this.print("Postman 목록 없음");
|
|
}
|
|
}
|
|
else {
|
|
this.print("Postman 설정 없음");
|
|
isValid = false;
|
|
}
|
|
}
|
|
else {
|
|
this.print("상대 agent 설정은 포함되어 있지 않습니다.");
|
|
isValid = false;
|
|
}
|
|
|
|
this.print("");
|
|
this.print("--------------------------------------");
|
|
if(isValid) {
|
|
this.print("설정과 환경이 정상적입니다.");
|
|
this.print("점검을 종료합니다.");
|
|
System.exit(0);
|
|
}
|
|
else {
|
|
this.print("설정 또는 환경에 오류가 포함되어 있습니다.");
|
|
this.print("점검을 종료합니다.");
|
|
System.exit(2);
|
|
}
|
|
this.print("--------------------------------------");
|
|
}
|
|
}
|
|
|
|
private void print(String text) {
|
|
System.out.println(text);
|
|
}
|
|
}
|
|
|