feat(export) - VTD-export - export data, fix parse number strategy

VitaDairy · Sep 5, 2024 · 686be32 · 686be32
1 parent 54bbd6a
commit 686be32
Show file tree

Hide file tree

Showing 11 changed files with 125 additions and 6 deletions.
diff --git a/.gitignore b/.gitignore
@@ -43,3 +43,4 @@ bin/
 
 .idea
 log/
+output/
diff --git a/person.xlsx b/person.xlsx
diff --git a/src/main/java/com/vitadairy/libraries/importexport/processor/WriteCellProcessor.java b/src/main/java/com/vitadairy/libraries/importexport/processor/WriteCellProcessor.java
@@ -26,7 +26,7 @@ public void write(Cell cell, Object input) throws Exception {
                 cell.setCellValue((String) cellData);
                 break;
             case NUMBER:
-                cell.setCellValue((double) cellData);
+                cell.setCellValue(((Number) cellData).doubleValue());
                 break;
             case DATE:
                 cell.setCellValue((Date) cellData);

diff --git a/src/main/java/com/vitadairy/libraries/importexport/service/WriteExportFileServiceImpl.java b/src/main/java/com/vitadairy/libraries/importexport/service/WriteExportFileServiceImpl.java
@@ -91,7 +91,7 @@ protected void writeExportFileHeader(Sheet sheet, Map<Integer, CellMetaData> cel
         cellMetadata.forEach((key, value) -> {
             // write header
             Cell cell = header.createCell(key);
-            cell.setCellValue(value.getFieldName());
+            cell.setCellValue(value.getName());
         });
     }
 

diff --git a/src/main/java/com/vitadairy/libraries/importexport/utils/WorkbookUtil.java b/src/main/java/com/vitadairy/libraries/importexport/utils/WorkbookUtil.java
@@ -28,6 +28,10 @@ public static void write(Workbook workbook, String filePath, boolean isOverride,
             if (file.exists() && !isOverride) {
                 return;
             }
+            if (!file.exists()) {
+                file.getParentFile().mkdirs();
+                file.createNewFile();
+            }
 
             outputStream = new FileOutputStream(file);
             workbook.write(outputStream);

diff --git a/src/test/java/example/Configuration.java b/src/test/java/example/Configuration.java
@@ -3,9 +3,12 @@
 import com.vitadairy.libraries.importexport.config.Log4j2Configurator;
 import com.vitadairy.libraries.importexport.helper.Log4j2Logger;
 import com.vitadairy.libraries.importexport.processor.ReadRowProcessor;
+import com.vitadairy.libraries.importexport.processor.WriteRowProcessor;
 import com.vitadairy.libraries.importexport.service.*;
 import example.common.Person;
 import example.configuration.PersonColumnMetaData;
+import example.dto.FetchPersonRequest;
+import example.service.FetchPersonService;
 import example.service.ParsePersonService;
 
 /**
@@ -33,6 +36,22 @@ public ReadImportFileService<Person> getReadImportFileService() {
         );
     }
 
+    public WriteExportFileService<Person, FetchPersonRequest> getWriteExportFileService() {
+        return new WriteExportFileServiceImpl<>(
+                new FetchPersonService(),
+                new WriteRowProcessor(
+                        new WriteDataServiceStrategy(
+                                new WriteDataDefaultService(),
+                                new WriteDataNumberService(),
+                                new WriteDataDateService("dd/MM/yyyy", logger)
+                        ),
+                        PersonColumnMetaData.getMetaDataMap(),
+                        logger
+                ),
+                logger
+        );
+    }
+
     public static Configuration getInstance() {
         return Singleton.INSTANCE;
     }

diff --git a/src/test/java/example/Main.java b/src/test/java/example/Main.java
@@ -1,9 +1,14 @@
 package example;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.vitadairy.libraries.importexport.common.ExportResponse;
 import com.vitadairy.libraries.importexport.common.ReadResponse;
-import example.common.Person;
+import com.vitadairy.libraries.importexport.dto.FetchRequest;
+import com.vitadairy.libraries.importexport.dto.Page;
 import com.vitadairy.libraries.importexport.service.ReadImportFileService;
+import com.vitadairy.libraries.importexport.service.WriteExportFileService;
+import example.common.Person;
+import example.dto.FetchPersonRequest;
 
 import java.io.File;
 
@@ -22,5 +27,14 @@ public static void main(String[] args) throws Exception {
 
         ReadResponse<Person> readResponse = readService.importFile(fileImport);
         Configuration.logger.info(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(readResponse));
+
+        WriteExportFileService<Person, FetchPersonRequest> writeService = Configuration.getInstance().getWriteExportFileService();
+        String folderPath = System.getProperty("user.dir") + File.separator + "output";
+        String fileName = "person_export";
+        FetchPersonRequest fetchPersonRequest = new FetchPersonRequest();
+        FetchRequest<FetchPersonRequest> fetchRequest = new FetchRequest<>(fetchPersonRequest, new Page(1, 10), false);
+
+        ExportResponse exportRes = writeService.exportFile(folderPath, fileName, fetchRequest);
+        Configuration.logger.info(objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(exportRes));
     }
 }
diff --git a/src/test/java/example/common/Person.java b/src/test/java/example/common/Person.java
@@ -40,4 +40,32 @@ public String getAddress() {
     public void setAddress(String address) {
         this.address = address;
     }
+
+    public static class Builder {
+        private String name;
+        private Integer age;
+        private String address;
+
+        public Builder() {
+        }
+
+        public Builder name(String name) {
+            this.name = name;
+            return this;
+        }
+
+        public Builder age(Integer age) {
+            this.age = age;
+            return this;
+        }
+
+        public Builder address(String address) {
+            this.address = address;
+            return this;
+        }
+
+        public Person build() {
+            return new Person(name, age, address);
+        }
+    }
 }
diff --git a/src/test/java/example/configuration/PersonColumnMetaData.java b/src/test/java/example/configuration/PersonColumnMetaData.java
@@ -16,9 +16,21 @@ public class PersonColumnMetaData {
     public static final List<CellMetaData> META_DATA_LIST = new ArrayList<>();
 
     static {
-        META_DATA_LIST.add(CellMetaData.builder().name("Name").dataType(DataType.STRING).build());
-        META_DATA_LIST.add(CellMetaData.builder().name("Age").dataType(DataType.NUMBER).build());
-        META_DATA_LIST.add(CellMetaData.builder().name("Address").dataType(DataType.STRING).build());
+        META_DATA_LIST.add(CellMetaData.builder()
+                .name("Name")
+                .dataType(DataType.STRING)
+                .fieldName("name")
+                .build());
+        META_DATA_LIST.add(CellMetaData.builder()
+                .name("Age")
+                .dataType(DataType.NUMBER)
+                .fieldName("age")
+                .build());
+        META_DATA_LIST.add(CellMetaData.builder()
+                .name("Address")
+                .dataType(DataType.STRING)
+                .fieldName("address")
+                .build());
     }
 
     public static Map<String, CellMetaData> getMetaDataMap() {

diff --git a/src/test/java/example/dto/FetchPersonRequest.java b/src/test/java/example/dto/FetchPersonRequest.java
@@ -0,0 +1,7 @@
+package example.dto;
+
+/**
+ * @author duyenthai
+ */
+public class FetchPersonRequest {
+}
diff --git a/src/test/java/example/service/FetchPersonService.java b/src/test/java/example/service/FetchPersonService.java
@@ -0,0 +1,34 @@
+package example.service;
+
+import com.vitadairy.libraries.importexport.dto.FetchRequest;
+import com.vitadairy.libraries.importexport.service.FetchDataService;
+import example.common.Person;
+import example.dto.FetchPersonRequest;
+
+import java.util.List;
+
+/**
+ * @author duyenthai
+ */
+public class FetchPersonService implements FetchDataService<Person, FetchPersonRequest> {
+    @Override
+    public List<Person> fetch(FetchRequest<FetchPersonRequest> request) {
+        return List.of(
+                new Person.Builder()
+                        .name("John")
+                        .age(20)
+                        .address("USA")
+                        .build(),
+                new Person.Builder()
+                        .name("Alice")
+                        .age(25)
+                        .address("UK")
+                        .build(),
+                new Person.Builder()
+                        .name("Bob")
+                        .age(30)
+                        .address("Canada")
+                        .build()
+        );
+    }
+}