X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FFreenetInterface.java;h=cfc130971c4157163084ed8844fe50b7ca3f9387;hb=f95e0b9700d6d50dedda280da2378cbc81e55efb;hp=7cb8a9aaec137075f3bd89cfefb7614c19bebbc3;hpb=d525d4ad65878cfd895945512650d978f6a5c034;p=Sone.git

diff --git a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
index 7cb8a9a..cfc1309 100644
--- a/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
+++ b/src/main/java/net/pterodactylus/sone/core/FreenetInterface.java
@@ -161,6 +161,7 @@ public class FreenetInterface {
 		FreenetURI targetUri = key.getInsertURI().setDocName(filenameHint);
 		InsertContext insertContext = client.getInsertContext(true);
 		RandomAccessBucket bucket = new ArrayBucket(temporaryImage.getImageData());
+		insertToken.setBucket(bucket);
 		ClientMetadata metadata = new ClientMetadata(temporaryImage.getMimeType());
 		InsertBlock insertBlock = new InsertBlock(bucket, metadata, targetUri);
 		try {
@@ -404,6 +405,7 @@ public class FreenetInterface {
 
 		/** The client putter. */
 		private ClientPutter clientPutter;
+		private Bucket bucket;
 
 		/** The final URI. */
 		private volatile FreenetURI resultingUri;
@@ -435,6 +437,10 @@ public class FreenetInterface {
 			eventBus.post(new ImageInsertStartedEvent(image));
 		}
 
+		public void setBucket(Bucket bucket) {
+			this.bucket = bucket;
+		}
+
 		//
 		// ACTIONS
 		//
@@ -473,6 +479,7 @@ public class FreenetInterface {
 			} else {
 				eventBus.post(new ImageInsertFailedEvent(image, insertException));
 			}
+			bucket.free();
 		}
 
 		/**
@@ -506,6 +513,7 @@ public class FreenetInterface {
 		@SuppressWarnings("synthetic-access")
 		public void onSuccess(BaseClientPutter clientPutter) {
 			eventBus.post(new ImageInsertFinishedEvent(image, resultingUri));
+			bucket.free();
 		}
 
 	}