Skip to content

Commit 23ceff1

Browse files
committed
Add trino query
1 parent 274c1f6 commit 23ceff1

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

src/main/java/com/treasuredata/client/model/TDJob.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ public class TDJob
3131
{
3232
public static enum Type
3333
{
34-
HIVE("hive"), MAPRED("mapred"), PRESTO("presto"), PIG("pig"), BULKLOAD("bulkload"), EXPORT("export"), UNKNOWN("none");
34+
HIVE("hive"), MAPRED("mapred"), PRESTO("presto"), TRINO("trino"), PIG("pig"), BULKLOAD("bulkload"), EXPORT("export"), UNKNOWN("none");
3535
private final String type;
3636

3737
private Type(String type)

src/main/java/com/treasuredata/client/model/TDJobRequest.java

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,36 @@ public static TDJobRequest newPrestoQuery(String database, String query, String
110110
.createTDJobRequest();
111111
}
112112

113+
public static TDJobRequest newTrinoQuery(String database, String query)
114+
{
115+
return new TDJobRequestBuilder()
116+
.setType(TDJob.Type.TRINO)
117+
.setDatabase(database)
118+
.setQuery(query)
119+
.createTDJobRequest();
120+
}
121+
122+
public static TDJobRequest newTrinoQuery(String database, String query, String resultOutput)
123+
{
124+
return new TDJobRequestBuilder()
125+
.setType(TDJob.Type.TRINO)
126+
.setDatabase(database)
127+
.setQuery(query)
128+
.setResultOutput(resultOutput)
129+
.createTDJobRequest();
130+
}
131+
132+
public static TDJobRequest newTrinoQuery(String database, String query, String resultOutput, String poolName)
133+
{
134+
return new TDJobRequestBuilder()
135+
.setType(TDJob.Type.TRINO)
136+
.setDatabase(database)
137+
.setQuery(query)
138+
.setResultOutput(resultOutput)
139+
.setPoolName(poolName)
140+
.createTDJobRequest();
141+
}
142+
113143
public static TDJobRequest newHiveQuery(String database, String query)
114144
{
115145
return new TDJobRequestBuilder()

0 commit comments

Comments
 (0)