add encrypt param when constructing CompactionTsFileReader (apache#16578)

* add encrypt param when constructing CompactionTsFileReader

* use get path instead of get databaseName

Author: zhujt20

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/execute/utils/MultiTsFileDeviceIterator.java

@@ -32,6 +32,7 @@
 import org.apache.iotdb.db.storageengine.dataregion.modification.TreeDeletionEntry;
 import org.apache.iotdb.db.storageengine.dataregion.read.control.FileReaderManager;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource;
+import org.apache.iotdb.db.utils.EncryptDBUtils;
 import org.apache.iotdb.db.utils.ModificationUtils;
 import org.apache.iotdb.db.utils.datastructure.PatternTreeMapFactory;
 
@@ -98,7 +99,9 @@ public MultiTsFileDeviceIterator(List<TsFileResource> tsFileResources) throws IO
       for (TsFileResource tsFileResource : this.tsFileResourcesSortedByDesc) {
         CompactionTsFileReader reader =
             new CompactionTsFileReader(
-                tsFileResource.getTsFilePath(), CompactionType.INNER_SEQ_COMPACTION);
+                tsFileResource.getTsFilePath(),
+                CompactionType.INNER_SEQ_COMPACTION,
+                EncryptDBUtils.getFirstEncryptParamFromTSFilePath(tsFileResource.getTsFilePath()));
         readerMap.put(tsFileResource, reader);
         deviceIteratorMap.put(tsFileResource, reader.getAllDevicesIteratorWithIsAligned());
       }
@@ -163,7 +166,10 @@ public MultiTsFileDeviceIterator(
 
     for (TsFileResource tsFileResource : tsFileResourcesSortedByDesc) {
       TsFileSequenceReader reader =
-          new CompactionTsFileReader(tsFileResource.getTsFilePath(), type);
+          new CompactionTsFileReader(
+              tsFileResource.getTsFilePath(),
+              type,
+              EncryptDBUtils.getFirstEncryptParamFromTSFilePath(tsFileResource.getTsFilePath()));
       readerMap.put(tsFileResource, reader);
       deviceIteratorMap.put(tsFileResource, reader.getAllDevicesIteratorWithIsAligned());
     }

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/io/CompactionTsFileReader.java

@@ -19,8 +19,10 @@
 
 package org.apache.iotdb.db.storageengine.dataregion.compaction.io;
 
+import org.apache.iotdb.commons.utils.TestOnly;
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.constant.CompactionType;
 
+import org.apache.tsfile.encrypt.EncryptParameter;
 import org.apache.tsfile.file.IMetadataIndexEntry;
 import org.apache.tsfile.file.header.ChunkHeader;
 import org.apache.tsfile.file.metadata.MetadataIndexNode;
@@ -42,16 +44,27 @@
  * data read and distinguishing between aligned and not aligned series during compaction.
  */
 public class CompactionTsFileReader extends TsFileSequenceReader {
+  @TestOnly
+  public CompactionTsFileReader(String file, CompactionType compactionType) throws IOException {
+    super(file, new EncryptParameter(config.getEncryptType(), config.getEncryptKey()));
+    CompactionTsFileInput compactionTsFileInput =
+        new CompactionTsFileInput(compactionType, tsFileInput);
+    this.tsFileInput = compactionTsFileInput;
+    compactionTsFileInput.setMetadataOffset(readFileMetadata().getMetaOffset());
+  }
 
   /**
    * Constructs a new instance of CompactionTsFileReader.
    *
    * @param file The file to be read.
    * @param compactionType The type of compaction running.
+   * @param encryptParameter The first encryption parameters for the file.
    * @throws IOException If an error occurs during file operations.
    */
-  public CompactionTsFileReader(String file, CompactionType compactionType) throws IOException {
-    super(file);
+  public CompactionTsFileReader(
+      String file, CompactionType compactionType, EncryptParameter encryptParameter)
+      throws IOException {
+    super(file, encryptParameter);
     CompactionTsFileInput compactionTsFileInput =
         new CompactionTsFileInput(compactionType, tsFileInput);
     this.tsFileInput = compactionTsFileInput;

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/repair/RepairDataFileScanUtil.java

@@ -29,6 +29,7 @@
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResourceStatus;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.timeindex.ArrayDeviceTimeIndex;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.timeindex.ITimeIndex;
+import org.apache.iotdb.db.utils.EncryptDBUtils;
 
 import org.apache.tsfile.common.conf.TSFileDescriptor;
 import org.apache.tsfile.common.constant.TsFileConstant;
@@ -110,7 +111,8 @@ public void scanTsFile(boolean checkTsFileResource) {
             tsfile.getPath(),
             resource.isSeq()
                 ? CompactionType.INNER_SEQ_COMPACTION
-                : CompactionType.INNER_UNSEQ_COMPACTION)) {
+                : CompactionType.INNER_UNSEQ_COMPACTION,
+            EncryptDBUtils.getFirstEncryptParamFromTSFilePath(tsfile.getAbsolutePath()))) {
       TsFileDeviceIterator deviceInFileIterator = reader.getAllDevicesIteratorWithIsAligned();
       Set<IDeviceID> deviceIdsInTimeIndex =
           checkTsFileResource ? new HashSet<>(timeIndex.getDevices()) : Collections.emptySet();

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/selector/estimator/AbstractCrossSpaceEstimator.java

@@ -23,6 +23,7 @@
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.CompactionScheduleContext;
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.constant.CompactionType;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource;
+import org.apache.iotdb.db.utils.EncryptDBUtils;
 
 import org.apache.tsfile.read.TsFileSequenceReader;
 
@@ -38,7 +39,10 @@ public abstract class AbstractCrossSpaceEstimator extends AbstractCompactionEsti
 
   @Override
   protected TsFileSequenceReader getReader(String filePath) throws IOException {
-    return new CompactionTsFileReader(filePath, CompactionType.CROSS_COMPACTION);
+    return new CompactionTsFileReader(
+        filePath,
+        CompactionType.CROSS_COMPACTION,
+        EncryptDBUtils.getFirstEncryptParamFromTSFilePath(filePath));
   }
 
   public long estimateCrossCompactionMemory(

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/selector/estimator/AbstractInnerSpaceEstimator.java

@@ -24,6 +24,7 @@
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.CompactionScheduleContext;
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.constant.CompactionType;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource;
+import org.apache.iotdb.db.utils.EncryptDBUtils;
 
 import org.apache.tsfile.read.TsFileSequenceReader;
 
@@ -39,9 +40,15 @@ public abstract class AbstractInnerSpaceEstimator extends AbstractCompactionEsti
   @Override
   protected TsFileSequenceReader getReader(String filePath) throws IOException {
     if (filePath.contains(IoTDBConstant.UNSEQUENCE_FOLDER_NAME)) {
-      return new CompactionTsFileReader(filePath, CompactionType.INNER_UNSEQ_COMPACTION);
+      return new CompactionTsFileReader(
+          filePath,
+          CompactionType.INNER_UNSEQ_COMPACTION,
+          EncryptDBUtils.getFirstEncryptParamFromTSFilePath(filePath));
     } else {
-      return new CompactionTsFileReader(filePath, CompactionType.INNER_SEQ_COMPACTION);
+      return new CompactionTsFileReader(
+          filePath,
+          CompactionType.INNER_SEQ_COMPACTION,
+          EncryptDBUtils.getFirstEncryptParamFromTSFilePath(filePath));
     }
   }
 

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/dataregion/compaction/selector/estimator/CompactionEstimateUtils.java

@@ -26,6 +26,7 @@
 import org.apache.iotdb.db.storageengine.dataregion.compaction.schedule.constant.CompactionType;
 import org.apache.iotdb.db.storageengine.dataregion.tsfile.TsFileResource;
 import org.apache.iotdb.db.storageengine.rescon.memory.SystemInfo;
+import org.apache.iotdb.db.utils.EncryptDBUtils;
 
 import org.apache.tsfile.enums.TSDataType;
 import org.apache.tsfile.file.metadata.ChunkMetadata;
@@ -169,7 +170,10 @@ static CompactionTaskMetadataInfo collectMetadataInfoFromDisk(
       for (TsFileResource resource : resources) {
         cost += resource.getTotalModSizeInByte();
         try (CompactionTsFileReader reader =
-            new CompactionTsFileReader(resource.getTsFilePath(), taskType)) {
+            new CompactionTsFileReader(
+                resource.getTsFilePath(),
+                taskType,
+                EncryptDBUtils.getFirstEncryptParamFromTSFilePath(resource.getTsFilePath()))) {
           for (Map.Entry<IDeviceID, Long> entry :
               getDeviceMetadataSizeMapAndCollectMetadataInfo(reader, metadataInfo).entrySet()) {
             deviceMetadataSizeMap.merge(entry.getKey(), entry.getValue(), Long::sum);