Return the pooled buffer when encoding fails

codec-java/src/main/java/com/kakao/actionbase/v2/core/code/AbstractEdgeEncoder.java

@@ -404,11 +404,15 @@ public String useAsHexString(Consumer<EdgeBuffer> block) {
 
   public byte[] useAsByteArray(Consumer<EdgeBuffer> block) {
     EdgeBuffer edgeBuffer = borrow();
-    edgeBuffer.reset();
-    block.accept(edgeBuffer);
-    byte[] byteArray = edgeBuffer.buffer.toByteArray();
-    release(edgeBuffer);
-    return byteArray;
+    try {
+      edgeBuffer.reset();
+      block.accept(edgeBuffer);
+      return edgeBuffer.buffer.toByteArray();
+    } finally {
+      if (edgeBuffer != null) {
+        release(edgeBuffer);
+      }
+    }
   }
 
   // --- private

codec-java/src/test/java/com/kakao/actionbase/v2/core/code/AbstractEdgeEncoderTest.java

@@ -0,0 +1,40 @@
+package com.kakao.actionbase.v2.core.code;
+
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertSame;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+
+import org.junit.jupiter.api.Test;
+
+class AbstractEdgeEncoderTest {
+
+  @Test
+  void shouldReturnBorrowedBufferAfterEncodingFails() {
+    EdgeEncoderFactory factory = new EdgeEncoderFactory(1);
+    BytesKeyValueEdgeEncoder encoder = factory.getBytesKeyValueEncoder();
+    EdgeBuffer borrowed = factory.pool.peek();
+
+    IllegalArgumentException failure =
+        assertThrows(
+            IllegalArgumentException.class,
+            () ->
+                encoder.useAsByteArray(
+                    buffer -> {
+                      buffer.encodeInt8((byte) 7);
+                      buffer.encodeAny(new Object());
+                    }));
+
+    assertEquals("Unexpected data of type : java.lang.Object", failure.getMessage());
+    assertEquals(1, encoder.getPoolSize());
+    assertSame(borrowed, factory.pool.peek());
+
+    byte[] encoded = encoder.useAsByteArray(buffer -> buffer.encodeInt8((byte) 1));
+    byte[] expected =
+        new EdgeEncoderFactory()
+            .getBytesKeyValueEncoder()
+            .useAsByteArray(buffer -> buffer.encodeInt8((byte) 1));
+    assertArrayEquals(expected, encoded);
+    assertSame(borrowed, factory.pool.peek());
+  }
+}