fix: data store ref count

Author: shizy818

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/source/relational/CteScanOperator.java

@@ -41,6 +41,7 @@ public class CteScanOperator extends AbstractSourceOperator {
       RamUsageEstimator.shallowSizeOfInstance(CteScanOperator.class);
 
   private final CteDataStore dataStore;
+  private final int dataStoreRefCount;
   private CteDataReader dataReader;
 
   private final long maxReturnSize =
@@ -50,8 +51,8 @@ public CteScanOperator(
       OperatorContext operatorContext, PlanNodeId sourceId, CteDataStore dataStore) {
     this.operatorContext = operatorContext;
     this.sourceId = sourceId;
-    dataStore.increaseRefCount();
     this.dataStore = dataStore;
+    this.dataStoreRefCount = dataStore.increaseRefCount();
     prepareReader();
   }
 
@@ -109,7 +110,7 @@ public long ramBytesUsed() {
     if (dataReader != null) {
       bytes += dataReader.bytesUsed();
     }
-    if (dataStore.getRefCount() == 1) {
+    if (dataStoreRefCount == 1) {
       bytes += dataStore.getCachedBytes();
     }
 

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/utils/cte/CteDataStore.java

@@ -95,11 +95,7 @@ public List<Integer> getColumnIndex2TsBlockColumnIndexList() {
     return columnIndex2TsBlockColumnIndexList;
   }
 
-  public int getRefCount() {
-    return count.get();
-  }
-
-  public void increaseRefCount() {
-    count.getAndIncrement();
+  public int increaseRefCount() {
+    return count.incrementAndGet();
   }
 }