Merge branch 'kserve_main' into 20230626_sync_main

Signed-off-by: Vaibhav Jain <[email protected]>

Author: vaibhavjainwiz

.github/workflows/build.yml

@@ -3,7 +3,6 @@ name: Build
 on:
   pull_request:
     branches:
-      - main
       - "release-[0-9].[0-9]+"
     paths-ignore:
       - "**.md"

src/main/java/com/ibm/watson/modelmesh/ModelMeshApi.java

@@ -768,14 +768,17 @@ public void onHalfClose() {
                         evictMethodDescriptor(methodName);
                     }
                 } finally {
+                    final boolean releaseResponse = status != OK;
                     if (payloadProcessor != null) {
                         ByteBuf data = null;
                         Metadata metadata = null;
                         if (response != null) {
                             data = response.data.readerIndex(respReaderIndex);
                             metadata = response.metadata;
                         }
-                        processPayload(data, requestId, modelId, methodName, metadata, status, false);
+                        processPayload(data, requestId, modelId, methodName, metadata, status, releaseResponse);
+                    } else if (releaseResponse && response != null) {
+                        response.release();
                     }
                     ReleaseAfterResponse.releaseAll();
                     clearThreadLocals();
@@ -805,7 +808,7 @@ private void processPayload(ByteBuf data, String payloadId, String modelId, Stri
                 try {
                     assert payloadProcessor != null;
                     if (!takeOwnership) {
-                        data.retain();
+                        ReferenceCountUtil.retain(data);
                     }
                     payload = new Payload(payloadId, modelId, methodName, metadata, data, status);
                     if (payloadProcessor.process(payload)) {