mirror of https://github.com/apache/druid.git
workinhg stuff
This commit is contained in:
Zoltan Haindrich
parent
8ee41f58d0
commit
55b2051f9d
|
|
@ -0,0 +1,613 @@
|
||||||
|
/*
|
||||||
|
* 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.base.Supplier;
|
||||||
|
import com.google.common.base.Suppliers;
|
||||||
|
import com.google.common.collect.ImmutableMap;
|
||||||
|
import com.google.common.collect.ImmutableSet;
|
||||||
|
import com.google.common.collect.Sets;
|
||||||
|
import com.google.inject.AbstractModule;
|
||||||
|
import com.google.inject.Binder;
|
||||||
|
import com.google.inject.Guice;
|
||||||
|
import com.google.inject.Injector;
|
||||||
|
import com.google.inject.Module;
|
||||||
|
import com.google.inject.Provides;
|
||||||
|
import com.google.inject.TypeLiteral;
|
||||||
|
import com.google.inject.name.Names;
|
||||||
|
import com.google.inject.util.Modules;
|
||||||
|
import com.google.inject.util.Providers;
|
||||||
|
import org.apache.calcite.avatica.server.AbstractAvaticaHandler;
|
||||||
|
import org.apache.druid.cli.CliBroker2;
|
||||||
|
import org.apache.druid.discovery.DiscoveryDruidNode;
|
||||||
|
import org.apache.druid.discovery.DruidNodeDiscovery;
|
||||||
|
import org.apache.druid.discovery.DruidNodeDiscoveryProvider;
|
||||||
|
import org.apache.druid.discovery.NodeRole;
|
||||||
|
import org.apache.druid.guice.DruidInjectorBuilder;
|
||||||
|
import org.apache.druid.guice.LazySingleton;
|
||||||
|
import org.apache.druid.guice.StartupInjectorBuilder;
|
||||||
|
import org.apache.druid.guice.annotations.Json;
|
||||||
|
import org.apache.druid.initialization.DruidModule;
|
||||||
|
import org.apache.druid.java.util.common.FileUtils;
|
||||||
|
import org.apache.druid.java.util.common.StringUtils;
|
||||||
|
import org.apache.druid.java.util.common.io.Closer;
|
||||||
|
import org.apache.druid.java.util.emitter.service.ServiceEmitter;
|
||||||
|
import org.apache.druid.query.DefaultQueryConfig;
|
||||||
|
import org.apache.druid.query.QueryRunnerFactoryConglomerate;
|
||||||
|
import org.apache.druid.query.QuerySegmentWalker;
|
||||||
|
import org.apache.druid.query.lookup.LookupExtractorFactoryContainerProvider;
|
||||||
|
import org.apache.druid.segment.join.JoinableFactoryWrapper;
|
||||||
|
import org.apache.druid.server.DruidNode;
|
||||||
|
import org.apache.druid.server.QueryLifecycleFactory;
|
||||||
|
import org.apache.druid.server.QueryScheduler;
|
||||||
|
import org.apache.druid.server.QuerySchedulerProvider;
|
||||||
|
import org.apache.druid.server.SpecificSegmentsQuerySegmentWalker;
|
||||||
|
import org.apache.druid.server.log.RequestLogger;
|
||||||
|
import org.apache.druid.server.log.TestRequestLogger;
|
||||||
|
import org.apache.druid.server.metrics.NoopServiceEmitter;
|
||||||
|
import org.apache.druid.server.security.AuthenticatorMapper;
|
||||||
|
import org.apache.druid.server.security.AuthorizerMapper;
|
||||||
|
import org.apache.druid.server.security.Escalator;
|
||||||
|
import org.apache.druid.sql.avatica.AvaticaMonitor;
|
||||||
|
import org.apache.druid.sql.avatica.DruidAvaticaJsonHandler;
|
||||||
|
import org.apache.druid.sql.avatica.DruidMeta;
|
||||||
|
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.planner.CalciteRulesManager;
|
||||||
|
import org.apache.druid.sql.calcite.planner.CatalogResolver;
|
||||||
|
import org.apache.druid.sql.calcite.planner.PlannerConfig;
|
||||||
|
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.DruidSchemaCatalog;
|
||||||
|
import org.apache.druid.sql.calcite.schema.DruidSchemaName;
|
||||||
|
import org.apache.druid.sql.calcite.util.CacheTestHelperModule;
|
||||||
|
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.http.NameValuePair;
|
||||||
|
import org.apache.http.client.utils.URIBuilder;
|
||||||
|
import org.apache.http.client.utils.URLEncodedUtils;
|
||||||
|
import org.eclipse.jetty.server.Server;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import java.io.Closeable;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.net.URI;
|
||||||
|
import java.net.URISyntaxException;
|
||||||
|
import java.nio.charset.StandardCharsets;
|
||||||
|
import java.sql.Connection;
|
||||||
|
import java.sql.DriverManager;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Properties;
|
||||||
|
import java.util.Set;
|
||||||
|
import java.util.function.BooleanSupplier;
|
||||||
|
|
||||||
|
public class Launcher2
|
||||||
|
{
|
||||||
|
|
||||||
|
public static final String URI_PREFIX = "druidtest://";
|
||||||
|
public static final String DEFAULT_URI = URI_PREFIX + "/";
|
||||||
|
|
||||||
|
static final SqlTestFrameworkConfigStore CONFIG_STORE = new SqlTestFrameworkConfigStore();
|
||||||
|
|
||||||
|
public Launcher2()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
public Connection connect(String url, Properties info) throws SQLException
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
SqlTestFrameworkConfig config = buildConfigfromURIParams(url);
|
||||||
|
|
||||||
|
ConfigurationInstance ci = CONFIG_STORE.getConfigurationInstance(
|
||||||
|
config,
|
||||||
|
x -> new AvaticaBasedTestConnectionSupplier(x)
|
||||||
|
);
|
||||||
|
|
||||||
|
AvaticaJettyServer server = ci.framework.injector().getInstance(AvaticaJettyServer.class);
|
||||||
|
return server.getConnection(info);
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
throw new SQLException("Can't create testconnection", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static class AvaticaBasedConnectionModule implements DruidModule, Closeable
|
||||||
|
{
|
||||||
|
Closer closer = Closer.create();
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
@LazySingleton
|
||||||
|
public DruidSchemaCatalog getLookupNodeService(QueryRunnerFactoryConglomerate conglomerate,
|
||||||
|
SpecificSegmentsQuerySegmentWalker walker, PlannerConfig plannerConfig)
|
||||||
|
{
|
||||||
|
return CalciteTests.createMockRootSchema(
|
||||||
|
conglomerate,
|
||||||
|
walker,
|
||||||
|
plannerConfig,
|
||||||
|
CalciteTests.TEST_AUTHORIZER_MAPPER
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
@LazySingleton
|
||||||
|
public DruidConnectionExtras getConnectionExtras(ObjectMapper objectMapper)
|
||||||
|
{
|
||||||
|
return new DruidConnectionExtras.DruidConnectionExtrasImpl(objectMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Provides
|
||||||
|
@LazySingleton
|
||||||
|
public AvaticaJettyServer getAvaticaServer(DruidMeta druidMeta, DruidConnectionExtras druidConnectionExtras)
|
||||||
|
throws Exception
|
||||||
|
{
|
||||||
|
AvaticaJettyServer avaticaJettyServer = new AvaticaJettyServer(druidMeta, druidConnectionExtras);
|
||||||
|
closer.register(avaticaJettyServer);
|
||||||
|
return avaticaJettyServer;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void configure(Binder binder)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close() throws IOException
|
||||||
|
{
|
||||||
|
closer.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
static class AvaticaJettyServer implements Closeable
|
||||||
|
{
|
||||||
|
final DruidMeta druidMeta;
|
||||||
|
final Server server;
|
||||||
|
final String url;
|
||||||
|
final DruidConnectionExtras connectionExtras;
|
||||||
|
|
||||||
|
AvaticaJettyServer(final DruidMeta druidMeta, DruidConnectionExtras druidConnectionExtras) throws Exception
|
||||||
|
{
|
||||||
|
this.druidMeta = druidMeta;
|
||||||
|
server = new Server(0);
|
||||||
|
server.setHandler(getAvaticaHandler(druidMeta));
|
||||||
|
server.start();
|
||||||
|
url = StringUtils.format(
|
||||||
|
"jdbc:avatica:remote:url=%s",
|
||||||
|
new URIBuilder(server.getURI()).setPath(DruidAvaticaJsonHandler.AVATICA_PATH).build()
|
||||||
|
);
|
||||||
|
connectionExtras = druidConnectionExtras;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Connection getConnection(Properties info) throws SQLException
|
||||||
|
{
|
||||||
|
Connection realConnection = DriverManager.getConnection(url, info);
|
||||||
|
Connection proxyConnection = DynamicComposite.make(
|
||||||
|
realConnection,
|
||||||
|
Connection.class,
|
||||||
|
connectionExtras,
|
||||||
|
DruidConnectionExtras.class
|
||||||
|
);
|
||||||
|
return proxyConnection;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close()
|
||||||
|
{
|
||||||
|
druidMeta.closeAllConnections();
|
||||||
|
try {
|
||||||
|
server.stop();
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
throw new RuntimeException("Can't stop server", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
protected AbstractAvaticaHandler getAvaticaHandler(final DruidMeta druidMeta)
|
||||||
|
{
|
||||||
|
return new DruidAvaticaJsonHandler(
|
||||||
|
druidMeta,
|
||||||
|
new DruidNode("dummy", "dummy", false, 1, null, true, false),
|
||||||
|
new AvaticaMonitor()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static class AvaticaBasedTestConnectionSupplier implements QueryComponentSupplier
|
||||||
|
{
|
||||||
|
private QueryComponentSupplier delegate;
|
||||||
|
private AvaticaBasedConnectionModule connectionModule;
|
||||||
|
|
||||||
|
public AvaticaBasedTestConnectionSupplier(QueryComponentSupplier delegate)
|
||||||
|
{
|
||||||
|
this.delegate = delegate;
|
||||||
|
this.connectionModule = new AvaticaBasedConnectionModule();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void gatherProperties(Properties properties)
|
||||||
|
{
|
||||||
|
delegate.gatherProperties(properties);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void configureGuice(DruidInjectorBuilder builder)
|
||||||
|
{
|
||||||
|
delegate.configureGuice(builder);
|
||||||
|
TestRequestLogger testRequestLogger = new TestRequestLogger();
|
||||||
|
// builder.addModule(connectionModule);
|
||||||
|
builder.addModule(
|
||||||
|
binder -> {
|
||||||
|
binder.bindConstant().annotatedWith(Names.named("serviceName")).to("test");
|
||||||
|
binder.bindConstant().annotatedWith(Names.named("servicePort")).to(0);
|
||||||
|
binder.bindConstant().annotatedWith(Names.named("tlsServicePort")).to(-1);
|
||||||
|
binder.bind(AuthenticatorMapper.class).toInstance(CalciteTests.TEST_AUTHENTICATOR_MAPPER);
|
||||||
|
binder.bind(AuthorizerMapper.class).toInstance(CalciteTests.TEST_AUTHORIZER_MAPPER);
|
||||||
|
binder.bind(Escalator.class).toInstance(CalciteTests.TEST_AUTHENTICATOR_ESCALATOR);
|
||||||
|
binder.bind(RequestLogger.class).toInstance(testRequestLogger);
|
||||||
|
binder.bind(String.class)
|
||||||
|
.annotatedWith(DruidSchemaName.class)
|
||||||
|
.toInstance(CalciteTests.DRUID_SCHEMA_NAME);
|
||||||
|
binder.bind(ServiceEmitter.class).to(NoopServiceEmitter.class);
|
||||||
|
binder.bind(QuerySchedulerProvider.class).in(LazySingleton.class);
|
||||||
|
binder.bind(QueryScheduler.class)
|
||||||
|
.toProvider(QuerySchedulerProvider.class)
|
||||||
|
.in(LazySingleton.class);
|
||||||
|
binder.install(new SqlModule.SqlStatementFactoryModule());
|
||||||
|
binder.bind(new TypeLiteral<Supplier<DefaultQueryConfig>>()
|
||||||
|
{
|
||||||
|
}).toInstance(Suppliers.ofInstance(new DefaultQueryConfig(ImmutableMap.of())));
|
||||||
|
binder.bind(CalciteRulesManager.class).toInstance(new CalciteRulesManager(ImmutableSet.of()));
|
||||||
|
binder.bind(CatalogResolver.class).toInstance(CatalogResolver.NULL_RESOLVER);
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@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
|
||||||
|
{
|
||||||
|
connectionModule.close();
|
||||||
|
delegate.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public PlannerComponentSupplier getPlannerComponentSupplier()
|
||||||
|
{
|
||||||
|
return delegate.getPlannerComponentSupplier();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
protected File createTempFolder(String prefix)
|
||||||
|
{
|
||||||
|
File tempDir = FileUtils.createTempDir(prefix);
|
||||||
|
Runtime.getRuntime().addShutdownHook(new Thread()
|
||||||
|
{
|
||||||
|
@Override
|
||||||
|
public void run()
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
FileUtils.deleteDirectory(tempDir);
|
||||||
|
}
|
||||||
|
catch (IOException ex) {
|
||||||
|
ex.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return tempDir;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static SqlTestFrameworkConfig buildConfigfromURIParams(String url) throws SQLException
|
||||||
|
{
|
||||||
|
Map<String, String> queryParams;
|
||||||
|
queryParams = new HashMap<>();
|
||||||
|
try {
|
||||||
|
List<NameValuePair> params = URLEncodedUtils.parse(new URI(url), StandardCharsets.UTF_8);
|
||||||
|
for (NameValuePair pair : params) {
|
||||||
|
queryParams.put(pair.getName(), pair.getValue());
|
||||||
|
}
|
||||||
|
// possible caveat: duplicate entries overwrite earlier ones
|
||||||
|
}
|
||||||
|
catch (URISyntaxException e) {
|
||||||
|
throw new SQLException("Can't decode URI", e);
|
||||||
|
}
|
||||||
|
|
||||||
|
return new SqlTestFrameworkConfig(queryParams);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void main(String[] args) throws Exception
|
||||||
|
{
|
||||||
|
ConfigurationInstance ci = getCI();
|
||||||
|
|
||||||
|
AvaticaJettyServer server = ci.framework.injector().getInstance(AvaticaJettyServer.class);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private static ConfigurationInstance getCI() throws SQLException, Exception
|
||||||
|
{
|
||||||
|
SqlTestFrameworkConfig config = buildConfigfromURIParams("druidtest:///");
|
||||||
|
|
||||||
|
ConfigurationInstance ci = CONFIG_STORE.getConfigurationInstance(
|
||||||
|
config,
|
||||||
|
x -> new AvaticaBasedTestConnectionSupplier(x)
|
||||||
|
);
|
||||||
|
return ci;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void runIt() throws Exception
|
||||||
|
{
|
||||||
|
Launcher2.main1(null);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static Module propOverrideModuel()
|
||||||
|
{
|
||||||
|
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");
|
||||||
|
|
||||||
|
|
||||||
|
Module m = binder -> binder.bind(Properties.class).toInstance(localProps);
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class DiscovertModule extends AbstractModule {
|
||||||
|
|
||||||
|
DiscovertModule() {
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void configure()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
@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 getProvider() {
|
||||||
|
final DruidNode coordinatorNode = new DruidNode("test-coordinator", "dummy", false, 8081, null, true, false);
|
||||||
|
FakeDruidNodeDiscoveryProvider provider = new FakeDruidNodeDiscoveryProvider(
|
||||||
|
ImmutableMap.of(
|
||||||
|
NodeRole.COORDINATOR, new FakeDruidNodeDiscovery(ImmutableMap.of(NodeRole.COORDINATOR, coordinatorNode))
|
||||||
|
)
|
||||||
|
);
|
||||||
|
return provider;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A fake {@link DruidNodeDiscoveryProvider} for {@link #createMockSystemSchema}.
|
||||||
|
*/
|
||||||
|
private static class FakeDruidNodeDiscoveryProvider extends DruidNodeDiscoveryProvider
|
||||||
|
{
|
||||||
|
private final Map<NodeRole, FakeDruidNodeDiscovery> nodeDiscoveries;
|
||||||
|
|
||||||
|
public FakeDruidNodeDiscoveryProvider(Map<NodeRole, FakeDruidNodeDiscovery> nodeDiscoveries)
|
||||||
|
{
|
||||||
|
this.nodeDiscoveries = nodeDiscoveries;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BooleanSupplier getForNode(DruidNode node, NodeRole nodeRole)
|
||||||
|
{
|
||||||
|
boolean get = nodeDiscoveries.getOrDefault(nodeRole, new FakeDruidNodeDiscovery())
|
||||||
|
.getAllNodes()
|
||||||
|
.stream()
|
||||||
|
.anyMatch(x -> x.getDruidNode().equals(node));
|
||||||
|
return () -> get;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DruidNodeDiscovery getForNodeRole(NodeRole nodeRole)
|
||||||
|
{
|
||||||
|
return nodeDiscoveries.getOrDefault(nodeRole, new FakeDruidNodeDiscovery());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class FakeDruidNodeDiscovery implements DruidNodeDiscovery
|
||||||
|
{
|
||||||
|
private final Set<DiscoveryDruidNode> nodes;
|
||||||
|
|
||||||
|
FakeDruidNodeDiscovery()
|
||||||
|
{
|
||||||
|
this.nodes = new HashSet<>();
|
||||||
|
}
|
||||||
|
|
||||||
|
FakeDruidNodeDiscovery(Map<NodeRole, DruidNode> nodes)
|
||||||
|
{
|
||||||
|
this.nodes = Sets.newHashSetWithExpectedSize(nodes.size());
|
||||||
|
nodes.forEach((k, v) -> {
|
||||||
|
addNode(v, k);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Collection<DiscoveryDruidNode> getAllNodes()
|
||||||
|
{
|
||||||
|
return nodes;
|
||||||
|
}
|
||||||
|
|
||||||
|
void addNode(DruidNode node, NodeRole role)
|
||||||
|
{
|
||||||
|
final DiscoveryDruidNode discoveryNode = new DiscoveryDruidNode(node, role, ImmutableMap.of());
|
||||||
|
this.nodes.add(discoveryNode);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void registerListener(Listener listener)
|
||||||
|
{
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private static DiscovertModule discoverModule()
|
||||||
|
{
|
||||||
|
|
||||||
|
// DruidNodeDiscoveryProvider instance = ;
|
||||||
|
// Module m = binder -> binder.bind(DruidNodeDiscoveryProvider.class).toInstance(instance);
|
||||||
|
return new DiscovertModule();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
static class CustomStartupInjectorBuilder extends StartupInjectorBuilder {
|
||||||
|
|
||||||
|
private List<com.google.inject.Module> overrideModules =new ArrayList<>();
|
||||||
|
|
||||||
|
public CustomStartupInjectorBuilder()
|
||||||
|
{
|
||||||
|
Module m = propOverrideModuel();
|
||||||
|
addOverride(m);
|
||||||
|
// addOverride(binder -> {
|
||||||
|
// binder.bind(SSLClientConfig.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLClientConfig.class).annotatedWith(Global.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLClientConfig.class).annotatedWith(EscalatedGlobal.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLContext.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLContextProvider.class).annotatedWith(Global.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLContextProvider.class).annotatedWith(EscalatedGlobal.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLContext.class).annotatedWith(Global.class).toProvider(Providers.of(null));
|
||||||
|
// binder.bind(SSLContext.class).annotatedWith(EscalatedGlobal.class).toProvider(Providers.of(null));
|
||||||
|
// }
|
||||||
|
// );
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private void addOverride(com.google.inject.Module m)
|
||||||
|
{
|
||||||
|
overrideModules.add(m);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Injector build()
|
||||||
|
{
|
||||||
|
return Guice.createInjector(
|
||||||
|
Modules.override(modules)
|
||||||
|
.with( overrideModules.toArray(new com.google.inject.Module[0])));
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void main1(Object object) throws Exception
|
||||||
|
{
|
||||||
|
final Injector injector = new CustomStartupInjectorBuilder()
|
||||||
|
.forTests()
|
||||||
|
.build();
|
||||||
|
|
||||||
|
|
||||||
|
SqlTestFramework framework = getCI().extracted(
|
||||||
|
propOverrideModuel(),
|
||||||
|
discoverModule(),
|
||||||
|
binder -> binder.bind(BrokerSegmentMetadataCache.class).toProvider(Providers.of(null))
|
||||||
|
|
||||||
|
)
|
||||||
|
|
||||||
|
;
|
||||||
|
|
||||||
|
// SSLContextProvider u = injector.getInstance(SSLContextProvider.class);
|
||||||
|
// System.out.println(u);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
CliBroker2 c = new CliBroker2() {
|
||||||
|
protected List<? extends Module> getModules() {
|
||||||
|
List<Module> ret = new ArrayList<>();
|
||||||
|
ret.add(discoverModule());
|
||||||
|
ret.add(propOverrideModuel());
|
||||||
|
ret.add(framework.testSetupModule());
|
||||||
|
// ret.add(new AvaticaBasedConnectionModule());
|
||||||
|
ret.add(binder -> binder.bind(RequestLogger.class).toInstance(new TestRequestLogger()));
|
||||||
|
ret.add(CacheTestHelperModule.ResultCacheMode.DISABLED.makeModule());
|
||||||
|
ret.addAll(super.getModules());
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
};
|
||||||
|
framework.injector().injectMembers(c);
|
||||||
|
// c.configure(new Properties());
|
||||||
|
c.run();
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -25,7 +25,7 @@
|
||||||
</Console>
|
</Console>
|
||||||
</Appenders>
|
</Appenders>
|
||||||
<Loggers>
|
<Loggers>
|
||||||
<Root level="info">
|
<Root level="debug">
|
||||||
<AppenderRef ref="Console"/>
|
<AppenderRef ref="Console"/>
|
||||||
</Root>
|
</Root>
|
||||||
<Logger name="org.apache.druid.segment.metadata" level="debug" additivity="false">
|
<Logger name="org.apache.druid.segment.metadata" level="debug" additivity="false">
|
||||||
|
|
|
||||||
|
|
@ -137,7 +137,7 @@ public class StatusResource
|
||||||
|
|
||||||
public Status(Collection<DruidModule> modules)
|
public Status(Collection<DruidModule> modules)
|
||||||
{
|
{
|
||||||
this.version = getDruidVersion();
|
this.version = "30.0.0-asd1";//getDruidVersion();
|
||||||
this.modules = getExtensionVersions(modules);
|
this.modules = getExtensionVersions(modules);
|
||||||
this.memory = new Memory(JvmUtils.getRuntimeInfo());
|
this.memory = new Memory(JvmUtils.getRuntimeInfo());
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue