adapted PutBuilder for confirming and rejecting prepared puts

tomp2p · Jun 4, 2014 · 3db803c · 3db803c
1 parent 044d462
commit 3db803c
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/core/src/main/java/net/tomp2p/p2p/builder/PutBuilder.java b/core/src/main/java/net/tomp2p/p2p/builder/PutBuilder.java
@@ -41,6 +41,9 @@ public class PutBuilder extends DHTBuilder<PutBuilder> {
 
     private boolean putMeta = false;
 
+    private boolean putConfim = false;
+    private boolean putReject = false;
+
     private PublicKey changePublicKey = null;
 
     public PutBuilder(Peer peer, Number160 locationKey) {
@@ -180,7 +183,25 @@ public PutBuilder changePublicKey(PublicKey changePublicKey) {
     	setSign();
     	return this;
     }
-
+
+	public boolean isPutConfirm() {
+		return putConfim;
+	}
+
+	public PutBuilder putConfirm() {
+		this.putConfim = true;
+		return this;
+	}
+
+	public boolean isPutReject() {
+		return putReject;
+	}
+
+	public PutBuilder putReject() {
+		this.putReject = true;
+		return this;
+	}
+
     public PublicKey changePublicKey() {
     	return changePublicKey;
     }
@@ -200,6 +221,18 @@ public FuturePut start() {
             throw new IllegalArgumentException(
                     "You must either set data via setDataMap() or setData(). Cannot add nothing.");
         }
+        if (!putConfim && dataMap == null && dataMapConvert == null) {
+            throw new IllegalArgumentException(
+                    "You must either set data via setDataMap() or setData(). Cannot add nothing.");
+        }
+        if (!putReject && dataMap == null && dataMapConvert == null) {
+            throw new IllegalArgumentException(
+                    "You must either set data via setDataMap() or setData(). Cannot add nothing.");
+        }
+        if (putConfim && putReject) {
+            throw new IllegalArgumentException(
+                    "You can't confirm and reject a put.");
+        }
         if (locationKey == null) {
             throw new IllegalArgumentException("You must provide a location key.");
         }