mirror of https://github.com/apache/druid.git
add
This commit is contained in:
parent
08b73d1969
commit
ba09e7d1de
|
@ -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;
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
|
@ -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);
|
||||||
}
|
}
|
||||||
|
|
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue