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()