--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/org/dancres/blitz/remote/perf/LeaseImpl.java	Sat Mar 21 11:00:06 2009 +0000
@@ -0,0 +1,93 @@
+package org.dancres.blitz.remote.perf;
+
+import java.rmi.RemoteException;
+
+import com.sun.jini.lease.AbstractLease;
+
+import net.jini.core.lease.UnknownLeaseException;
+import net.jini.core.lease.LeaseDeniedException;
+import net.jini.core.lease.LeaseMap;
+import net.jini.core.lease.Lease;
+
+import net.jini.id.Uuid;
+import net.jini.id.ReferentUuid;
+
+import org.dancres.blitz.lease.SpaceUID;
+
+/**
+   Generic lease implementation used to wrap all Blitz's internal lease
+   implementations.
+ */
+class LeaseImpl extends AbstractLease implements ReferentUuid {
+    Landlord theStub;
+    Uuid theUuid;
+
+    private SpaceUID theUID;
+
+    LeaseImpl(Landlord aLandlord, Uuid aUuid, SpaceUID aUID,
+              long anExpiration) {
+
+        super(anExpiration);
+        theStub = aLandlord;
+        theUID = aUID;
+        theUuid = aUuid;
+    }
+
+    public Uuid getReferentUuid() {
+        return theUuid;
+    }
+
+    protected long doRenew(long duration)
+        throws UnknownLeaseException, LeaseDeniedException, RemoteException {
+
+        return theStub.renew(theUID, duration);
+    }
+
+    public void cancel()
+        throws UnknownLeaseException, RemoteException {
+
+        theStub.cancel(theUID);
+    }
+
+    void setExpiration(long anExpiry) {
+        expiration = anExpiry;
+    }
+
+    SpaceUID getUID() {
+        return theUID;
+    }
+
+    /**
+       Each entry in a LeaseMap needs a duration to renew for so, when we
+       create the leasemap, we insert the current lease as having the specified
+       renewal duration.
+     */
+    public LeaseMap createLeaseMap(long aDuration) {
+        return new LeaseMapImpl(this, aDuration, theStub);
+    }
+
+    public boolean canBatch(Lease aLease) {
+
+        if (aLease instanceof LeaseImpl) {
+            LeaseImpl myOther = (LeaseImpl) aLease;
+
+            return myOther.theStub.equals(theStub);
+        }
+
+        return false;
+    }
+
+    public int hashCode() {
+        return theUID.hashCode();
+    }
+
+    public boolean equals(Object anObject) {
+        if (anObject instanceof LeaseImpl) {
+            if (theUuid.equals(((LeaseImpl) anObject).theUuid)) {
+                return (theUID.equals(((LeaseImpl) anObject).theUID));
+            }
+        }
+
+        return false;
+    }
+}