Java MockNode类代码示例

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
private static void assertTribeNodeSuccessfullyCreated(Settings extraSettings) throws Exception {
    //The tribe clients do need it to make sure they can find their corresponding tribes using the proper transport
    Settings settings = Settings.builder().put(NetworkModule.HTTP_ENABLED.getKey(), false).put("node.name", "tribe_node")
            .put("transport.type", MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME).put("discovery.type", "local")
            .put("tribe.t1.transport.type", MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME)
            .put("tribe.t2.transport.type",MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME)
            .put(Environment.PATH_HOME_SETTING.getKey(), createTempDir())
            .put(extraSettings).build();

    try (Node node = new MockNode(settings, Arrays.asList(MockTcpTransportPlugin.class, TestZenDiscovery.TestPlugin.class)).start()) {
        try (Client client = node.client()) {
            assertBusy(() -> {
                ClusterState state = client.admin().cluster().prepareState().clear().setNodes(true).get().getState();
                assertThat(state.getClusterName().value(), equalTo("tribe_node_cluster"));
                assertThat(state.getNodes().getSize(), equalTo(5));
                for (DiscoveryNode discoveryNode : state.getNodes()) {
                    assertThat(discoveryNode.getName(), either(equalTo("tribe1_node")).or(equalTo("tribe2_node"))
                            .or(equalTo("tribe_node")).or(equalTo("tribe_node/t1")).or(equalTo("tribe_node/t2")));
                }
            });
        }
    }
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
@BeforeClass
public static void createTribes() throws NodeValidationException {
    Settings baseSettings = Settings.builder()
        .put(NetworkModule.HTTP_ENABLED.getKey(), false)
        .put("transport.type", MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME)
        .put(Environment.PATH_HOME_SETTING.getKey(), createTempDir())
        .put(NodeEnvironment.MAX_LOCAL_STORAGE_NODES_SETTING.getKey(), 2)
        .build();

    final List<Class<? extends Plugin>> mockPlugins = Arrays.asList(MockTcpTransportPlugin.class, TestZenDiscovery.TestPlugin.class);
    tribe1 = new MockNode(
        Settings.builder()
            .put(baseSettings)
            .put("cluster.name", "tribe1")
            .put("node.name", "tribe1_node")
                .put(NodeEnvironment.NODE_ID_SEED_SETTING.getKey(), random().nextLong())
            .build(), mockPlugins).start();
    tribe2 = new MockNode(
        Settings.builder()
            .put(baseSettings)
            .put("cluster.name", "tribe2")
            .put("node.name", "tribe2_node")
                .put(NodeEnvironment.NODE_ID_SEED_SETTING.getKey(), random().nextLong())
            .build(), mockPlugins).start();
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
FilterJoinBenchmark() {
  Settings settings = Settings.builder()
    .put(FilterJoinCache.SIREN_FILTERJOIN_CACHE_ENABLED, false)
    .put("index.engine.robin.refreshInterval", "-1")
    .put("path.home", "./target/elasticsearch-benchmark/home/")
    .put("node.local", true)
    .put(SETTING_NUMBER_OF_SHARDS, NUM_SHARDS)
    .put(SETTING_NUMBER_OF_REPLICAS, NUM_REPLICAS)
    .put(IndexCacheModule.QUERY_CACHE_EVERYTHING, true)
    .build();

  this.nodes = new MockNode[2];
  this.nodes[0] = new MockNode(Settings.builder().put(settings).put("name", "node1").build(),
          Version.CURRENT, Collections.<Class<? extends Plugin>>singletonList(SirenJoinPlugin.class)).start();
  this.nodes[1] = new MockNode(Settings.builder().put(settings).put("name", "node2").build(),
          Version.CURRENT, Collections.<Class<? extends Plugin>>singletonList(SirenJoinPlugin.class)).start();
  this.client = nodes[0].client();
  this.random = new Random(System.currentTimeMillis());
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
TermsByQueryBenchmark() {
  Settings settings = Settings.builder()
    .put(FilterJoinCache.SIREN_FILTERJOIN_CACHE_ENABLED, false)
    .put("index.engine.robin.refreshInterval", "-1")
    .put("path.home", "./target/elasticsearch-benchmark/home/")
    .put("node.local", true)
    .put(SETTING_NUMBER_OF_SHARDS, NUM_SHARDS)
    .put(SETTING_NUMBER_OF_REPLICAS, NUM_REPLICAS)
    .build();

  this.nodes = new MockNode[2];
  this.nodes[0] = new MockNode(Settings.builder().put(settings).put("name", "node1").build(),
          Version.CURRENT, Collections.<Class<? extends Plugin>>singletonList(SirenJoinPlugin.class)).start();
  this.nodes[1] = new MockNode(Settings.builder().put(settings).put("name", "node2").build(),
          Version.CURRENT, Collections.<Class<? extends Plugin>>singletonList(SirenJoinPlugin.class)).start();
  this.client = nodes[0].client();
  this.random = new Random(System.currentTimeMillis());
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public static void main(String[] args) throws Throwable {
    Settings.Builder settings = Settings.builder();
    settings.put("http.cors.enabled", "true");
    settings.put("http.cors.allow-origin", "*");
    settings.put("cluster.name", AzureRepositoryF.class.getSimpleName());

    // Example for azure repo settings
    // settings.put("cloud.azure.storage.my_account1.account", "account_name");
    // settings.put("cloud.azure.storage.my_account1.key", "account_key");
    // settings.put("cloud.azure.storage.my_account1.default", true);
    // settings.put("cloud.azure.storage.my_account2.account", "account_name");
    // settings.put("cloud.azure.storage.my_account2.key", "account_key_secondary");

    final CountDownLatch latch = new CountDownLatch(1);
    final Node node = new MockNode(settings.build(), Collections.singletonList(AzureRepositoryPlugin.class));
    Runtime.getRuntime().addShutdownHook(new Thread() {
        @Override
        public void run() {
            try {
                IOUtils.close(node);
            } catch (IOException e) {
                throw new ElasticsearchException(e);
            } finally {
                latch.countDown();
            }
        }
    });
    node.start();
    latch.await();
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public static Node createNode() {
    Settings nodeSettings = Settings.settingsBuilder()
            .put("path.home", System.getProperty("path.home"))
            .put("client.type", "node")
            .put("index.number_of_shards", 5)
            .put("index.number_of_replica", 0)
            .build();
    // ES 2.1 renders NodeBuilder as useless
    Node node = new MockNode(nodeSettings, AggregationPlugin.class);
    node.start();
    return node;
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public static Node createNode() {
    Settings nodeSettings = Settings.settingsBuilder()
            .put("path.home", System.getProperty("path.home"))
            .put("client.type", "node")
            .put("index.number_of_shards", 1)
            .put("index.number_of_replica", 0)
            .build();
    // ES 2.1 renders NodeBuilder as useless
    Node node = new MockNode(nodeSettings, SyslogPlugin.class);
    node.start();
    return node;
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public Node buildNodeWithoutPlugins() throws IOException {
    Settings nodeSettings = Settings.builder()
            .put(getNodeSettings())
            .build();
    logger.info("settings={}", nodeSettings.getAsMap());
    return new MockNode(nodeSettings, Collections.emptyList());
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public Node buildNode() throws IOException {
    Settings nodeSettings = Settings.builder()
            .put(getNodeSettings())
            .build();
    logger.info("settings={}", nodeSettings.getAsMap());
    return new MockNode(nodeSettings, Collections.singletonList(ReferencePlugin.class));
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public Node buildNodeWithoutBundlePlugin() throws IOException {
    Settings nodeSettings = Settings.builder()
            .put(getNodeSettings())
            .build();
    logger.info("settings={}", nodeSettings.getAsMap());
    return new MockNode(nodeSettings, Collections.singletonList(Netty4Plugin.class));
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public Node buildNode() throws IOException {
    Settings nodeSettings = Settings.builder()
            .put(getNodeSettings())
            .build();
    logger.info("settings={}", nodeSettings.getAsMap());
    return new MockNode(nodeSettings, Arrays.asList(Netty4Plugin.class, BundlePlugin.class));
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
private Node buildNode(String id) throws IOException {
    Settings nodeSettings = settingsBuilder()
            .put(getNodeSettings())
            .put("name", id)
            .build();
    logger.info("settings={}", nodeSettings.getAsMap());
    // ES 2.1 renders NodeBuilder as useless
    Node node = new MockNode(nodeSettings, HelperPlugin.class);
    AbstractClient client = (AbstractClient)node.client();
    nodes.put(id, node);
    clients.put(id, client);
    logger.info("clients={}", clients);
    return node;
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
public static void main(String[] args) throws Exception {
    Settings settings = Settings.builder()
        .put("cluster.name", "test-cluster")
        .put("path.home", "target/home")
        .put("path.data", "target/data")
        .put("path.plugins", "src")            
        .build();
    Collection<Class<? extends Plugin>> plugins = 
        Collections.<Class<? extends Plugin>> singletonList(ClusteringPlugin.class);
    try (Node node = new MockNode(settings, plugins)) {
      node.start();
      new CountDownLatch(1).await();
    }
}
 

import org.elasticsearch.node.MockNode; //导入依赖的package包/类
NodeAndClient(String name, MockNode node, int nodeAndClientId) {
    this.node = node;
    this.name = name;
    this.nodeAndClientId = nodeAndClientId;
    markNodeDataDirsAsNotEligableForWipe(node);
}