initial commit

Author: inv-jishnu

data-loader/cli/src/main/java/com/scalar/db/dataloader/cli/command/dataexport/ExportCommand.java

@@ -6,6 +6,7 @@
 import static java.nio.file.StandardOpenOption.CREATE;
 
 import com.scalar.db.api.DistributedStorage;
+import com.scalar.db.api.DistributedTransactionManager;
 import com.scalar.db.api.TableMetadata;
 import com.scalar.db.dataloader.cli.exception.DirectoryValidationException;
 import com.scalar.db.dataloader.cli.util.DirectoryUtils;
@@ -36,6 +37,8 @@
 import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.Callable;
+
+import com.scalar.db.service.TransactionFactory;
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -64,6 +67,7 @@ public Integer call() throws Exception {
       validatePositiveValue(spec.commandLine(), maxThreads, DataLoaderError.INVALID_MAX_THREADS);
 
       StorageFactory storageFactory = StorageFactory.create(scalarDbPropertiesFilePath);
+      TransactionFactory transactionFactory = TransactionFactory.create(scalarDbPropertiesFilePath)
       TableMetadataService metaDataService =
           new TableMetadataService(storageFactory.getStorageAdmin());
       ScalarDbDao scalarDbDao = new ScalarDbDao();
@@ -146,10 +150,10 @@ private void validateOutputDirectory() throws DirectoryValidationException {
   }
 
   private ExportManager createExportManager(
-      StorageFactory storageFactory, ScalarDbDao scalarDbDao, FileFormat fileFormat) {
+      TransactionFactory transactionFactory, ScalarDbDao scalarDbDao, FileFormat fileFormat) {
     ProducerTaskFactory taskFactory =
         new ProducerTaskFactory(delimiter, includeTransactionMetadata, prettyPrintJson);
-    DistributedStorage storage = storageFactory.getStorage();
+      DistributedTransactionManager manager = transactionFactory.getTransactionManager();
     switch (fileFormat) {
       case JSON:
         return new JsonExportManager(storage, scalarDbDao, taskFactory);

data-loader/core/src/main/java/com/scalar/db/dataloader/core/dataexport/CsvExportManager.java

@@ -1,6 +1,7 @@
 package com.scalar.db.dataloader.core.dataexport;
 
 import com.scalar.db.api.DistributedStorage;
+import com.scalar.db.api.DistributedTransactionManager;
 import com.scalar.db.api.TableMetadata;
 import com.scalar.db.dataloader.core.dataexport.producer.ProducerTaskFactory;
 import com.scalar.db.dataloader.core.dataimport.dao.ScalarDbDao;
@@ -18,13 +19,13 @@ public class CsvExportManager extends ExportManager {
    * Constructs a {@code CsvExportManager} with the specified {@link DistributedStorage}, {@link
    * ScalarDbDao}, and {@link ProducerTaskFactory}.
    *
-   * @param storage the {@code DistributedStorage} instance used to read data from the database
+   * @param manager the {@code DistributedTransactionManager} instance used to read data from the database
    * @param dao the {@code ScalarDbDao} used to execute export-related database operations
    * @param producerTaskFactory the factory used to create producer tasks for exporting data
    */
   public CsvExportManager(
-      DistributedStorage storage, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
-    super(storage, dao, producerTaskFactory);
+          DistributedTransactionManager manager, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
+    super(manager, dao, producerTaskFactory);
   }
 
   /**

data-loader/core/src/main/java/com/scalar/db/dataloader/core/dataexport/ExportManager.java

@@ -1,6 +1,7 @@
 package com.scalar.db.dataloader.core.dataexport;
 
 import com.scalar.db.api.DistributedStorage;
+import com.scalar.db.api.DistributedTransactionManager;
 import com.scalar.db.api.Result;
 import com.scalar.db.api.Scanner;
 import com.scalar.db.api.TableMetadata;
@@ -34,7 +35,7 @@
 public abstract class ExportManager {
   private static final Logger logger = LoggerFactory.getLogger(ExportManager.class);
 
-  private final DistributedStorage storage;
+  private final DistributedTransactionManager distributedTransactionManager;
   private final ScalarDbDao dao;
   private final ProducerTaskFactory producerTaskFactory;
   private final Object lock = new Object();

data-loader/core/src/main/java/com/scalar/db/dataloader/core/dataexport/JsonExportManager.java

@@ -1,6 +1,7 @@
 package com.scalar.db.dataloader.core.dataexport;
 
 import com.scalar.db.api.DistributedStorage;
+import com.scalar.db.api.DistributedTransactionManager;
 import com.scalar.db.api.TableMetadata;
 import com.scalar.db.dataloader.core.dataexport.producer.ProducerTaskFactory;
 import com.scalar.db.dataloader.core.dataimport.dao.ScalarDbDao;
@@ -11,16 +12,16 @@
 public class JsonExportManager extends ExportManager {
 
   /**
-   * Constructs a {@code JsonExportManager} with the specified {@link DistributedStorage}, {@link
+   * Constructs a {@code JsonExportManager} with the specified {@link DistributedTransactionManager}, {@link
    * ScalarDbDao}, and {@link ProducerTaskFactory}.
    *
-   * @param storage the {@code DistributedStorage} instance used to read data from the database
+   * @param manager the {@code DistributedTransactionManager} instance used to read data from the database
    * @param dao the {@code ScalarDbDao} used to execute export-related database operations
    * @param producerTaskFactory the factory used to create producer tasks for exporting data
    */
   public JsonExportManager(
-      DistributedStorage storage, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
-    super(storage, dao, producerTaskFactory);
+          DistributedTransactionManager manager, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
+    super(manager, dao, producerTaskFactory);
   }
 
   /**

data-loader/core/src/main/java/com/scalar/db/dataloader/core/dataexport/JsonLineExportManager.java

@@ -1,6 +1,7 @@
 package com.scalar.db.dataloader.core.dataexport;
 
 import com.scalar.db.api.DistributedStorage;
+import com.scalar.db.api.DistributedTransactionManager;
 import com.scalar.db.api.TableMetadata;
 import com.scalar.db.dataloader.core.dataexport.producer.ProducerTaskFactory;
 import com.scalar.db.dataloader.core.dataimport.dao.ScalarDbDao;
@@ -13,16 +14,16 @@
 public class JsonLineExportManager extends ExportManager {
 
   /**
-   * Constructs a {@code JsonLineExportManager} with the specified {@link DistributedStorage},
+   * Constructs a {@code JsonLineExportManager} with the specified {@link DistributedTransactionManager},
    * {@link ScalarDbDao}, and {@link ProducerTaskFactory}.
    *
-   * @param storage the {@code DistributedStorage} instance used to read data from the database
+   * @param manager the {@code DistributedTransactionManager} instance used to read data from the database
    * @param dao the {@code ScalarDbDao} used to execute export-related database operations
    * @param producerTaskFactory the factory used to create producer tasks for exporting data
    */
   public JsonLineExportManager(
-      DistributedStorage storage, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
-    super(storage, dao, producerTaskFactory);
+          DistributedTransactionManager manager, ScalarDbDao dao, ProducerTaskFactory producerTaskFactory) {
+    super(manager, dao, producerTaskFactory);
   }
 
   /**