solve the problem of direct memory leak

hprose · Jul 1, 2020 · 05968ee · 05968ee
1 parent 0f8d00f
commit 05968ee
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/hprose/io/ByteBufferStream.java b/src/main/java/hprose/io/ByteBufferStream.java
@@ -20,6 +20,8 @@
 package hprose.io;
 
 import hprose.common.HproseException;
+import sun.nio.ch.DirectBuffer;
+
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
@@ -75,9 +77,11 @@ private void free(ByteBuffer buffer) {
                         if (pos < POOLSIZE - 1) {
                             pool[index][++pos] = buffer;
                             position[index] = pos;
+                            return;
                         }
                     }
                 }
+                ((DirectBuffer) buffer).cleaner().clean();
             }
         }
     }