#919 - io.ebean package initial

src/main/java/io/ebeaninternal/server/deploy/ChainedBeanPersistController.java

@@ -0,0 +1,185 @@
+package io.ebeaninternal.server.deploy;
+
+import io.ebean.event.BeanPersistController;
+import io.ebean.event.BeanPersistRequest;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Comparator;
+import java.util.List;
+
+/**
+ * Chains multiple BeanPersistController's together.
+ *
+ * Used when multiple BeanPersistController register for the same bean type.
+ */
+public class ChainedBeanPersistController implements BeanPersistController {
+
+	private static final Sorter SORTER = new Sorter();
+
+	private final List<BeanPersistController> list;
+
+	private final BeanPersistController[] chain;
+
+	/**
+	 * Construct adding 2 BeanPersistController's.
+	 */
+	public ChainedBeanPersistController(BeanPersistController c1, BeanPersistController c2) {
+		this(addList(c1, c2));
+	}
+
+	/**
+	 * Helper method used to create a list from 2 BeanPersistController's.
+	 */
+	private static List<BeanPersistController> addList(BeanPersistController c1, BeanPersistController c2) {
+		ArrayList<BeanPersistController> addList = new ArrayList<>(2);
+		addList.add(c1);
+		addList.add(c2);
+		return addList;
+	}
+
+	/**
+	 * Construct given the list of BeanPersistController's.
+	 */
+	public ChainedBeanPersistController(List<BeanPersistController> list) {
+		this.list = list;
+		BeanPersistController[] c = list.toArray(new BeanPersistController[list.size()]);
+		Arrays.sort(c, SORTER);
+		this.chain = c;
+	}
+
+  /**
+   * Return the size of the chain.
+   */
+	protected int size() {
+		return chain.length;
+	}
+
+	/**
+	 * Register a new BeanPersistController and return the resulting chain.
+	 */
+	public ChainedBeanPersistController register(BeanPersistController c) {
+		if (list.contains(c)){
+			return this;
+		} else {
+			ArrayList<BeanPersistController> newList = new ArrayList<>();
+			newList.addAll(list);
+			newList.add(c);
+
+			return new ChainedBeanPersistController(newList);
+		}
+	}
+
+	/**
+	 * De-register a BeanPersistController and return the resulting chain.
+	 */
+	public ChainedBeanPersistController deregister(BeanPersistController c) {
+		if (!list.contains(c)){
+			return this;
+		} else {
+			ArrayList<BeanPersistController> newList = new ArrayList<>();
+			newList.addAll(list);
+			newList.remove(c);
+
+			return new ChainedBeanPersistController(newList);
+		}
+	}
+
+	/**
+	 * Always returns 0 (not used for this object).
+	 */
+  @Override
+	public int getExecutionOrder() {
+		return 0;
+	}
+
+	/**
+	 * Always returns false (not used for this object).
+	 */
+  @Override
+	public boolean isRegisterFor(Class<?> cls) {
+		return false;
+	}
+
+  @Override
+	public void postDelete(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      aChain.postDelete(request);
+    }
+	}
+
+  @Override
+	public void postInsert(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      aChain.postInsert(request);
+    }
+	}
+
+  @Override
+	public void postUpdate(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      aChain.postUpdate(request);
+    }
+	}
+
+	@Override
+	public void postSoftDelete(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      aChain.postSoftDelete(request);
+    }
+	}
+
+  @Override
+	public boolean preDelete(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      if (!aChain.preDelete(request)) {
+        return false;
+      }
+    }
+		return true;
+	}
+
+	@Override
+	public boolean preSoftDelete(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      if (!aChain.preSoftDelete(request)) {
+        return false;
+      }
+    }
+		return true;
+	}
+
+  @Override
+	public boolean preInsert(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      if (!aChain.preInsert(request)) {
+        return false;
+      }
+    }
+		return true;
+	}
+
+  @Override
+	public boolean preUpdate(BeanPersistRequest<?> request) {
+    for (BeanPersistController aChain : chain) {
+      if (!aChain.preUpdate(request)) {
+        return false;
+      }
+    }
+		return true;
+	}
+
+	/**
+	 * Helper to order the BeanPersistController's in a chain.
+	 */
+	private static class Sorter implements Comparator<BeanPersistController> {
+
+		public int compare(BeanPersistController o1, BeanPersistController o2) {
+
+			int i1 = o1.getExecutionOrder() ;
+			int i2 = o2.getExecutionOrder() ;
+			return (i1<i2 ? -1 : (i1==i2 ? 0 : 1));
+		}
+
+	}
+}