#919 - io.ebean package initial

src/main/java/io/ebeaninternal/server/cluster/BinaryMessage.java

@@ -0,0 +1,58 @@
+package io.ebeaninternal.server.cluster;
+
+import java.io.ByteArrayOutputStream;
+import java.io.DataOutputStream;
+
+/**
+ * Represents a relatively small independent message.
+ * <p>
+ * In general terms we break up a potentially large object like
+ * RemoteTransactionEvent into many smaller BinaryMessages. This is so that if
+ * they don't all fit on a single Packet we can easily break them up and put
+ * them on multiple packets.
+ * </p>
+ * <p>
+ * Also note that for the Multicast approach a Packet will generally contain
+ * many messages each directed to different members of the cluster. So it would
+ * be common for many Ack, Resend and Control messages to all be contained in a
+ * single packet.
+ * </p>
+ */
+public class BinaryMessage {
+
+  public static final int TYPE_MSGCONTROL = 0;
+  public static final int TYPE_BEANIUD = 1;
+  public static final int TYPE_TABLEIUD = 2;
+
+  public static final int TYPE_MSGACK = 8;
+  public static final int TYPE_MSGRESEND = 9;
+
+  private final ByteArrayOutputStream buffer;
+  private final DataOutputStream os;
+  private byte[] bytes;
+
+  /**
+   * Create with an estimated buffer size.
+   */
+  public BinaryMessage(int bufSize) {
+    this.buffer = new ByteArrayOutputStream(bufSize);
+    this.os = new DataOutputStream(buffer);
+  }
+
+  /**
+   * Return the DataOutputStream to write content to.
+   */
+  public DataOutputStream getOs() {
+    return os;
+  }
+
+  /**
+   * Return all the content as a byte array.
+   */
+  public byte[] getByteArray() {
+    if (bytes == null) {
+      bytes = buffer.toByteArray();
+    }
+    return bytes;
+  }
+}

src/main/java/io/ebeaninternal/server/cluster/BinaryMessageList.java

@@ -0,0 +1,23 @@
+package io.ebeaninternal.server.cluster;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Holds a List of BinaryMessage's.
+ *
+ * @author rbygrave
+ */
+public class BinaryMessageList {
+
+  final ArrayList<BinaryMessage> list = new ArrayList<>();
+
+  public void add(BinaryMessage msg) {
+    list.add(msg);
+  }
+
+  public List<BinaryMessage> getList() {
+    return list;
+  }
+
+}

src/main/java/io/ebeaninternal/server/cluster/ClusterBroadcast.java

@@ -0,0 +1,28 @@
+package io.ebeaninternal.server.cluster;
+
+import io.ebeaninternal.server.transaction.RemoteTransactionEvent;
+
+
+/**
+ * Sends messages to the cluster members.
+ */
+public interface ClusterBroadcast {
+
+  /**
+   * Inform the other cluster members that this instance has come online and
+   * start any listeners etc.
+   */
+  void startup();
+
+  /**
+   * Inform the other cluster members that this instance is leaving and
+   * shutdown any listeners.
+   */
+  void shutdown();
+
+  /**
+   * Send a transaction event to all the members of the cluster.
+   */
+  void broadcast(RemoteTransactionEvent remoteTransEvent);
+
+}

src/main/java/io/ebeaninternal/server/cluster/ClusterBroadcastFactory.java

@@ -0,0 +1,14 @@
+package io.ebeaninternal.server.cluster;
+
+import java.util.Properties;
+
+/**
+ * Factory to create the cluster broadcast service.
+ */
+public interface ClusterBroadcastFactory {
+
+  /**
+   * Create the cluster transport with the manager and deployment properties.
+   */
+  ClusterBroadcast create(ClusterManager manager, Properties properties);
+}

src/main/java/io/ebeaninternal/server/cluster/ClusterManager.java

@@ -0,0 +1,107 @@
+package io.ebeaninternal.server.cluster;
+
+import io.ebean.EbeanServer;
+import io.ebean.config.ContainerConfig;
+import io.ebeaninternal.server.transaction.RemoteTransactionEvent;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.Iterator;
+import java.util.ServiceLoader;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * Manages the cluster service.
+ */
+public class ClusterManager {
+
+  private static final Logger clusterLogger = LoggerFactory.getLogger("org.avaje.ebean.Cluster");
+
+  private static final Logger logger = LoggerFactory.getLogger(ClusterManager.class);
+
+  private final ConcurrentHashMap<String, EbeanServer> serverMap = new ConcurrentHashMap<>();
+
+  private final Object monitor = new Object();
+
+  private final ClusterBroadcast broadcast;
+
+  private boolean started;
+
+  public ClusterManager(ContainerConfig config) {
+    if (!config.isClusterActive()) {
+      broadcast = null;
+    } else {
+      ClusterBroadcastFactory factory = createFactory();
+      broadcast = factory.create(this, config.getProperties());
+    }
+  }
+
+  /**
+   * Return the ClusterTransportFactory via ServiceLoader.
+   */
+  private ClusterBroadcastFactory createFactory() {
+
+    ServiceLoader<ClusterBroadcastFactory> load = ServiceLoader.load(ClusterBroadcastFactory.class);
+    ClusterBroadcastFactory factory = null;
+    Iterator<ClusterBroadcastFactory> iterator = load.iterator();
+    if (iterator.hasNext()) {
+      factory = iterator.next();
+    }
+    if (factory == null) {
+      throw new IllegalStateException("No ClusterTransportFactory found in classpath. "
+        + " Probably need to add the avaje-ebeanorm-cluster dependency");
+    }
+    return factory;
+  }
+
+  public void registerServer(EbeanServer server) {
+    synchronized (monitor) {
+      serverMap.put(server.getName(), server);
+      if (!started) {
+        startup();
+      }
+    }
+  }
+
+  public EbeanServer getServer(String name) {
+    synchronized (monitor) {
+      return serverMap.get(name);
+    }
+  }
+
+  private void startup() {
+    started = true;
+    if (broadcast != null) {
+      broadcast.startup();
+    }
+  }
+
+  /**
+   * Return true if clustering is on.
+   */
+  public boolean isClustering() {
+    return broadcast != null;
+  }
+
+  /**
+   * Send the message headers and payload to every server in the cluster.
+   */
+  public void broadcast(RemoteTransactionEvent event) {
+    if (broadcast != null) {
+      if (clusterLogger.isDebugEnabled()) {
+        clusterLogger.debug("sending: {}", event);
+      }
+      broadcast.broadcast(event);
+    }
+  }
+
+  /**
+   * Shutdown the service and Deregister from the cluster.
+   */
+  public void shutdown() {
+    if (broadcast != null) {
+      logger.info("ClusterManager shutdown ");
+      broadcast.shutdown();
+    }
+  }
+}

src/main/java/io/ebeaninternal/server/cluster/package.html

@@ -0,0 +1,12 @@
+<HTML>
+<HEAD>
+  <META HTTP-EQUIV="CONTENT-TYPE" CONTENT="text/html; charset=iso-8859-1">
+  <TITLE>AvajeLib</TITLE>
+</HEAD>
+<Body BGCOLOR="#ffffff">
+Clustering service for an application.
+<P>
+  A framework for supporting clustering of servers.
+</P>
+</Body>
+</HTML>