Checked if near cache was available before access

hazelcast/src/main/java/com/hazelcast/internal/nearcache/NearCache.java

@@ -116,6 +116,9 @@ public interface NearCache<K, V> extends InitializingObject {
      */
     int size();
 
+
+    boolean isAvailable();
+
     /**
      * Gets the {@link com.hazelcast.config.InMemoryFormat} of the storage for internal records.
      *

hazelcast/src/main/java/com/hazelcast/internal/nearcache/NearCacheRecordStore.java

@@ -136,4 +136,6 @@ public interface NearCacheRecordStore<K, V> extends InitializingObject {
     long tryReserveForUpdate(K key, Data keyData);
 
     V tryPublishReserved(K key, V value, long reservationId, boolean deserialize);
+
+    boolean isAvailable();
 }

hazelcast/src/main/java/com/hazelcast/internal/nearcache/impl/DefaultNearCache.java

@@ -177,6 +177,11 @@ public int size() {
         return nearCacheRecordStore.size();
     }
 
+    @Override
+    public boolean isAvailable() {
+        return nearCacheRecordStore.isAvailable();
+    }
+
     @Override
     public void preload(DataStructureAdapter<Object, ?> adapter) {
         nearCacheRecordStore.loadKeys(adapter);

hazelcast/src/main/java/com/hazelcast/internal/nearcache/impl/store/AbstractNearCacheRecordStore.java

@@ -130,6 +130,7 @@ public void setStaleReadDetector(StaleReadDetector staleReadDetector) {
 
     @Override
     public StaleReadDetector getStaleReadDetector() {
+        checkAvailable();
         return staleReadDetector;
     }
 
@@ -167,7 +168,8 @@ protected void checkAvailable() {
         }
     }
 
-    protected boolean isAvailable() {
+    @Override
+    public boolean isAvailable() {
         return records != null;
     }
 
@@ -397,6 +399,7 @@ public long tryReserveForUpdate(K key, Data keyData) {
 
     @Override
     public V tryPublishReserved(K key, V value, long reservationId, boolean deserialize) {
+        checkAvailable();
         return updateAndGetReserved(key, value, reservationId, deserialize);
     }
 

hazelcast/src/test/java/com/hazelcast/internal/nearcache/NearCacheTestSupport.java

@@ -400,5 +400,10 @@ public long tryReserveForUpdate(Integer key, Data keyData) {
         public String tryPublishReserved(Integer key, String value, long reservationId, boolean deserialize) {
             return null;
         }
+
+        @Override
+        public boolean isAvailable() {
+            return true;
+        }
     }
 }

hazelcast/src/test/java/com/hazelcast/internal/nearcache/NearCacheTestUtils.java

@@ -443,9 +443,12 @@ public void run() {
     public static void assertNearCacheSize(NearCacheTestContext<?, ?, ?, ?> context, long expectedSize, String... messages) {
         String message = messages.length > 0 ? messages[0] + " " : "";
 
-        long nearCacheSize = context.nearCache.size();
-        assertEquals(format("%sNear Cache size didn't reach the desired value (%d vs. %d) (%s)",
-                message, expectedSize, nearCacheSize, context.stats), expectedSize, nearCacheSize);
+        if (context.nearCache.isAvailable()) {
+            // if near cache is destroyed, it will not be available for size check.
+            long nearCacheSize = context.nearCache.size();
+            assertEquals(format("%sNear Cache size didn't reach the desired value (%d vs. %d) (%s)",
+                    message, expectedSize, nearCacheSize, context.stats), expectedSize, nearCacheSize);
+        }
 
         long ownedEntryCount = context.stats.getOwnedEntryCount();
         assertEquals(format("%sNear Cache owned entry count didn't reach the desired value (%d vs. %d) (%s)",