ObjectKeyGenerator.java

package org.radargun.stages.cache.generators;

import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;

import org.radargun.config.DefinitionElement;

/**
 * @author Mircea Markus <Mircea.Markus@jboss.com>
 */
@DefinitionElement(name = "object", doc = "Generates externalizable keys wrapping long identifier of the key.")
public class ObjectKeyGenerator implements KeyGenerator {
   @Override
   public Object generateKey(long keyIndex) {
      return new ObjectKey(keyIndex);
   }

   /**
    * @author Mircea Markus <Mircea.Markus@jboss.com>
    */
   public static class ObjectKey implements Externalizable {

      private long keyIndex;

      public ObjectKey(long keyIndex) {
         this.keyIndex = keyIndex;
      }

      @Override
      public void writeExternal(ObjectOutput objectOutput) throws IOException {
         objectOutput.writeLong(keyIndex);
      }

      @Override
      public void readExternal(ObjectInput objectInput) throws IOException, ClassNotFoundException {
         this.keyIndex = objectInput.readLong();
      }

      @Override
      public String toString() {
         return String.format("ObjectKey{%016X}", keyIndex);
      }

      @Override
      public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;

         ObjectKey objectKey = (ObjectKey) o;

         if (keyIndex != objectKey.keyIndex) return false;
         return true;
      }

      @Override
      public int hashCode() {
         return (int) keyIndex;
      }

      /**
       * This is an index that uniquely identifies this key in the cluster.
       */
      public long getKeyIndexInCluster(int threadCountPerNode, int keysPerThread) {
         return keyIndex;
      }
   }
}