From b23c3e26899fde1724674a49753e3a033197ef5a Mon Sep 17 00:00:00 2001
From: dean <byungnam.baek@dabee.com>
Date: Wed, 15 Apr 2026 12:02:17 +0900
Subject: [PATCH] =?UTF-8?q?oom=EC=B2=98=EB=A6=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tasklet/PostShapefileUpdateTasklet.java   | 64 +++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 shp-exporter/src/main/java/com/kamco/makesample/batch/tasklet/PostShapefileUpdateTasklet.java

diff --git a/shp-exporter/src/main/java/com/kamco/makesample/batch/tasklet/PostShapefileUpdateTasklet.java b/shp-exporter/src/main/java/com/kamco/makesample/batch/tasklet/PostShapefileUpdateTasklet.java
new file mode 100644
index 0000000..22a6f85
--- /dev/null
+++ b/shp-exporter/src/main/java/com/kamco/makesample/batch/tasklet/PostShapefileUpdateTasklet.java
@@ -0,0 +1,64 @@
+package com.kamco.makesample.batch.tasklet;
+
+import java.util.List;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.batch.core.StepContribution;
+import org.springframework.batch.core.configuration.annotation.StepScope;
+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.jdbc.core.JdbcTemplate;
+import org.springframework.stereotype.Component;
+
+/**
+ * Shapefile 생성 완료 후 실행되는 UPDATE Tasklet
+ *
+ * <p>Job Flow 상 generateShapefileStep 이후에 실행됩니다.
+ *
+ * <p>실행할 SQL은 이 클래스의 execute() 메서드 안에 작성하세요.
+ */
+@Component
+@StepScope
+public class PostShapefileUpdateTasklet implements Tasklet {
+
+  private static final Logger log = LoggerFactory.getLogger(PostShapefileUpdateTasklet.class);
+
+  private final JdbcTemplate jdbcTemplate;
+
+  @Value("#{jobParameters['inferenceId']}")
+  private String inferenceId;
+
+  @Value("#{jobParameters['batchIds']}")
+  private String batchIds;
+
+  public PostShapefileUpdateTasklet(JdbcTemplate jdbcTemplate) {
+    this.jdbcTemplate = jdbcTemplate;
+  }
+
+  @Override
+  public RepeatStatus execute(StepContribution contribution, ChunkContext chunkContext)
+      throws Exception {
+
+    log.info(
+        "Executing post-shapefile UPDATE for inferenceId={}, batchIds={}", inferenceId, batchIds);
+
+    List<Long> batchIdList =
+        List.of(batchIds.split(",")).stream().map(Long::parseLong).toList();
+
+    // TODO: 실행할 UPDATE SQL을 여기에 작성하세요.
+    // 예시:
+    // int updated = jdbcTemplate.update(
+    //     "UPDATE some_table SET status = 'EXPORTED', inference_id = ? WHERE batch_id = ANY(?)",
+    //     ps -> {
+    //       ps.setString(1, inferenceId);
+    //       ps.setArray(2, ps.getConnection().createArrayOf("bigint", batchIdList.toArray()));
+    //     });
+    // log.info("Updated {} rows", updated);
+
+    log.info("Post-shapefile UPDATE completed");
+
+    return RepeatStatus.FINISHED;
+  }
+}
\ No newline at end of file