temp

Author: THUMarkLau

server/src/main/java/org/apache/iotdb/db/engine/StorageEngine.java

@@ -555,7 +555,8 @@ public DataRegion buildNewStorageGroupProcessor(
             systemDir + File.separator + logicalStorageGroupName,
             virtualStorageGroupId,
             fileFlushPolicy,
-            storageGroupMNode.getFullPath());
+            storageGroupMNode.getFullPath(),
+            false);
     processor.setDataTTL(storageGroupMNode.getDataTTL());
     processor.setCustomFlushListeners(customFlushListeners);
     processor.setCustomCloseFileListeners(customCloseFileListeners);
@@ -1055,6 +1056,10 @@ protected void getSeriesSchemas(InsertPlan insertPlan, DataRegion processor)
     }
   }
 
+  public String getSystemDir() {
+    return systemDir;
+  }
+
   static class InstanceHolder {
 
     private static final StorageEngine INSTANCE = new StorageEngine();

server/src/main/java/org/apache/iotdb/db/engine/StorageEngineV2.java

@@ -481,7 +481,8 @@ public DataRegion buildNewDataRegion(
             systemDir + File.separator + logicalStorageGroupName,
             dataRegionId,
             fileFlushPolicy,
-            logicalStorageGroupName);
+            logicalStorageGroupName,
+            false);
     dataRegion.setDataTTL(ttl);
     dataRegion.setCustomFlushListeners(customFlushListeners);
     dataRegion.setCustomCloseFileListeners(customCloseFileListeners);

server/src/main/java/org/apache/iotdb/db/engine/snapshot/SnapshotLoader.java

@@ -0,0 +1,56 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.iotdb.db.engine.snapshot;
+
+import org.apache.iotdb.db.engine.StorageEngine;
+import org.apache.iotdb.db.engine.flush.TsFileFlushPolicy;
+import org.apache.iotdb.db.engine.storagegroup.DataRegion;
+import org.apache.iotdb.db.exception.DataRegionException;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.File;
+
+public class SnapshotLoader {
+  private static final Logger LOGGER = LoggerFactory.getLogger(SnapshotLoader.class);
+  private String dataDir;
+  private String storageGroupName;
+  private String dataRegionId;
+
+  public SnapshotLoader(String dataDir, String storageGroupName, String dataRegionId) {
+    this.dataDir = dataDir;
+    this.storageGroupName = storageGroupName;
+    this.dataRegionId = dataRegionId;
+  }
+
+  public DataRegion loadSnapshot() {
+    try {
+      return new DataRegion(
+          StorageEngine.getInstance().getSystemDir() + File.separator + storageGroupName,
+          dataRegionId,
+          new TsFileFlushPolicy.DirectFlushPolicy(),
+          storageGroupName,
+          true);
+    } catch (DataRegionException e) {
+      LOGGER.error("Exception occurs while loading snapshot", e);
+    }
+    return null;
+  }
+}

server/src/main/java/org/apache/iotdb/db/engine/storagegroup/DataRegion.java

@@ -263,6 +263,8 @@ public class DataRegion {
   /** used to collect TsFiles in this virtual storage group */
   private TsFileSyncManager tsFileSyncManager = TsFileSyncManager.getInstance();
 
+  private boolean recoverFromSnapshot = false;
+
   /**
    * constrcut a storage group processor
    *
@@ -275,11 +277,13 @@ public DataRegion(
       String systemDir,
       String dataRegionId,
       TsFileFlushPolicy fileFlushPolicy,
-      String logicalStorageGroupName)
+      String logicalStorageGroupName,
+      boolean recoverFromSnapshot)
       throws DataRegionException {
     this.dataRegionId = dataRegionId;
     this.logicalStorageGroupName = logicalStorageGroupName;
     this.fileFlushPolicy = fileFlushPolicy;
+    this.recoverFromSnapshot = recoverFromSnapshot;
 
     storageGroupSysDir = SystemFileFactory.INSTANCE.getFile(systemDir, dataRegionId);
     this.tsFileManager =

server/src/test/java/org/apache/iotdb/db/engine/compaction/cross/RewriteCrossSpaceCompactionTest.java

@@ -580,7 +580,8 @@ public void testOneDeletionDuringCompaction() throws Exception {
             STORAGE_GROUP_DIR.getPath(),
             "0",
             new TsFileFlushPolicy.DirectFlushPolicy(),
-            COMPACTION_TEST_SG);
+            COMPACTION_TEST_SG,
+            false);
     registerTimeseriesInMManger(4, 5, true);
     createFiles(2, 2, 3, 300, 0, 0, 50, 50, true, true);
     createFiles(2, 4, 5, 300, 700, 700, 50, 50, true, true);
@@ -689,7 +690,8 @@ public void testSeveralDeletionsDuringCompaction() throws Exception {
             STORAGE_GROUP_DIR.getPath(),
             "0",
             new TsFileFlushPolicy.DirectFlushPolicy(),
-            COMPACTION_TEST_SG);
+            COMPACTION_TEST_SG,
+            false);
     registerTimeseriesInMManger(4, 5, true);
     createFiles(2, 2, 3, 300, 0, 0, 50, 50, true, true);
     createFiles(2, 4, 5, 300, 700, 700, 50, 50, true, true);

server/src/test/java/org/apache/iotdb/db/engine/storagegroup/DataRegionTest.java

@@ -916,7 +916,7 @@ record = new TSRecord(1, deviceId);
   static class DummyDataRegion extends DataRegion {
 
     DummyDataRegion(String systemInfoDir, String storageGroupName) throws DataRegionException {
-      super(systemInfoDir, "0", new TsFileFlushPolicy.DirectFlushPolicy(), storageGroupName);
+      super(systemInfoDir, "0", new TsFileFlushPolicy.DirectFlushPolicy(), storageGroupName, false);
     }
   }
 }

server/src/test/java/org/apache/iotdb/db/engine/storagegroup/StorageGroupProcessorTest.java

@@ -863,7 +863,7 @@ record = new TSRecord(1, deviceId);
   class DummySGP extends DataRegion {
 
     DummySGP(String systemInfoDir, String storageGroupName) throws DataRegionException {
-      super(systemInfoDir, "0", new TsFileFlushPolicy.DirectFlushPolicy(), storageGroupName);
+      super(systemInfoDir, "0", new TsFileFlushPolicy.DirectFlushPolicy(), storageGroupName, false);
     }
   }
 }

server/src/test/java/org/apache/iotdb/db/engine/storagegroup/TTLTest.java

@@ -109,7 +109,8 @@ private void createSchemas() throws MetadataException, DataRegionException {
             IoTDBDescriptor.getInstance().getConfig().getSystemDir(),
             sg1,
             new DirectFlushPolicy(),
-            sg1);
+            sg1,
+            false);
     IoTDB.schemaProcessor.createTimeseries(
         new PartialPath(g1s1),
         TSDataType.INT64,