org.elasticsearch.cluster.ClusterModule Java Examples

private void registerSettings(ClusterModule clusterModule, Collection<? extends Setting> settings) {
    for (Setting setting : settings) {
        /**
         * validation is done in
         * {@link io.crate.analyze.SettingsAppliers.AbstractSettingsApplier#apply(org.elasticsearch.common.settings.Settings.Builder, Object[], io.crate.sql.tree.Expression)}
         * here we use Validator.EMPTY, since ES ignores invalid settings silently
         **/

        clusterModule.registerClusterDynamicSetting(setting.settingName(), Validator.EMPTY);
        registerSettings(clusterModule, setting.children());
    }
}
 

public static MapperService createMapperService(String type) throws IOException {
    IndexMetaData indexMetaData = IndexMetaData.builder("test")
        .settings(Settings.builder()
            .put(IndexMetaData.SETTING_VERSION_CREATED, Version.CURRENT)
            .put(IndexMetaData.SETTING_NUMBER_OF_SHARDS, 1).put(IndexMetaData.SETTING_NUMBER_OF_REPLICAS, 1))
        .putMapping(type, "{\"properties\": {}}")
        .build();
    MapperService mapperService = MapperTestUtils.newMapperService(new NamedXContentRegistry(ClusterModule.getNamedXWriteables()),
        createTempDir(), Settings.EMPTY, "test");
    mapperService.merge(indexMetaData, MapperService.MergeReason.MAPPING_UPDATE, false);
    return mapperService;
}
 

@Override
protected NamedXContentRegistry xContentRegistry() {
    if (isInternalCluster() && cluster().size() > 0) {
        // If it's internal cluster - using existing registry in case plugin registered custom data
        return internalCluster().getInstance(NamedXContentRegistry.class);
    } else {
        // If it's external cluster - fall back to the standard set
        return new NamedXContentRegistry(ClusterModule.getNamedXWriteables());
    }
}
 

public static MockTcpTransport newMockTransport(Settings settings, Version version, ThreadPool threadPool) {
    // some tests use MockTransportService to do network based testing. Yet, we run tests in multiple JVMs that means
    // concurrent tests could claim port that another JVM just released and if that test tries to simulate a disconnect it might
    // be smart enough to re-connect depending on what is tested. To reduce the risk, since this is very hard to debug we use
    // a different default port range per JVM unless the incoming settings override it
    int basePort = 10300 + (JVM_ORDINAL * 100); // use a non-default port otherwise some cluster in this JVM might reuse a port
    settings = Settings.builder().put(TransportSettings.PORT.getKey(), basePort + "-" + (basePort + 100)).put(settings).build();
    NamedWriteableRegistry namedWriteableRegistry = new NamedWriteableRegistry(ClusterModule.getNamedWriteables());
    return new MockTcpTransport(settings, threadPool, BigArrays.NON_RECYCLING_INSTANCE,
        new NoneCircuitBreakerService(), namedWriteableRegistry, new NetworkService(Collections.emptyList()), version);
}
 

private NamedXContentRegistry xContentRegistry() {
  return new NamedXContentRegistry(ClusterModule.getNamedXWriteables());
}
 

public ElasticsearchNodeCommand(String description) {
    super(description);
    nodeOrdinalOption = parser.accepts("ordinal", "Optional node ordinal, 0 if not specified")
            .withRequiredArg().ofType(Integer.class);
    namedXContentRegistry = new NamedXContentRegistry(ClusterModule.getNamedXWriteables());
}
 

private Manifest loadManifest() throws IOException {
    return Manifest.FORMAT.loadLatestState(logger, new NamedXContentRegistry(ClusterModule.getNamedXWriteables()), nodePaths);
}
 

/**
 * The {@link NamedXContentRegistry} to use for this test. Subclasses should override and use liberally.
 */
protected NamedXContentRegistry xContentRegistry() {
    return new NamedXContentRegistry(ClusterModule.getNamedXWriteables());
}
 

/**
 * The {@link NamedWriteableRegistry} to use for this test. Subclasses should override and use liberally.
 */
protected NamedWriteableRegistry writableRegistry() {
    return new NamedWriteableRegistry(ClusterModule.getNamedWriteables());
}
 

protected NamedWriteableRegistry writeableRegistry() {
    return new NamedWriteableRegistry(ClusterModule.getNamedWriteables());
}
 

@Override
protected NamedWriteableRegistry getNamedWriteableRegistry() {
    return new NamedWriteableRegistry(ClusterModule.getNamedWriteables());
}
 

private NamedXContentRegistry getNamedXContentRegistry() {
    List<NamedXContentRegistry.Entry> registry = new ArrayList<>();
    registry.addAll(new SQLPlugin(Settings.EMPTY).getNamedXContent());
    registry.addAll(ClusterModule.getNamedXWriteables());
    return new NamedXContentRegistry(registry);
}
 

public void onModule(ClusterModule clusterModule) {
    // add our dynamic cluster settings
    clusterModule.registerClusterDynamicSetting(Constants.CUSTOM_ANALYSIS_SETTINGS_PREFIX + "*", Validator.EMPTY);

    clusterModule.registerClusterDynamicSetting(CrateCircuitBreakerService.QUERY_CIRCUIT_BREAKER_LIMIT_SETTING, Validator.MEMORY_SIZE);
    clusterModule.registerClusterDynamicSetting(CrateCircuitBreakerService.QUERY_CIRCUIT_BREAKER_OVERHEAD_SETTING, Validator.NON_NEGATIVE_DOUBLE);

    clusterModule.registerClusterDynamicSetting("crate.internal.decommission.*", Validator.EMPTY);

    registerSettings(clusterModule, CrateSettings.CRATE_SETTINGS);

    clusterModule.registerAllocationDecider(DecommissionAllocationDecider.class);
}