Remove thread local message frame to fix the connection close issue

ballerina-platform · Dec 5, 2024 · 630d8a7 · 630d8a7
1 parent 057624b
commit 630d8a7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ballerina-tests/tests/resources/.keep b/ballerina-tests/tests/resources/.keep
diff --git a/native/src/main/java/io/ballerina/stdlib/grpc/OutboundMessage.java b/native/src/main/java/io/ballerina/stdlib/grpc/OutboundMessage.java
@@ -49,11 +49,11 @@ public class OutboundMessage {
     private final HttpCarbonMessage responseMessage;
     private int statusCode = NULL_STATUS_CODE;
     private boolean outboundClosed;
-    private final ThreadLocal<MessageFramer> framer;
+    private final MessageFramer framer;
 
     public OutboundMessage(HttpCarbonMessage responseMessage) {
         this.responseMessage = responseMessage;
-        this.framer = ThreadLocal.withInitial(() -> new MessageFramer(responseMessage));
+        this.framer = new MessageFramer(responseMessage);
     }
 
     OutboundMessage(InboundMessage inboundMessage) {
@@ -68,7 +68,7 @@ public OutboundMessage(HttpCarbonMessage responseMessage) {
         responseMessage.setHeader(CONTENT_TYPE_KEY, GrpcConstants.CONTENT_TYPE_GRPC);
         responseMessage.setHeader(TE_KEY, GrpcConstants.TE_TRAILERS);
         this.responseMessage = responseMessage;
-        this.framer = ThreadLocal.withInitial(() -> new MessageFramer(responseMessage));
+        this.framer = new MessageFramer(responseMessage);
     }
 
     /**
@@ -237,7 +237,7 @@ final void halfClose() {
     }
 
     final MessageFramer framer() {
-        return framer.get();
+        return framer;
     }
 
     public boolean isReady() {