Exemplo n.º 1
0
 @Override
 public CompletableFuture<Void> close() {
   CompletableFuture<Void> future = new CompletableFuture<>();
   lock.lock()
       .thenCompose(v -> balancer.replace(server.server().cluster()))
       .whenComplete(
           (r1, e1) -> {
             balancer.close();
             lock.unlock()
                 .whenComplete(
                     (r2, e2) -> {
                       super.close()
                           .whenComplete(
                               (r3, e3) -> {
                                 server
                                     .close()
                                     .whenComplete(
                                         (r4, e4) -> {
                                           if (e4 == null) {
                                             future.complete(null);
                                           } else {
                                             future.completeExceptionally(e4);
                                           }
                                         });
                               });
                     });
           });
   return future;
 }
Exemplo n.º 2
0
  /** Registers membership change listeners on the cluster. */
  private void registerListeners() {
    server
        .server()
        .cluster()
        .members()
        .forEach(
            m -> {
              m.onTypeChange(t -> balance());
              m.onStatusChange(s -> balance());
            });

    server.server().cluster().onLeaderElection(l -> balance());

    server
        .server()
        .cluster()
        .onJoin(
            m -> {
              m.onTypeChange(t -> balance());
              m.onStatusChange(s -> balance());
              balance();
            });

    server.server().cluster().onLeave(m -> balance());
  }
Exemplo n.º 3
0
 /** Balances the cluster. */
 private void balance() {
   if (lock != null
       && !locking
       && server.server().cluster().member().equals(server.server().cluster().leader())) {
     locking = true;
     lock.lock()
         .thenCompose(v -> balancer.balance(server.server().cluster()))
         .whenComplete((r1, e1) -> lock.unlock().whenComplete((r2, e2) -> locking = false));
   }
 }
Exemplo n.º 4
0
 @Override
 public CompletableFuture<Atomix> open() {
   return server
       .open()
       .thenRun(this::registerListeners)
       .thenCompose(v -> super.open())
       .thenCompose(v -> client.get("", DistributedLock.class))
       .thenApply(
           lock -> {
             this.lock = lock;
             return this;
           });
 }