Fix null pointer exception when 'infoRecord' is null

ballerina-platform · Oct 24, 2023 · bca1b13 · bca1b13
1 parent 4460551
commit bca1b13
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/...rina-runtime/src/main/java/io/ballerina/runtime/transactions/TransactionLocalContext.java b/...rina-runtime/src/main/java/io/ballerina/runtime/transactions/TransactionLocalContext.java
@@ -64,8 +64,14 @@ private TransactionLocalContext(String globalTransactionId, String url, String p
         this.rollbackOnlyError = null;
         this.isTransactional = true;
         this.transactionId = ValueCreator.createArrayValue(globalTransactionId.getBytes());
-        transactionResourceManager.transactionInfoMap.put(ByteBuffer.wrap(transactionId.getBytes().clone()),
-                infoRecord);
+        validateAndPutTransactionInfo(ByteBuffer.wrap(transactionId.getBytes().clone()), infoRecord);
+    }
+
+    private void validateAndPutTransactionInfo(ByteBuffer transactionIdBytes, Object infoRecord) {
+        if (infoRecord == null) {
+            return;
+        }
+        transactionResourceManager.transactionInfoMap.put(transactionIdBytes, infoRecord);
     }
 
     public static TransactionLocalContext createTransactionParticipantLocalCtx(String globalTransactionId,
@@ -228,7 +234,7 @@ public void setResourceParticipant(boolean resourceParticipant) {
     }
 
     public Object getInfoRecord() {
-        return transactionResourceManager.transactionInfoMap.get(ByteBuffer.wrap(transactionId.getBytes()));
+        return transactionResourceManager.getTransactionRecord(transactionId);
     }
 
     public boolean isTransactional() {

diff --git a/...a-runtime/src/main/java/io/ballerina/runtime/transactions/TransactionResourceManager.java b/...a-runtime/src/main/java/io/ballerina/runtime/transactions/TransactionResourceManager.java
@@ -606,6 +606,11 @@ public void notifyLocalParticipantFailure(String gTransactionId, String blockId)
     }
 
     public Object getTransactionRecord(BArray xid) {
-        return transactionInfoMap.get(ByteBuffer.wrap(xid.getBytes()));
+        synchronized (transactionInfoMap) {
+            if (transactionInfoMap.containsKey(ByteBuffer.wrap(xid.getBytes()))) {
+                return transactionInfoMap.get(ByteBuffer.wrap(xid.getBytes()));
+            }
+            return null;
+        }
     }
 }