Allow enabling tpcds connector in HiveQueryRunner

It is useful to have tpcds connector enabled when trying to debug tpcds queries locally
trinodb · Feb 26, 2022 · 7d0506a · 7d0506a
1 parent 43e0431
commit 7d0506a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/plugin/trino-hive/pom.xml b/plugin/trino-hive/pom.xml
@@ -371,6 +371,12 @@
             <scope>test</scope>
         </dependency>
 
+        <dependency>
+            <groupId>io.trino</groupId>
+            <artifactId>trino-tpcds</artifactId>
+            <scope>test</scope>
+        </dependency>
+
         <dependency>
             <groupId>io.trino</groupId>
             <artifactId>trino-tpch</artifactId>

diff --git a/plugin/trino-hive/src/test/java/io/trino/plugin/hive/HiveQueryRunner.java b/plugin/trino-hive/src/test/java/io/trino/plugin/hive/HiveQueryRunner.java
@@ -26,6 +26,7 @@
 import io.trino.plugin.hive.metastore.MetastoreConfig;
 import io.trino.plugin.hive.metastore.file.FileHiveMetastore;
 import io.trino.plugin.hive.metastore.file.FileHiveMetastoreConfig;
+import io.trino.plugin.tpcds.TpcdsPlugin;
 import io.trino.plugin.tpch.TpchPlugin;
 import io.trino.spi.security.Identity;
 import io.trino.spi.security.PrincipalType;
@@ -110,6 +111,7 @@ public static class Builder<SELF extends Builder<?>>
         };
         private Module module = EMPTY_MODULE;
         private Optional<CachingDirectoryLister> cachingDirectoryLister = Optional.empty();
+        private boolean tpcdsCatalogEnabled;
 
         protected Builder()
         {
@@ -182,6 +184,12 @@ public SELF setCachingDirectoryLister(CachingDirectoryLister cachingDirectoryLis
             return self();
         }
 
+        public SELF setTpcdsCatalogEnabled(boolean tpcdsCatalogEnabled)
+        {
+            this.tpcdsCatalogEnabled = tpcdsCatalogEnabled;
+            return self();
+        }
+
         @Override
         public DistributedQueryRunner build()
                 throws Exception
@@ -194,6 +202,11 @@ public DistributedQueryRunner build()
                 queryRunner.installPlugin(new TpchPlugin());
                 queryRunner.createCatalog("tpch", "tpch");
 
+                if (tpcdsCatalogEnabled) {
+                    queryRunner.installPlugin(new TpcdsPlugin());
+                    queryRunner.createCatalog("tpcds", "tpcds");
+                }
+
                 HiveMetastore metastore = this.metastore.apply(queryRunner);
                 queryRunner.installPlugin(new TestingHivePlugin(metastore, module, cachingDirectoryLister));
 
@@ -358,6 +371,7 @@ public static void main(String[] args)
                 .setHiveProperties(ImmutableMap.of())
                 .setInitialTables(TpchTable.getTables())
                 .setBaseDataDir(baseDataDir)
+                .setTpcdsCatalogEnabled(true)
                 .build();
         Thread.sleep(10);
         log.info("======== SERVER STARTED ========");