From 33a3c960f8be58c730b48378ec9920d3214495d4 Mon Sep 17 00:00:00 2001
From: Yuta Iwama <ganmacs@gmail.com>
Date: Wed, 4 Jun 2025 11:34:39 +0900
Subject: [PATCH] Add `header` params to getResult

https://api-docs.treasuredata.com/pages/td-api/tag/Jobs/#tag/Jobs/operation/getJobResults
---
 .../java/com/treasuredata/client/TDClient.java     |  8 ++++++++
 .../java/com/treasuredata/client/TDClientApi.java  | 14 ++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/src/main/java/com/treasuredata/client/TDClient.java b/src/main/java/com/treasuredata/client/TDClient.java
index a601f8f5..306cf76a 100644
--- a/src/main/java/com/treasuredata/client/TDClient.java
+++ b/src/main/java/com/treasuredata/client/TDClient.java
@@ -748,10 +748,18 @@ public TDJob jobInfo(String jobId)
     @Override
     public <Result> Result jobResult(String jobId, TDResultFormat format, Function<InputStream, Result> resultStreamHandler)
             throws TDClientException
+    {
+        return jobResult(jobId, format, false, resultStreamHandler);
+    }
+
+    @Override
+    public <Result> Result jobResult(String jobId, TDResultFormat format, boolean includeHeader, Function<InputStream, Result> resultStreamHandler)
+            throws TDClientException
     {
         TDApiRequest request = TDApiRequest.Builder
                 .GET(buildUrl("/v3/job/result", jobId))
                 .addQueryParam("format", format.getName())
+                .addQueryParam("header", Boolean.toString(includeHeader))
                 .build();
         return httpClient.<Result>call(request, apiKeyCache, resultStreamHandler);
     }
diff --git a/src/main/java/com/treasuredata/client/TDClientApi.java b/src/main/java/com/treasuredata/client/TDClientApi.java
index 57a82860..7cbf86c4 100644
--- a/src/main/java/com/treasuredata/client/TDClientApi.java
+++ b/src/main/java/com/treasuredata/client/TDClientApi.java
@@ -281,6 +281,20 @@ default <Result> Result jobResult(String jobId, TDResultFormat format, com.googl
      */
     <Result> Result jobResult(String jobId, TDResultFormat format, Function<InputStream, Result> resultStreamHandler);
 
+    /**
+     * Open an input stream to retrieve the job result.
+     * The input stream will be closed after this method
+     *
+     * You will receive an empty stream if the query has not finished yet.
+     *
+     * @param jobId
+     * @param format
+     * @param resultStreamHandler
+     * @param includeHeader
+     * @return
+     */
+    <Result> Result jobResult(String jobId, TDResultFormat format,  boolean includeHeader, Function<InputStream, Result> resultStreamHandler);
+
     // bulk import API
     List<TDBulkImportSession> listBulkImportSessions();