u[pdate

integration-tests/src/main/java/org/apache/druid/quidem/ExposedAsBrokerQueryComponentSupplierWrapper.java

@@ -185,6 +185,7 @@ public class ExposedAsBrokerQueryComponentSupplierWrapper implements QueryCompon
 
     overrideModules.addAll(ExposedAsBrokerQueryComponentSupplierWrapper.brokerModules());
     overrideModules.add(new DiscovertModule());
+    builder.add(QuidemCaptureModule.class);
 
   }
 

integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java

@@ -99,7 +99,7 @@ public class Launcher
       chk1();
       chkStatus();
 
-      lifecycle.stop();
+      lifecycle.join();
     } else {
 
     }

integration-tests/src/main/java/org/apache/druid/quidem/QuidemCaptureModule.java

@@ -29,6 +29,6 @@ public class QuidemCaptureModule implements Module
   @Override
   public void configure(Binder binder)
   {
-    Jerseys.addResource(binder, QuidemCapture.class);
+    Jerseys.addResource(binder, QuidemCaptureResource.class);
   }
 }

integration-tests/src/main/java/org/apache/druid/quidem/QuidemCaptureResource.java

@@ -24,10 +24,12 @@ import javax.ws.rs.Path;
 import javax.ws.rs.Produces;
 import javax.ws.rs.core.MediaType;
 
-@Path("/quidem")
-public class QuidemCapture
-{
+import java.io.IOException;
+import java.io.PrintStream;
 
+@Path("/quidem")
+public class QuidemCaptureResource
+{
   private QuidemRecorder recorder = null;
 
   @GET
@@ -41,16 +43,16 @@ public class QuidemCapture
   @GET
   @Path("/start")
   @Produces(MediaType.TEXT_PLAIN)
-  public synchronized String getSome1()
+  public synchronized String getSome1() throws IOException
   {
     stopIfRunning();
     start();
-    return null;
+    return recorder.toString();
   }
 
-  private void start()
+  private void start() throws IOException
   {
-    recorder = new QuidemRecorder();
+    recorder = new QuidemRecorder(new PrintStream("/tmp/new.iq"));
   }
 
   private void stopIfRunning()

integration-tests/src/main/java/org/apache/druid/quidem/QuidemRecorder.java

@@ -19,10 +19,16 @@
 
 package org.apache.druid.quidem;
 
+import java.io.PrintStream;
+
 public class QuidemRecorder implements AutoCloseable, DruidHook
 {
-  public QuidemRecorder()
+  private PrintStream printStream;
+
+  public QuidemRecorder(PrintStream printStream)
   {
+    this.printStream = printStream;
+    printStream.println("#started");
     DruidHook.register(DruidHook.SQL, this);
     DruidHook.register(DruidHook.RESULTSET, this);
   }
@@ -37,6 +43,15 @@ public class QuidemRecorder implements AutoCloseable, DruidHook
   @Override
   public <T> void dispatch1(HookKey<T> key, T object)
   {
-    System.out.println(object);
+    if(DruidHook.SQL.equals(key)) {
+      printStream.println(object);
+      printStream.print(";");
+      return;
+    }
+    if (DruidHook.RESULTSET.equals(key)) {
+      printStream.println(object);
+      printStream.println("!ok");
+      return;
+    }
   }
 }

sql/src/test/java/org/apache/druid/quidem/DruidHook.java

@@ -104,5 +104,4 @@ public interface DruidHook
       }
     }
   }
-
 }