DistributedGroupExample.java

/*
 * Copyright 2015 the original author or authors.
 *
 * 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.examples.group;

import io.atomix.AtomixReplica;
import io.atomix.catalyst.transport.Address;
import io.atomix.catalyst.transport.netty.NettyTransport;
import io.atomix.copycat.server.storage.Storage;
import io.atomix.group.DistributedGroup;
import io.atomix.group.messaging.MessageConsumer;

import java.net.InetAddress;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

/**
 * Distributed group example.
 *
 * @author <a href="http://github.com/kuujo>Jordan Halterman</a>
 */
public class DistributedGroupExample {

  /**
   * Starts the server.
   */
  public static void main(String[] args) throws Exception {
    if (args.length < 2)
      throw new IllegalArgumentException("must supply a local port and at least one remote host:port tuple");

    int port = Integer.valueOf(args[0]);

    Address address = new Address(InetAddress.getLocalHost().getHostName(), port);

    List<Address> cluster = new ArrayList<>();
    for (int i = 1; i < args.length; i++) {
      String[] parts = args[i].split(":");
      cluster.add(new Address(parts[0], Integer.valueOf(parts[1])));
    }

    AtomixReplica atomix = AtomixReplica.builder(address)
      .withTransport(new NettyTransport())
      .withStorage(Storage.builder()
        .withDirectory(System.getProperty("user.dir") + "/logs/" + UUID.randomUUID().toString())
        .build())
      .build();

    atomix.bootstrap(cluster).join();

    System.out.println("Creating membership group");
    DistributedGroup group = atomix.getGroup("group").get();

    System.out.println("Joining membership group");
    group.join().thenAccept(member -> {
      System.out.println("Joined group with member ID: " + member.id());
      MessageConsumer<String> consumer = member.messaging().consumer("tasks");
      consumer.onMessage(task -> {
        System.out.println("Received message");
        try {
          Thread.sleep(100);
        } catch (InterruptedException e) {
          task.ack();
        }
      });
    });

    group.onJoin(member -> {
      System.out.println(member.id() + " joined the group!");

      member.messaging().producer("tasks").send("hello").thenRun(() -> {
        System.out.println("Task complete!");
      });
    });

    for (;;) {
      Thread.sleep(1000);
    }
  }

}