io.atomix.cluster.impl.DefaultNodeDiscoveryService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zeebe-atomix-cluster Show documentation
The newest version!
/*
 * Copyright 2018-present Open Networking Foundation
 * Copyright © 2020 camunda services GmbH ([email protected])
 *
 * Licensed 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 io.atomix.cluster.impl;

import io.atomix.cluster.BootstrapService;
import io.atomix.cluster.Node;
import io.atomix.cluster.discovery.ManagedNodeDiscoveryService;
import io.atomix.cluster.discovery.NodeDiscoveryEvent;
import io.atomix.cluster.discovery.NodeDiscoveryEventListener;
import io.atomix.cluster.discovery.NodeDiscoveryProvider;
import io.atomix.cluster.discovery.NodeDiscoveryService;
import io.atomix.utils.event.AbstractListenerManager;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicBoolean;

/** Default node discovery service. */
public class DefaultNodeDiscoveryService
    extends AbstractListenerManager
    implements ManagedNodeDiscoveryService {

  private final BootstrapService bootstrapService;
  private final Node localNode;
  private final NodeDiscoveryProvider provider;
  private final AtomicBoolean started = new AtomicBoolean();
  private final NodeDiscoveryEventListener discoveryEventListener = this::post;

  public DefaultNodeDiscoveryService(
      final BootstrapService bootstrapService,
      final Node localNode,
      final NodeDiscoveryProvider provider) {
    this.bootstrapService = bootstrapService;
    this.localNode = localNode;
    this.provider = provider;
  }

  @Override
  public Set getNodes() {
    return provider.getNodes();
  }

  @Override
  public CompletableFuture start() {
    if (started.compareAndSet(false, true)) {
      provider.addListener(discoveryEventListener);
      final Node node =
          Node.builder().withId(localNode.id().id()).withAddress(localNode.address()).build();
      return provider.join(bootstrapService, node).thenApply(v -> this);
    }
    return CompletableFuture.completedFuture(this);
  }

  @Override
  public boolean isRunning() {
    return started.get();
  }

  @Override
  public CompletableFuture stop() {
    if (started.compareAndSet(true, false)) {
      return provider
          .leave(localNode)
          .thenRun(
              () -> {
                provider.removeListener(discoveryEventListener);
              });
    }
    return CompletableFuture.completedFuture(null);
  }
}