io.atomix.cluster.impl.DefaultNodeDiscoveryService Maven / Gradle / Ivy
/*
* 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);
}
}