updates

kgyrtkirk · May 27, 2024 · ec2ecde · ec2ecde
1 parent 4595b0c
commit ec2ecde
Show file tree

Hide file tree

Showing 5 changed files with 425 additions and 11 deletions.
diff --git a/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java b/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java
@@ -99,6 +99,8 @@ private static void main3(Object object) throws Exception
       chk1();
       chkStatus();
 
+      System.out.println("-------------------booted up-------------------");
+
       lifecycle.join();
     } else {
 

diff --git a/integration-tests/src/main/java/org/apache/druid/quidem/QuidemRecorder.java b/integration-tests/src/main/java/org/apache/druid/quidem/QuidemRecorder.java
@@ -29,27 +29,22 @@ public QuidemRecorder(PrintStream printStream)
   {
     this.printStream = printStream;
     printStream.println("#started");
+    printStream.println("!connect druidtest:///");
     DruidHook.register(DruidHook.SQL, this);
-    DruidHook.register(DruidHook.RESULTSET, this);
   }
 
   @Override
   public void close()
   {
     DruidHook.unregister(DruidHook.SQL, this);
-    DruidHook.unregister(DruidHook.RESULTSET, this);
   }
 
   @Override
   public <T> void dispatch1(HookKey<T> key, T object)
   {
     if(DruidHook.SQL.equals(key)) {
-      printStream.println(object);
-      printStream.print(";");
-      return;
-    }
-    if (DruidHook.RESULTSET.equals(key)) {
-      printStream.println(object);
+      printStream.print(object);
+      printStream.println(";");
       printStream.println("!ok");
       return;
     }

diff --git a/processing/src/main/java/org/apache/druid/segment/data/CompressionStrategy.java b/processing/src/main/java/org/apache/druid/segment/data/CompressionStrategy.java
@@ -20,6 +20,7 @@
 package org.apache.druid.segment.data;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+
 import com.fasterxml.jackson.annotation.JsonValue;
 import com.github.luben.zstd.Zstd;
 import com.ning.compress.BufferRecycler;

diff --git a/sql/src/test/java/org/apache/druid/quidem/DruidHook.java b/sql/src/test/java/org/apache/druid/quidem/DruidHook.java
@@ -66,7 +66,6 @@ public boolean equals(Object obj)
 
   public static final HookKey<RelNode> DRUID_PLAN = new HookKey<>("druidPlan", RelNode.class);
   public static final HookKey<String> SQL = new HookKey<>("sql", String.class);
-  public static final HookKey<String> RESULTSET = new HookKey<>("resultSet", String.class);
 
   <T> void dispatch1(HookKey<T> key, T object);