quangdn

pom.xml

@@ -189,6 +189,18 @@
             <artifactId>modelmapper</artifactId>
             <version>2.3.2</version>
         </dependency>
+        <!-- https://mvnrepository.com/artifact/org.apache.poi/poi -->
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi</artifactId>
+            <version>3.17</version>
+        </dependency>
+        <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi-ooxml</artifactId>
+            <version>3.17</version>
+        </dependency>
     </dependencies>
 
     <dependencyManagement>

src/main/java/com/viettel/campaign/web/rest/controller/CustomerController.java

@@ -7,13 +7,24 @@ import com.viettel.campaign.service.CustomerService;
 import com.viettel.campaign.web.dto.request_dto.SearchCustomerRequestDTO;
 import com.viettel.campaign.web.dto.request_dto.CustomerRequestDTO;
 import org.apache.log4j.Logger;
+import org.apache.poi.xssf.usermodel.XSSFRow;
+import org.apache.poi.xssf.usermodel.XSSFSheet;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
+import org.springframework.util.ResourceUtils;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.validation.Valid;
+import java.io.File;
+import java.nio.file.Files;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Map;
 
 @Controller
@@ -159,4 +170,60 @@ public class CustomerController {
         ResultDTO result = customerService.getCustomerContact(customerId, contactType, contact);
         return new ResponseEntity(result, HttpStatus.OK);
     }
+
+    @GetMapping(value = "/downloadFileTemplate")
+    public ResponseEntity<byte[]> downloadFileTemplate() {
+        LOGGER.debug("--------DOWNLOAD FILE TEMPLATE---------");
+        try {
+            File file = ResourceUtils.getFile("classpath:templates/test_download.xlsx");
+            byte[] content = Files.readAllBytes(file.toPath());
+            return ResponseEntity.ok()
+                    .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName())
+                    .contentType(MediaType.APPLICATION_OCTET_STREAM)
+                    .body(content);
+        } catch (Exception e) {
+            LOGGER.error(e.getMessage());
+            return new ResponseEntity<>(HttpStatus.BAD_REQUEST);
+        }
+    }
+
+    @PostMapping(value = "/importFile")
+    public ResponseEntity<byte[]> importFile(@RequestParam("file") MultipartFile file) {
+        LOGGER.debug("--------IMPORT FILE TEMPLATE---------");
+        try {
+            List<CustomerDTO> listCustomer = new ArrayList<>();
+            XSSFWorkbook workbook = new XSSFWorkbook(file.getInputStream());
+            XSSFSheet sheet = workbook.getSheetAt(0);
+            for (int i = 0; i < sheet.getPhysicalNumberOfRows(); i++) {
+                CustomerDTO customer = new CustomerDTO();
+                XSSFRow row = sheet.getRow(i);
+                customer.setCustomerId(Double.valueOf(row.getCell(0).getNumericCellValue()).longValue());
+                customer.setCreateDate(row.getCell(1).getDateCellValue());
+                customer.setName(row.getCell(2).getStringCellValue());
+                listCustomer.add(customer);
+            }
+//            for (int i = 0; i < listCustomer.size(); i++) {
+//                validate du lieu
+//            }
+//            if (okay) {
+//                for (int i = 0; i < listCustomer.size(); i++) {
+//                    customerService.createCustomer(listCustomer.get(i));
+//                }
+//                return ResponseEntity.ok()
+//                        .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName())
+//                        .contentType(MediaType.APPLICATION_OCTET_STREAM)
+//                        .body(file.getBytes());
+//            } else {
+//                return ResponseEntity.ok()
+//                        .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=" + file.getName())
+//                        .header("Message", "Validate failed!")
+//                        .contentType(MediaType.APPLICATION_OCTET_STREAM)
+//                        .body(file.getBytes());
+//            }
+            return null;
+        } catch (Exception e) {
+            LOGGER.error(e.getMessage());
+            return new ResponseEntity<>(HttpStatus.BAD_REQUEST);
+        }
+    }
 }