fix UT errors

Author: HTHou

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/planner/plan/node/write/RelationalInsertRowsNode.java

@@ -226,7 +226,7 @@ private void handleObjectValue(
           ByteBuffer valueBytes =
               ByteBuffer.allocate(relativePath.getSerializedSize() + Long.BYTES);
           valueBytes.putLong(offset + content.length);
-          relativePath.serialize(valueBytes);
+          relativePath.serializeToObjectValue(valueBytes);
           ((Binary) values[j]).setValues(valueBytes.array());
           insertRowNode.setValues(values);
         } else {

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/planner/plan/node/write/RelationalInsertTabletNode.java

@@ -477,7 +477,7 @@ private void handleObjectValue(
       if (isEoF) {
         ByteBuffer valueBytes = ByteBuffer.allocate(relativePath.getSerializedSize() + Long.BYTES);
         valueBytes.putLong(offset + content.length);
-        relativePath.serialize(valueBytes);
+        relativePath.serializeToObjectValue(valueBytes);
         ((Binary[]) columns[column])[j] = new Binary(valueBytes.array());
       } else {
         ((Binary[]) columns[column])[j] = null;

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/Base32ObjectPath.java

@@ -48,6 +48,11 @@ public IObjectPath deserializeFrom(ByteBuffer byteBuffer) {
         public IObjectPath deserializeFrom(InputStream inputStream) throws IOException {
           return deserialize(inputStream);
         }
+
+        @Override
+        public IObjectPath deserializeFromObjectValue(ByteBuffer byteBuffer) {
+          return deserialize(byteBuffer);
+        }
       };
 
   private static final Factory FACTORY = Base32ObjectPath::new;
@@ -108,6 +113,11 @@ public int getSerializedSize() {
     return cnt;
   }
 
+  @Override
+  public void serializeToObjectValue(ByteBuffer byteBuffer) {
+    serialize(byteBuffer);
+  }
+
   public static Base32ObjectPath deserialize(ByteBuffer byteBuffer) {
     int cnt = ReadWriteForEncodingUtils.readUnsignedVarInt(byteBuffer);
     String first = ReadWriteIOUtils.readVarIntString(byteBuffer);

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/IObjectPath.java

@@ -39,6 +39,8 @@ public interface IObjectPath {
 
   int getSerializedSize();
 
+  void serializeToObjectValue(ByteBuffer byteBuffer);
+
   interface Factory {
 
     IObjectPath create(int regionId, long time, IDeviceID iDeviceID, String measurement);
@@ -55,6 +57,8 @@ interface Deserializer {
 
     IObjectPath deserializeFrom(InputStream inputStream) throws IOException;
 
+    IObjectPath deserializeFromObjectValue(ByteBuffer byteBuffer);
+
     Deserializer DESERIALIZER =
         CONFIG.getRestrictObjectLimit()
             ? PlainObjectPath.getDESERIALIZER()

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/PlainObjectPath.java

@@ -27,6 +27,7 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.nio.ByteBuffer;
+import java.nio.charset.StandardCharsets;
 
 public class PlainObjectPath implements IObjectPath {
 
@@ -43,6 +44,11 @@ public IObjectPath deserializeFrom(ByteBuffer byteBuffer) {
         public IObjectPath deserializeFrom(InputStream inputStream) throws IOException {
           return deserialize(inputStream);
         }
+
+        @Override
+        public IObjectPath deserializeFromObjectValue(ByteBuffer byteBuffer) {
+          return deserializeObjectValue(byteBuffer);
+        }
       };
 
   private static final Factory FACTORY = PlainObjectPath::new;
@@ -81,6 +87,11 @@ public int getSerializedSize() {
     return ReadWriteIOUtils.sizeToWrite(filePath);
   }
 
+  @Override
+  public void serializeToObjectValue(ByteBuffer byteBuffer) {
+    byteBuffer.put(filePath.getBytes(StandardCharsets.UTF_8));
+  }
+
   public static PlainObjectPath deserialize(ByteBuffer byteBuffer) {
     String filePath = ReadWriteIOUtils.readString(byteBuffer);
     return new PlainObjectPath(filePath);
@@ -91,6 +102,12 @@ public static PlainObjectPath deserialize(InputStream stream) throws IOException
     return new PlainObjectPath(filePath);
   }
 
+  public static PlainObjectPath deserializeObjectValue(ByteBuffer byteBuffer) {
+    byte[] bytes = new byte[byteBuffer.remaining()];
+    byteBuffer.get(bytes);
+    return new PlainObjectPath(new String(bytes, StandardCharsets.UTF_8));
+  }
+
   @Override
   public String toString() {
     return filePath;

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

@@ -169,7 +169,7 @@ public static Pair<Long, String> parseObjectBinary(Binary binary) {
     ByteBuffer buffer = ByteBuffer.wrap(bytes);
     long length = buffer.getLong();
     String relativeObjectFilePath =
-        IObjectPath.Deserializer.DESERIALIZER.deserializeFrom(buffer).toString();
+        IObjectPath.Deserializer.DESERIALIZER.deserializeFromObjectValue(buffer).toString();
     return new Pair<>(length, relativeObjectFilePath);
   }
 
@@ -184,7 +184,7 @@ public static Optional<File> getNullableObjectPathFromBinary(
     byte[] bytes = binary.getValues();
     ByteBuffer buffer = ByteBuffer.wrap(bytes, 8, bytes.length - 8);
     String relativeObjectFilePath =
-        IObjectPath.Deserializer.DESERIALIZER.deserializeFrom(buffer).toString();
+        IObjectPath.Deserializer.DESERIALIZER.deserializeFromObjectValue(buffer).toString();
     return TIER_MANAGER.getAbsoluteObjectFilePath(relativeObjectFilePath, needTempFile);
   }
 

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/plan/function/RecordObjectTypeTest.java

@@ -19,6 +19,8 @@
 
 package org.apache.iotdb.db.queryengine.plan.function;
 
+import org.apache.iotdb.db.conf.IoTDBConfig;
+import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.exception.DiskSpaceInsufficientException;
 import org.apache.iotdb.db.queryengine.execution.operator.process.function.partition.Slice;
 import org.apache.iotdb.db.queryengine.execution.operator.source.relational.aggregation.RecordIterator;
@@ -50,10 +52,13 @@
 
 public class RecordObjectTypeTest {
 
+  private final IoTDBConfig config = IoTDBDescriptor.getInstance().getConfig();
+
   private File objectDir;
 
   @Before
   public void setup() {
+    config.setRestrictObjectLimit(true);
     try {
       objectDir = new File(TierManager.getInstance().getNextFolderForObjectFile());
     } catch (DiskSpaceInsufficientException e) {
@@ -69,6 +74,7 @@ public void tearDown() throws IOException {
         Files.delete(file.toPath());
       }
     }
+    config.setRestrictObjectLimit(true);
   }
 
   @Test

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/transformation/dag/column/unary/scalar/ObjectTypeFunctionTest.java

@@ -19,6 +19,8 @@
 
 package org.apache.iotdb.db.queryengine.transformation.dag.column.unary.scalar;
 
+import org.apache.iotdb.db.conf.IoTDBConfig;
+import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.exception.DiskSpaceInsufficientException;
 import org.apache.iotdb.db.queryengine.transformation.dag.column.ColumnTransformer;
 import org.apache.iotdb.db.storageengine.rescon.disk.TierManager;
@@ -48,10 +50,13 @@
 
 public class ObjectTypeFunctionTest {
 
+  private final IoTDBConfig config = IoTDBDescriptor.getInstance().getConfig();
+
   private File objectDir;
 
   @Before
   public void setup() {
+    config.setRestrictObjectLimit(true);
     try {
       objectDir = new File(TierManager.getInstance().getNextFolderForObjectFile());
     } catch (DiskSpaceInsufficientException e) {
@@ -67,6 +72,7 @@ public void tearDown() throws IOException {
         Files.delete(file.toPath());
       }
     }
+    config.setRestrictObjectLimit(false);
   }
 
   @Test

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/storageengine/dataregion/compaction/object/ObjectTypeCompactionTest.java

@@ -23,6 +23,8 @@
 import org.apache.iotdb.commons.schema.table.TsTable;
 import org.apache.iotdb.commons.schema.table.column.FieldColumnSchema;
 import org.apache.iotdb.commons.schema.table.column.TagColumnSchema;
+import org.apache.iotdb.db.conf.IoTDBConfig;
+import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.exception.DiskSpaceInsufficientException;
 import org.apache.iotdb.db.exception.StorageEngineException;
 import org.apache.iotdb.db.schemaengine.table.DataNodeTableCache;
@@ -77,10 +79,13 @@ public class ObjectTypeCompactionTest extends AbstractCompactionTest {
   private String threadName;
   private File objectDir;
 
+  private final IoTDBConfig config = IoTDBDescriptor.getInstance().getConfig();
+
   @Before
   @Override
   public void setUp()
       throws IOException, WriteProcessException, MetadataException, InterruptedException {
+    config.setRestrictObjectLimit(true);
     this.threadName = Thread.currentThread().getName();
     Thread.currentThread().setName("pool-1-IoTDB-Compaction-Worker-1");
     DataNodeTableCache.getInstance().invalid(this.COMPACTION_TEST_SG);
@@ -105,6 +110,7 @@ public void tearDown() throws IOException, StorageEngineException {
         Files.delete(file.toPath());
       }
     }
+    config.setRestrictObjectLimit(false);
   }
 
   public void createTable(String tableName, long ttl) {