Load: Add check for reset memory size to 0 in LoadTsFileMemoryManager (#16940)

Author: luoluoyuyu

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/load/memory/LoadTsFileMemoryManager.java

@@ -82,6 +82,10 @@ public synchronized long tryAllocateFromQuery(final long sizeInBytes) {
   }
 
   public synchronized void releaseToQuery(final long sizeInBytes) {
+    if (sizeInBytes <= 0) {
+      throw new IllegalArgumentException(
+          String.format("Load: Invalid memory size %d bytes, must be positive", sizeInBytes));
+    }
     if (usedMemorySizeInBytes.get() < sizeInBytes) {
       LOGGER.error(
           "Load: Attempting to release more memory ({}) than allocated ({})",
@@ -96,6 +100,10 @@ public synchronized void releaseToQuery(final long sizeInBytes) {
 
   public synchronized LoadTsFileMemoryBlock allocateMemoryBlock(long sizeInBytes)
       throws LoadRuntimeOutOfMemoryException {
+    if (sizeInBytes <= 0) {
+      throw new IllegalArgumentException(
+          String.format("Load: Invalid memory size %d bytes, must be positive", sizeInBytes));
+    }
     try {
       forceAllocateFromQuery(sizeInBytes);
       if (LOGGER.isDebugEnabled()) {
@@ -120,7 +128,16 @@ public synchronized LoadTsFileMemoryBlock allocateMemoryBlock(long sizeInBytes)
    */
   synchronized void forceResize(LoadTsFileMemoryBlock memoryBlock, long newSizeInBytes)
       throws LoadRuntimeOutOfMemoryException {
-    if (memoryBlock.getTotalMemorySizeInBytes() >= newSizeInBytes) {
+    if (newSizeInBytes < 0) {
+      throw new IllegalArgumentException(
+          String.format(
+              "Load: Invalid memory size %d bytes, must be non-negative", newSizeInBytes));
+    }
+    if (memoryBlock.getTotalMemorySizeInBytes() == newSizeInBytes) {
+      return;
+    }
+
+    if (memoryBlock.getTotalMemorySizeInBytes() > newSizeInBytes) {
 
       if (memoryBlock.getMemoryUsageInBytes() > newSizeInBytes) {
         LOGGER.error(