This commit is contained in:
Zoltan Haindrich 2024-05-21 14:00:02 +00:00
parent 08b73d1969
commit ba09e7d1de
4 changed files with 173 additions and 3 deletions

View File

@ -28,6 +28,8 @@ import javax.ws.rs.core.MediaType;
public class QuidemCapture public class QuidemCapture
{ {
private QuidemRecorder recorder = null;
@GET @GET
@Path("/") @Path("/")
@Produces(MediaType.TEXT_PLAIN) @Produces(MediaType.TEXT_PLAIN)
@ -37,10 +39,26 @@ public class QuidemCapture
} }
@GET @GET
@Path("/asd") @Path("/start")
@Produces(MediaType.TEXT_PLAIN) @Produces(MediaType.TEXT_PLAIN)
public String getSome1() public synchronized String getSome1()
{ {
return "Asd"; stopIfRunning();
start();
return null;
}
private void start()
{
recorder = new QuidemRecorder();
}
private void stopIfRunning()
{
if (recorder != null) {
recorder.close();
recorder = null;
}
} }
} }

View File

@ -0,0 +1,42 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.druid.quidem;
public class QuidemRecorder implements AutoCloseable, DruidHook
{
public QuidemRecorder()
{
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)
{
System.out.println(object);
}
}

View File

@ -39,6 +39,7 @@ import org.apache.calcite.tools.FrameworkConfig;
import org.apache.calcite.tools.Frameworks; import org.apache.calcite.tools.Frameworks;
import org.apache.druid.guice.annotations.Json; import org.apache.druid.guice.annotations.Json;
import org.apache.druid.math.expr.ExprMacroTable; import org.apache.druid.math.expr.ExprMacroTable;
import org.apache.druid.quidem.DruidHook;
import org.apache.druid.segment.join.JoinableFactoryWrapper; import org.apache.druid.segment.join.JoinableFactoryWrapper;
import org.apache.druid.server.security.Access; import org.apache.druid.server.security.Access;
import org.apache.druid.server.security.AuthConfig; import org.apache.druid.server.security.AuthConfig;
@ -112,6 +113,7 @@ public class PlannerFactory extends PlannerToolbox
queryContext, queryContext,
hook hook
); );
DruidHook.dispatch(DruidHook.SQL, sql);
return new DruidPlanner(buildFrameworkConfig(context), context, engine, hook); return new DruidPlanner(buildFrameworkConfig(context), context, engine, hook);
} }

View File

@ -0,0 +1,108 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.druid.quidem;
import org.apache.calcite.rel.RelNode;
import java.io.Closeable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
public interface DruidHook
{
static class HookKey<T>
{
private String label;
private Class<T> type;
public HookKey(String label, Class<T> type)
{
this.label = label;
this.type = type;
}
@Override
public int hashCode()
{
return Objects.hash(label, type);
}
@Override
public boolean equals(Object obj)
{
if (this == obj) {
return true;
}
if (obj == null) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
HookKey other = (HookKey) obj;
return Objects.equals(label, other.label) && Objects.equals(type, other.type);
}
}
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);
static Map<HookKey<?>, List<DruidHook>> GLOBAL = new HashMap<>();
public static void register(HookKey<?> label, DruidHook hook)
{
GLOBAL.computeIfAbsent(label, k -> new ArrayList<>()).add(hook);
}
public static void unregister(HookKey<?> key, DruidHook hook)
{
GLOBAL.get(key).remove(hook);
}
public static Closeable withHook(HookKey<?> key, DruidHook hook)
{
register(key, hook);
return new Closeable()
{
@Override
public void close()
{
unregister(key, hook);
}
};
}
public static <T> void dispatch(HookKey<T> key, T object)
{
List<DruidHook> hooks = GLOBAL.get(key);
if (hooks != null) {
for (DruidHook hook : hooks) {
hook.dispatch1(key, object);
}
}
}
}