1 files changed, 136 insertions, 0 deletions
diff --git a/simple/simple-transport/src/main/java/org/simpleframework/transport/reactor/PartitionDistributor.java b/simple/simple-transport/src/main/java/org/simpleframework/transport/reactor/PartitionDistributor.java
new file mode 100644
index 0000000..b0d24ac
--- /dev/null
+++ b/simple/simple-transport/src/main/java/org/simpleframework/transport/reactor/PartitionDistributor.java
@@ -0,0 +1,136 @@
+/*
+ * PartitionDistributor.java February 2007
+ *
+ * Copyright (C) 2007, Niall Gallagher <niallg@users.sf.net>
+ *
+ * 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 org.simpleframework.transport.reactor;
+
+import java.io.IOException;
+import java.nio.channels.SelectableChannel;
+import java.util.concurrent.Executor;
+
+/**
+ * The <code>PartitionDistributor</code> object is a distributor that
+ * partitions the selection process in to several threads. Each of
+ * the threads has a single selector, and operations are distributed
+ * amongst the threads using the hash code of the socket. Partitions
+ * ensure that several selector threads can share a higher load and
+ * respond to a more I/O events.
+ * 
+ * @author Niall Gallagher
+ */
+class PartitionDistributor implements OperationDistributor {
+
+   /**
+    * This contains the distributors that represent a partition. 
+    */
+   private final OperationDistributor[] list;
+   
+   /**
+    * Constructor for the <code>PartitionDistributor</code> object. 
+    * This will create a distributor that partitions the operations
+    * amongst a pool of selectors using the channels hash code.
+    * 
+    * @param executor this is the executor used to run operations
+    * @param count this is the number of partitions to be used
+    */
+   public PartitionDistributor(Executor executor, int count) throws IOException {
+      this(executor, count, 120000);      
+   }
+   
+   /**
+    * Constructor for the <code>PartitionDistributor</code> object. 
+    * This will create a distributor that partitions the operations
+    * amongst a pool of selectors using the channels hash code.
+    * 
+    * @param executor this is the executor used to run operations
+    * @param count this is the number of partitions to be used
+    * @param expiry this is the expiry duration that is to be used
+    */   
+   public PartitionDistributor(Executor executor, int count, long expiry) throws IOException {      
+      this.list = new OperationDistributor[count];
+      this.start(executor, expiry);
+   }
+   
+   /**
+    * This is used to create the partitions that represent a thread
+    * used for selection. Operations will index to a particular one
+    * using the hash code of the operations channel. If there is only
+    * one partition all operations will index to the partition.
+    * 
+    * @param executor the executor used to run the operations
+    * @param expiry this is the expiry duration that is to be used
+    */
+   private void start(Executor executor, long expiry) throws IOException {
+      for(int i = 0; i < list.length; i++) {
+         list[i] = new ActionDistributor(executor, true, expiry);
+      }
+   }
+
+   /**
+    * This is used to process the <code>Operation</code> object. This
+    * will wake up the selector if it is currently blocked selecting
+    * and register the operations associated channel. Once the 
+    * selector is awake it will acquire the operation from the queue
+    * and register the associated <code>SelectableChannel</code> for
+    * selection. The operation will then be executed when the channel
+    * is ready for the interested I/O events.
+    * 
+    * @param task this is the task that is scheduled for distribution   
+    * @param require this is the bit-mask value for interested events
+    */    
+   public void process(Operation task, int require) throws IOException {
+      int length = list.length;
+      
+      if(length == 1) {
+         list[0].process(task, require);
+      } else {
+         process(task, require, length);
+      }
+   }
+   
+   /**
+    * This is used to process the <code>Operation</code> object. This
+    * will wake up the selector if it is currently blocked selecting
+    * and register the operations associated channel. Once the 
+    * selector is awake it will acquire the operation from the queue
+    * and register the associated <code>SelectableChannel</code> for
+    * selection. The operation will then be executed when the channel
+    * is ready for the interested I/O events.
+    * 
+    * @param task this is the task that is scheduled for distribution   
+    * @param require this is the bit-mask value for interested events
+    * @param length this is the number of distributors to hash with
+    */    
+   private void process(Operation task, int require, int length) throws IOException {
+      SelectableChannel channel = task.getChannel();
+      int hash = channel.hashCode();
+      
+      list[hash % length].process(task, require);
+   }   
+
+   /**
+    * This is used to close the distributor such that it cancels all
+    * of the registered channels and closes down the selector. This
+    * is used when the distributor is no longer required, after the
+    * close further attempts to process operations will fail.
+    */    
+   public void close() throws IOException {
+      for(OperationDistributor entry : list) {
+         entry.close();
+      }      
+   }
+}