Files
sake_batch/src/main/java/com/wb/domainserver/acikey/AciKeyTokenTasklet.java

83 lines
2.9 KiB
Java

package com.wb.domainserver.acikey;
import lombok.extern.slf4j.Slf4j;
import org.springframework.batch.core.StepContribution;
import org.springframework.batch.core.scope.context.ChunkContext;
import org.springframework.batch.core.step.tasklet.Tasklet;
import org.springframework.batch.repeat.RepeatStatus;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.Map;
// ACIKey 토큰 발급 테스트 배치 - API 호출 및 토큰 확인 로직
@Slf4j
@Component
public class AciKeyTokenTasklet implements Tasklet {
@Value("${acikey.url}")
private String aciKeyUrl;
@Value("${acikey.userid}")
private String userId;
@Value("${acikey.apikey}")
private String apiKey;
@Override
public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext) {
log.info("[AciKeyBatch] 배치 시작");
log.info("[AciKeyBatch] API URL: {}", aciKeyUrl);
try {
// Header 설정
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.APPLICATION_JSON);
headers.set("USERID", userId);
headers.set("APIKEY", apiKey);
HttpEntity<String> requestEntity = new HttpEntity<>("{}", headers);
// API 호출
RestTemplate restTemplate = new RestTemplate();
ResponseEntity<String> response = restTemplate.exchange(
aciKeyUrl,
HttpMethod.POST,
requestEntity,
String.class
);
ObjectMapper objectMapper = new ObjectMapper();
Map<String, Object> body = objectMapper.readValue(response.getBody(), Map.class);
log.info("[AciKeyBatch] 응답 전체: {}", body);
String status = (String) body.get("Status");
String aciKey = (String) body.get("ACIKey");
String errMsg = (String) body.get("ErrMsg");
if ("success".equals(status)) {
log.info("[AciKeyBatch] ✅ 토큰 발급 성공!");
log.info("[AciKeyBatch] ACIKey: {}", aciKey);
contribution.incrementWriteCount(1);
} else {
log.error("[AciKeyBatch] ❌ 토큰 발급 실패. Status: {}, ErrMsg: {}", status, errMsg);
throw new RuntimeException("ACIKey 토큰 발급 실패. Status: " + status + ", ErrMsg: " + errMsg);
}
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
log.error("[AciKeyBatch] ❌ API 호출 중 오류 발생: {}", e.getMessage(), e);
throw new RuntimeException("AciKey API 호출 오류", e);
}
log.info("[AciKeyBatch] 배치 종료");
return RepeatStatus.FINISHED;
}
}