From c6f227105b2957b0273080c61300b5b21d35a08d Mon Sep 17 00:00:00 2001
From: Son Do <son.do@treasure-data.com>
Date: Thu, 12 Jun 2025 09:29:08 +0700
Subject: [PATCH] Add trino query

---
 .../com/treasuredata/client/model/TDJob.java  |  2 +-
 .../client/model/TDJobRequest.java            | 30 +++++++++++++++++++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/treasuredata/client/model/TDJob.java b/src/main/java/com/treasuredata/client/model/TDJob.java
index 5eb3b44d..24567550 100644
--- a/src/main/java/com/treasuredata/client/model/TDJob.java
+++ b/src/main/java/com/treasuredata/client/model/TDJob.java
@@ -31,7 +31,7 @@ public class TDJob
 {
     public static enum Type
     {
-        HIVE("hive"), MAPRED("mapred"), PRESTO("presto"), PIG("pig"), BULKLOAD("bulkload"), EXPORT("export"), UNKNOWN("none");
+        HIVE("hive"), MAPRED("mapred"), PRESTO("presto"), TRINO("trino"), PIG("pig"), BULKLOAD("bulkload"), EXPORT("export"), UNKNOWN("none");
         private final String type;
 
         private Type(String type)
diff --git a/src/main/java/com/treasuredata/client/model/TDJobRequest.java b/src/main/java/com/treasuredata/client/model/TDJobRequest.java
index cbcf66eb..5f2ece57 100644
--- a/src/main/java/com/treasuredata/client/model/TDJobRequest.java
+++ b/src/main/java/com/treasuredata/client/model/TDJobRequest.java
@@ -110,6 +110,36 @@ public static TDJobRequest newPrestoQuery(String database, String query, String
                 .createTDJobRequest();
     }
 
+    public static TDJobRequest newTrinoQuery(String database, String query)
+    {
+        return new TDJobRequestBuilder()
+                .setType(TDJob.Type.TRINO)
+                .setDatabase(database)
+                .setQuery(query)
+                .createTDJobRequest();
+    }
+
+    public static TDJobRequest newTrinoQuery(String database, String query, String resultOutput)
+    {
+        return new TDJobRequestBuilder()
+                .setType(TDJob.Type.TRINO)
+                .setDatabase(database)
+                .setQuery(query)
+                .setResultOutput(resultOutput)
+                .createTDJobRequest();
+    }
+
+    public static TDJobRequest newTrinoQuery(String database, String query, String resultOutput, String poolName)
+    {
+        return new TDJobRequestBuilder()
+                .setType(TDJob.Type.TRINO)
+                .setDatabase(database)
+                .setQuery(query)
+                .setResultOutput(resultOutput)
+                .setPoolName(poolName)
+                .createTDJobRequest();
+    }
+
     public static TDJobRequest newHiveQuery(String database, String query)
     {
         return new TDJobRequestBuilder()