mirror of https://github.com/apache/druid.git
undo integration-tests module changes
This commit is contained in:
Zoltan Haindrich
parent
90c88aeab0
commit
d75dcea4dc
|
|
@ -562,22 +562,7 @@
|
||||||
</execution>
|
</execution>
|
||||||
</executions>
|
</executions>
|
||||||
</plugin>
|
</plugin>
|
||||||
|
</plugins>
|
||||||
|
|
||||||
<plugin>
|
|
||||||
<groupId>org.codehaus.mojo</groupId>
|
|
||||||
<artifactId>exec-maven-plugin</artifactId>
|
|
||||||
<configuration>
|
|
||||||
<executable>java</executable>
|
|
||||||
<arguments>
|
|
||||||
<argument>-classpath</argument>
|
|
||||||
<classpath/>
|
|
||||||
<argument>org.apache.druid.quidem.Launcher</argument>
|
|
||||||
</arguments>
|
|
||||||
</configuration>
|
|
||||||
</plugin>
|
|
||||||
|
|
||||||
</plugins>
|
|
||||||
</build>
|
</build>
|
||||||
|
|
||||||
<profiles>
|
<profiles>
|
||||||
|
|
|
||||||
|
|
@ -1,373 +0,0 @@
|
||||||
/*
|
|
||||||
* 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 com.fasterxml.jackson.databind.ObjectMapper;
|
|
||||||
import com.google.common.collect.ImmutableList;
|
|
||||||
import com.google.common.collect.ImmutableSet;
|
|
||||||
import com.google.inject.AbstractModule;
|
|
||||||
import com.google.inject.Injector;
|
|
||||||
import com.google.inject.Key;
|
|
||||||
import com.google.inject.Module;
|
|
||||||
import com.google.inject.Provides;
|
|
||||||
import com.google.inject.name.Names;
|
|
||||||
import org.apache.druid.cli.CliBroker;
|
|
||||||
import org.apache.druid.cli.QueryJettyServerInitializer;
|
|
||||||
import org.apache.druid.client.BrokerSegmentWatcherConfig;
|
|
||||||
import org.apache.druid.client.BrokerServerView;
|
|
||||||
import org.apache.druid.client.InternalQueryConfig;
|
|
||||||
import org.apache.druid.client.TimelineServerView;
|
|
||||||
import org.apache.druid.client.selector.CustomTierSelectorStrategyConfig;
|
|
||||||
import org.apache.druid.client.selector.ServerSelectorStrategy;
|
|
||||||
import org.apache.druid.client.selector.TierSelectorStrategy;
|
|
||||||
import org.apache.druid.curator.CuratorModule;
|
|
||||||
import org.apache.druid.curator.discovery.DiscoveryModule;
|
|
||||||
import org.apache.druid.discovery.DruidNodeDiscoveryProvider;
|
|
||||||
import org.apache.druid.guice.AnnouncerModule;
|
|
||||||
import org.apache.druid.guice.BrokerProcessingModule;
|
|
||||||
import org.apache.druid.guice.BrokerServiceModule;
|
|
||||||
import org.apache.druid.guice.CoordinatorDiscoveryModule;
|
|
||||||
import org.apache.druid.guice.DruidInjectorBuilder;
|
|
||||||
import org.apache.druid.guice.ExpressionModule;
|
|
||||||
import org.apache.druid.guice.ExtensionsModule;
|
|
||||||
import org.apache.druid.guice.FirehoseModule;
|
|
||||||
import org.apache.druid.guice.JacksonConfigManagerModule;
|
|
||||||
import org.apache.druid.guice.JavaScriptModule;
|
|
||||||
import org.apache.druid.guice.Jerseys;
|
|
||||||
import org.apache.druid.guice.JoinableFactoryModule;
|
|
||||||
import org.apache.druid.guice.JsonConfigProvider;
|
|
||||||
import org.apache.druid.guice.LazySingleton;
|
|
||||||
import org.apache.druid.guice.LifecycleModule;
|
|
||||||
import org.apache.druid.guice.LocalDataStorageDruidModule;
|
|
||||||
import org.apache.druid.guice.MetadataConfigModule;
|
|
||||||
import org.apache.druid.guice.NestedDataModule;
|
|
||||||
import org.apache.druid.guice.QueryRunnerFactoryModule;
|
|
||||||
import org.apache.druid.guice.SegmentWranglerModule;
|
|
||||||
import org.apache.druid.guice.ServerModule;
|
|
||||||
import org.apache.druid.guice.ServerTypeConfig;
|
|
||||||
import org.apache.druid.guice.ServerViewModule;
|
|
||||||
import org.apache.druid.guice.StartupLoggingModule;
|
|
||||||
import org.apache.druid.guice.StorageNodeModule;
|
|
||||||
import org.apache.druid.guice.annotations.Client;
|
|
||||||
import org.apache.druid.guice.annotations.EscalatedClient;
|
|
||||||
import org.apache.druid.guice.annotations.Json;
|
|
||||||
import org.apache.druid.guice.http.HttpClientModule;
|
|
||||||
import org.apache.druid.guice.security.AuthenticatorModule;
|
|
||||||
import org.apache.druid.guice.security.AuthorizerModule;
|
|
||||||
import org.apache.druid.guice.security.DruidAuthModule;
|
|
||||||
import org.apache.druid.initialization.CoreInjectorBuilder;
|
|
||||||
import org.apache.druid.initialization.Log4jShutterDownerModule;
|
|
||||||
import org.apache.druid.initialization.ServerInjectorBuilder;
|
|
||||||
import org.apache.druid.initialization.TombstoneDataStorageModule;
|
|
||||||
import org.apache.druid.java.util.common.io.Closer;
|
|
||||||
import org.apache.druid.java.util.emitter.service.ServiceEmitter;
|
|
||||||
import org.apache.druid.metadata.storage.derby.DerbyMetadataStorageDruidModule;
|
|
||||||
import org.apache.druid.query.QueryRunnerFactoryConglomerate;
|
|
||||||
import org.apache.druid.query.QuerySegmentWalker;
|
|
||||||
import org.apache.druid.query.RetryQueryRunnerConfig;
|
|
||||||
import org.apache.druid.query.lookup.LookupExtractorFactoryContainerProvider;
|
|
||||||
import org.apache.druid.rpc.guice.ServiceClientModule;
|
|
||||||
import org.apache.druid.segment.join.JoinableFactoryWrapper;
|
|
||||||
import org.apache.druid.segment.writeout.SegmentWriteOutMediumModule;
|
|
||||||
import org.apache.druid.server.BrokerQueryResource;
|
|
||||||
import org.apache.druid.server.ClientInfoResource;
|
|
||||||
import org.apache.druid.server.DruidNode;
|
|
||||||
import org.apache.druid.server.QueryLifecycleFactory;
|
|
||||||
import org.apache.druid.server.QuerySchedulerProvider;
|
|
||||||
import org.apache.druid.server.ResponseContextConfig;
|
|
||||||
import org.apache.druid.server.SpecificSegmentsQuerySegmentWalker;
|
|
||||||
import org.apache.druid.server.SubqueryGuardrailHelper;
|
|
||||||
import org.apache.druid.server.SubqueryGuardrailHelperProvider;
|
|
||||||
import org.apache.druid.server.coordination.ServerType;
|
|
||||||
import org.apache.druid.server.http.BrokerResource;
|
|
||||||
import org.apache.druid.server.http.SelfDiscoveryResource;
|
|
||||||
import org.apache.druid.server.initialization.AuthorizerMapperModule;
|
|
||||||
import org.apache.druid.server.initialization.ExternalStorageAccessSecurityModule;
|
|
||||||
import org.apache.druid.server.initialization.jetty.JettyServerInitializer;
|
|
||||||
import org.apache.druid.server.initialization.jetty.JettyServerModule;
|
|
||||||
import org.apache.druid.server.log.NoopRequestLogger;
|
|
||||||
import org.apache.druid.server.log.RequestLogger;
|
|
||||||
import org.apache.druid.server.metrics.NoopServiceEmitter;
|
|
||||||
import org.apache.druid.server.metrics.QueryCountStatsProvider;
|
|
||||||
import org.apache.druid.server.metrics.SubqueryCountStatsProvider;
|
|
||||||
import org.apache.druid.server.router.TieredBrokerConfig;
|
|
||||||
import org.apache.druid.server.security.AuthenticatorMapper;
|
|
||||||
import org.apache.druid.server.security.Escalator;
|
|
||||||
import org.apache.druid.server.security.TLSCertificateCheckerModule;
|
|
||||||
import org.apache.druid.sql.calcite.planner.CalciteRulesManager;
|
|
||||||
import org.apache.druid.sql.calcite.planner.CatalogResolver;
|
|
||||||
import org.apache.druid.sql.calcite.run.NativeSqlEngine;
|
|
||||||
import org.apache.druid.sql.calcite.run.SqlEngine;
|
|
||||||
import org.apache.druid.sql.calcite.schema.BrokerSegmentMetadataCache;
|
|
||||||
import org.apache.druid.sql.calcite.schema.DruidSchemaName;
|
|
||||||
import org.apache.druid.sql.calcite.util.CalciteTests;
|
|
||||||
import org.apache.druid.sql.calcite.util.SqlTestFramework;
|
|
||||||
import org.apache.druid.sql.calcite.util.SqlTestFramework.Builder;
|
|
||||||
import org.apache.druid.sql.calcite.util.SqlTestFramework.PlannerComponentSupplier;
|
|
||||||
import org.apache.druid.sql.calcite.util.SqlTestFramework.QueryComponentSupplier;
|
|
||||||
import org.apache.druid.sql.guice.SqlModule;
|
|
||||||
import org.apache.druid.storage.StorageConnectorModule;
|
|
||||||
import org.apache.druid.timeline.PruneLoadSpec;
|
|
||||||
import org.eclipse.jetty.server.Server;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Properties;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* A wrapper class to expose a {@link QueryComponentSupplier} as a Broker service.
|
|
||||||
*/
|
|
||||||
public class ExposedAsBrokerQueryComponentSupplierWrapper implements QueryComponentSupplier
|
|
||||||
{
|
|
||||||
private QueryComponentSupplier delegate;
|
|
||||||
|
|
||||||
public ExposedAsBrokerQueryComponentSupplierWrapper(QueryComponentSupplier delegate)
|
|
||||||
{
|
|
||||||
this.delegate = delegate;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void gatherProperties(Properties properties)
|
|
||||||
{
|
|
||||||
delegate.gatherProperties(properties);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void configureGuice(DruidInjectorBuilder builder)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void configureGuice(CoreInjectorBuilder builder, List<Module> overrideModules)
|
|
||||||
{
|
|
||||||
delegate.configureGuice(builder);
|
|
||||||
|
|
||||||
installForServerModules(builder);
|
|
||||||
|
|
||||||
overrideModules.addAll(ExposedAsBrokerQueryComponentSupplierWrapper.brokerModules());
|
|
||||||
overrideModules.add(new BrokerTestModule());
|
|
||||||
builder.add(QuidemCaptureModule.class);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public QueryRunnerFactoryConglomerate createCongolmerate(Builder builder, Closer closer)
|
|
||||||
{
|
|
||||||
return delegate.createCongolmerate(builder, closer);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public SpecificSegmentsQuerySegmentWalker createQuerySegmentWalker(QueryRunnerFactoryConglomerate conglomerate,
|
|
||||||
JoinableFactoryWrapper joinableFactory, Injector injector)
|
|
||||||
{
|
|
||||||
return delegate.createQuerySegmentWalker(conglomerate, joinableFactory, injector);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public SqlEngine createEngine(QueryLifecycleFactory qlf, ObjectMapper objectMapper, Injector injector)
|
|
||||||
{
|
|
||||||
return delegate.createEngine(qlf, objectMapper, injector);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void configureJsonMapper(ObjectMapper mapper)
|
|
||||||
{
|
|
||||||
delegate.configureJsonMapper(mapper);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public JoinableFactoryWrapper createJoinableFactoryWrapper(LookupExtractorFactoryContainerProvider lookupProvider)
|
|
||||||
{
|
|
||||||
return delegate.createJoinableFactoryWrapper(lookupProvider);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void finalizeTestFramework(SqlTestFramework sqlTestFramework)
|
|
||||||
{
|
|
||||||
delegate.finalizeTestFramework(sqlTestFramework);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void close() throws IOException
|
|
||||||
{
|
|
||||||
delegate.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public PlannerComponentSupplier getPlannerComponentSupplier()
|
|
||||||
{
|
|
||||||
return delegate.getPlannerComponentSupplier();
|
|
||||||
}
|
|
||||||
|
|
||||||
public class BrokerTestModule extends AbstractModule
|
|
||||||
{
|
|
||||||
@Override
|
|
||||||
protected void configure()
|
|
||||||
{
|
|
||||||
bind(AuthenticatorMapper.class).toInstance(CalciteTests.TEST_AUTHENTICATOR_MAPPER);
|
|
||||||
bind(Escalator.class).toInstance(CalciteTests.TEST_AUTHENTICATOR_ESCALATOR);
|
|
||||||
bind(RequestLogger.class).toInstance(new NoopRequestLogger());
|
|
||||||
bind(String.class)
|
|
||||||
.annotatedWith(DruidSchemaName.class)
|
|
||||||
.toInstance(CalciteTests.DRUID_SCHEMA_NAME);
|
|
||||||
bind(ServiceEmitter.class).to(NoopServiceEmitter.class);
|
|
||||||
bind(QuerySchedulerProvider.class).in(LazySingleton.class);
|
|
||||||
bind(CalciteRulesManager.class).toInstance(new CalciteRulesManager(ImmutableSet.of()));
|
|
||||||
bind(CatalogResolver.class).toInstance(CatalogResolver.NULL_RESOLVER);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
|
||||||
@LazySingleton
|
|
||||||
public BrokerSegmentMetadataCache provideCache()
|
|
||||||
{
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
|
||||||
@LazySingleton
|
|
||||||
public Properties getProps()
|
|
||||||
{
|
|
||||||
Properties localProps = new Properties();
|
|
||||||
localProps.put("druid.enableTlsPort", "false");
|
|
||||||
localProps.put("druid.zk.service.enabled", "false");
|
|
||||||
localProps.put("druid.plaintextPort", "12345");
|
|
||||||
localProps.put("druid.host", "localhost");
|
|
||||||
localProps.put("druid.broker.segment.awaitInitializationOnStart", "false");
|
|
||||||
return localProps;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
|
||||||
@LazySingleton
|
|
||||||
public SqlEngine createMockSqlEngine(
|
|
||||||
final QuerySegmentWalker walker,
|
|
||||||
final QueryRunnerFactoryConglomerate conglomerate,
|
|
||||||
@Json ObjectMapper jsonMapper)
|
|
||||||
{
|
|
||||||
return new NativeSqlEngine(CalciteTests.createMockQueryLifecycleFactory(walker, conglomerate), jsonMapper);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Provides
|
|
||||||
@LazySingleton
|
|
||||||
DruidNodeDiscoveryProvider getDruidNodeDiscoveryProvider()
|
|
||||||
{
|
|
||||||
final DruidNode coordinatorNode = CalciteTests.mockCoordinatorNode();
|
|
||||||
return CalciteTests.mockDruidNodeDiscoveryProvider(coordinatorNode);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Closely related to {@link CoreInjectorBuilder#forServer()}
|
|
||||||
*/
|
|
||||||
private void installForServerModules(CoreInjectorBuilder builder)
|
|
||||||
{
|
|
||||||
builder.add(
|
|
||||||
new Log4jShutterDownerModule(),
|
|
||||||
new LifecycleModule(),
|
|
||||||
ExtensionsModule.SecondaryModule.class,
|
|
||||||
new DruidAuthModule(),
|
|
||||||
TLSCertificateCheckerModule.class,
|
|
||||||
// EmitterModule.class,
|
|
||||||
HttpClientModule.global(),
|
|
||||||
HttpClientModule.escalatedGlobal(),
|
|
||||||
new HttpClientModule("druid.broker.http", Client.class, true),
|
|
||||||
new HttpClientModule("druid.broker.http", EscalatedClient.class, true),
|
|
||||||
new CuratorModule(),
|
|
||||||
new AnnouncerModule(),
|
|
||||||
new SegmentWriteOutMediumModule(),
|
|
||||||
new ServerModule(),
|
|
||||||
new StorageNodeModule(),
|
|
||||||
new JettyServerModule(),
|
|
||||||
new ExpressionModule(),
|
|
||||||
new NestedDataModule(),
|
|
||||||
new DiscoveryModule(),
|
|
||||||
new ServerViewModule(),
|
|
||||||
new MetadataConfigModule(),
|
|
||||||
new DerbyMetadataStorageDruidModule(),
|
|
||||||
new JacksonConfigManagerModule(),
|
|
||||||
new CoordinatorDiscoveryModule(),
|
|
||||||
new LocalDataStorageDruidModule(),
|
|
||||||
new TombstoneDataStorageModule(),
|
|
||||||
new FirehoseModule(),
|
|
||||||
new JavaScriptModule(),
|
|
||||||
new AuthenticatorModule(),
|
|
||||||
new AuthorizerModule(),
|
|
||||||
new AuthorizerMapperModule(),
|
|
||||||
new StartupLoggingModule(),
|
|
||||||
new ExternalStorageAccessSecurityModule(),
|
|
||||||
new ServiceClientModule(),
|
|
||||||
new StorageConnectorModule(),
|
|
||||||
new SqlModule(),
|
|
||||||
ServerInjectorBuilder.registerNodeRoleModule(ImmutableSet.of())
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Closely related to {@link CliBroker#getModules}.
|
|
||||||
*/
|
|
||||||
static List<? extends Module> brokerModules()
|
|
||||||
{
|
|
||||||
return ImmutableList.of(
|
|
||||||
new BrokerProcessingModule(),
|
|
||||||
new QueryRunnerFactoryModule(),
|
|
||||||
new SegmentWranglerModule(),
|
|
||||||
new JoinableFactoryModule(),
|
|
||||||
new BrokerServiceModule(),
|
|
||||||
binder -> {
|
|
||||||
|
|
||||||
binder.bindConstant().annotatedWith(Names.named("serviceName")).to(
|
|
||||||
TieredBrokerConfig.DEFAULT_BROKER_SERVICE_NAME
|
|
||||||
);
|
|
||||||
binder.bindConstant().annotatedWith(Names.named("servicePort")).to(8082);
|
|
||||||
binder.bindConstant().annotatedWith(Names.named("tlsServicePort")).to(8282);
|
|
||||||
binder.bindConstant().annotatedWith(PruneLoadSpec.class).to(true);
|
|
||||||
binder.bind(ResponseContextConfig.class).toInstance(ResponseContextConfig.newConfig(false));
|
|
||||||
|
|
||||||
binder.bind(TimelineServerView.class).to(BrokerServerView.class).in(LazySingleton.class);
|
|
||||||
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.select", TierSelectorStrategy.class);
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.select.tier.custom", CustomTierSelectorStrategyConfig.class);
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.balancer", ServerSelectorStrategy.class);
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.retryPolicy", RetryQueryRunnerConfig.class);
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.segment", BrokerSegmentWatcherConfig.class);
|
|
||||||
JsonConfigProvider.bind(binder, "druid.broker.internal.query.config", InternalQueryConfig.class);
|
|
||||||
binder.bind(JettyServerInitializer.class).to(QueryJettyServerInitializer.class).in(LazySingleton.class);
|
|
||||||
|
|
||||||
binder.bind(BrokerQueryResource.class).in(LazySingleton.class);
|
|
||||||
Jerseys.addResource(binder, BrokerQueryResource.class);
|
|
||||||
binder.bind(SubqueryGuardrailHelper.class).toProvider(SubqueryGuardrailHelperProvider.class);
|
|
||||||
binder.bind(QueryCountStatsProvider.class).to(BrokerQueryResource.class).in(LazySingleton.class);
|
|
||||||
binder.bind(SubqueryCountStatsProvider.class).toInstance(new SubqueryCountStatsProvider());
|
|
||||||
Jerseys.addResource(binder, BrokerResource.class);
|
|
||||||
Jerseys.addResource(binder, ClientInfoResource.class);
|
|
||||||
|
|
||||||
LifecycleModule.register(binder, BrokerQueryResource.class);
|
|
||||||
|
|
||||||
LifecycleModule.register(binder, Server.class);
|
|
||||||
binder.bind(ServerTypeConfig.class).toInstance(new ServerTypeConfig(ServerType.BROKER));
|
|
||||||
|
|
||||||
binder.bind(String.class)
|
|
||||||
.annotatedWith(DruidSchemaName.class)
|
|
||||||
.toInstance(CalciteTests.DRUID_SCHEMA_NAME);
|
|
||||||
|
|
||||||
Jerseys.addResource(binder, SelfDiscoveryResource.class);
|
|
||||||
LifecycleModule.registerKey(binder, Key.get(SelfDiscoveryResource.class));
|
|
||||||
}
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,69 +0,0 @@
|
||||||
/*
|
|
||||||
* 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.druid.cli.GuiceRunnable;
|
|
||||||
import org.apache.druid.java.util.common.lifecycle.Lifecycle;
|
|
||||||
import org.apache.druid.java.util.common.logger.Logger;
|
|
||||||
import org.apache.druid.sql.calcite.SqlTestFrameworkConfig;
|
|
||||||
import org.apache.druid.sql.calcite.SqlTestFrameworkConfig.ConfigurationInstance;
|
|
||||||
import org.apache.druid.sql.calcite.SqlTestFrameworkConfig.SqlTestFrameworkConfigStore;
|
|
||||||
import org.apache.druid.sql.calcite.util.SqlTestFramework;
|
|
||||||
|
|
||||||
public class Launcher
|
|
||||||
{
|
|
||||||
static final SqlTestFrameworkConfigStore CONFIG_STORE = new SqlTestFrameworkConfigStore(
|
|
||||||
x -> new ExposedAsBrokerQueryComponentSupplierWrapper(x)
|
|
||||||
);
|
|
||||||
private static final String QUIDEM_URI = "quidem.uri";
|
|
||||||
private static Logger log = new Logger(Launcher.class);
|
|
||||||
private final SqlTestFramework framework;
|
|
||||||
private final ConfigurationInstance configurationInstance;
|
|
||||||
private Lifecycle lifecycle;
|
|
||||||
|
|
||||||
public Launcher(String uri) throws Exception
|
|
||||||
{
|
|
||||||
SqlTestFrameworkConfig config = SqlTestFrameworkConfig.fromURL(uri);
|
|
||||||
configurationInstance = CONFIG_STORE.getConfigurationInstance(config);
|
|
||||||
framework = configurationInstance.framework;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void start()
|
|
||||||
{
|
|
||||||
lifecycle = GuiceRunnable.initLifecycle(framework.injector(), log);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void shutdown()
|
|
||||||
{
|
|
||||||
lifecycle.stop();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void main(String[] args) throws Exception
|
|
||||||
{
|
|
||||||
if(true) {
|
|
||||||
throw new RuntimeException("Asd");
|
|
||||||
}
|
|
||||||
String quidemUri = System.getProperty(QUIDEM_URI, "druidtest:///");
|
|
||||||
|
|
||||||
Launcher launcher = new Launcher(quidemUri );
|
|
||||||
launcher.start();
|
|
||||||
launcher.lifecycle.join();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,34 +0,0 @@
|
||||||
/*
|
|
||||||
* 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 com.google.inject.Binder;
|
|
||||||
import com.google.inject.Module;
|
|
||||||
import org.apache.druid.guice.Jerseys;
|
|
||||||
|
|
||||||
public class QuidemCaptureModule implements Module
|
|
||||||
{
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void configure(Binder binder)
|
|
||||||
{
|
|
||||||
Jerseys.addResource(binder, QuidemCaptureResource.class);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,66 +0,0 @@
|
||||||
/*
|
|
||||||
* 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 javax.ws.rs.GET;
|
|
||||||
import javax.ws.rs.Path;
|
|
||||||
import javax.ws.rs.Produces;
|
|
||||||
import javax.ws.rs.core.MediaType;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.PrintStream;
|
|
||||||
|
|
||||||
@Path("/quidem")
|
|
||||||
public class QuidemCaptureResource
|
|
||||||
{
|
|
||||||
private QuidemRecorder recorder = null;
|
|
||||||
|
|
||||||
@GET
|
|
||||||
@Path("/")
|
|
||||||
@Produces(MediaType.TEXT_PLAIN)
|
|
||||||
public String getSome()
|
|
||||||
{
|
|
||||||
return "Asd";
|
|
||||||
}
|
|
||||||
|
|
||||||
@GET
|
|
||||||
@Path("/start")
|
|
||||||
@Produces(MediaType.TEXT_PLAIN)
|
|
||||||
public synchronized String getSome1() throws IOException
|
|
||||||
{
|
|
||||||
stopIfRunning();
|
|
||||||
start();
|
|
||||||
return recorder.toString();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void start() throws IOException
|
|
||||||
{
|
|
||||||
recorder = new QuidemRecorder(new PrintStream("/tmp/new.iq"));
|
|
||||||
}
|
|
||||||
|
|
||||||
private void stopIfRunning()
|
|
||||||
{
|
|
||||||
if (recorder != null) {
|
|
||||||
recorder.close();
|
|
||||||
recorder = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,54 +0,0 @@
|
||||||
/*
|
|
||||||
* 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.druid.sql.calcite.run.DruidHook;
|
|
||||||
|
|
||||||
import java.io.PrintStream;
|
|
||||||
|
|
||||||
public class QuidemRecorder implements AutoCloseable, DruidHook
|
|
||||||
{
|
|
||||||
private PrintStream printStream;
|
|
||||||
|
|
||||||
public QuidemRecorder(PrintStream printStream)
|
|
||||||
{
|
|
||||||
this.printStream = printStream;
|
|
||||||
printStream.println("#started");
|
|
||||||
printStream.println("!connect druidtest:///");
|
|
||||||
DruidHook.register(DruidHook.SQL, this);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void close()
|
|
||||||
{
|
|
||||||
DruidHook.unregister(DruidHook.SQL, this);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public <T> void invoke(HookKey<T> key, T object)
|
|
||||||
{
|
|
||||||
if (DruidHook.SQL.equals(key)) {
|
|
||||||
printStream.print(object);
|
|
||||||
printStream.println(";");
|
|
||||||
printStream.println("!ok");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,75 +0,0 @@
|
||||||
/*
|
|
||||||
* 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.junit.AfterClass;
|
|
||||||
import org.junit.BeforeClass;
|
|
||||||
import org.junit.Test;
|
|
||||||
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.net.URI;
|
|
||||||
import java.net.http.HttpClient;
|
|
||||||
import java.net.http.HttpRequest;
|
|
||||||
import java.net.http.HttpRequest.BodyPublishers;
|
|
||||||
import java.net.http.HttpResponse;
|
|
||||||
import static org.testng.Assert.assertEquals;
|
|
||||||
|
|
||||||
public class LauncherSmokeTest
|
|
||||||
{
|
|
||||||
private static Launcher launcher;
|
|
||||||
|
|
||||||
@BeforeClass
|
|
||||||
public static void setUp() throws Exception
|
|
||||||
{
|
|
||||||
launcher = new Launcher("druidtest:///");
|
|
||||||
launcher.start();
|
|
||||||
}
|
|
||||||
|
|
||||||
@AfterClass
|
|
||||||
public static void tearDown()
|
|
||||||
{
|
|
||||||
launcher.shutdown();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void chkSelectFromFoo() throws IOException, InterruptedException
|
|
||||||
{
|
|
||||||
HttpRequest request = HttpRequest.newBuilder()
|
|
||||||
.uri(URI.create("http://localhost:12345/druid/v2/sql"))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.POST(BodyPublishers.ofString("{\"query\":\"Select * from foo\"}"))
|
|
||||||
.build();
|
|
||||||
HttpClient hc = HttpClient.newHttpClient();
|
|
||||||
HttpResponse<String> a = hc.send(request, HttpResponse.BodyHandlers.ofString());
|
|
||||||
assertEquals(200, a.statusCode());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void chkStatusWorks() throws IOException, InterruptedException
|
|
||||||
{
|
|
||||||
HttpRequest request = HttpRequest.newBuilder()
|
|
||||||
.uri(URI.create("http://localhost:12345/status"))
|
|
||||||
.header("Content-Type", "application/json")
|
|
||||||
.GET()
|
|
||||||
.build();
|
|
||||||
HttpClient hc = HttpClient.newHttpClient();
|
|
||||||
HttpResponse<String> a = hc.send(request, HttpResponse.BodyHandlers.ofString());
|
|
||||||
assertEquals(200, a.statusCode());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue