Merge pull request #5 from anicetkeric/dev

complete code

Author: anicetkeric

src/main/java/com/springbatch/excel/tutorial/batch/BatchConfiguration.java

@@ -68,7 +68,7 @@ public MongoItemWriter<Employee> writer(MongoTemplate mongoTemplate) {
 
 
     /**
-     * Declaration step
+     * step declaration
      * @return {@link Step}
      */
     @Bean
@@ -84,7 +84,7 @@ public Step employeeStep(MongoItemWriter<Employee> itemWriter) {
 
 
     /**
-     * Declaration job
+     * job declaration
      * @param listener {@link JobCompletionListener}
      * @return {@link Job}
      */

src/main/java/com/springbatch/excel/tutorial/batch/EmployeeJobLauncher.java

@@ -28,18 +28,19 @@ public class EmployeeJobLauncher {
 
     private final JobLauncher jobLauncher;
 
-    @Value("${employee.excel.path}")
-    private String excelFilePath;
+    @Value("${employee.excel.processingfolder}")
+    private String processingDir;
 
     EmployeeJobLauncher(Job job, JobLauncher jobLauncher) {
         this.job = job;
         this.jobLauncher = jobLauncher;
     }
 
-    //@Scheduled(cron = "*/2 * * * *")
+    // run every 2 min
     @Scheduled(fixedRate = 120000)
     void launchFileToJob() throws JobParametersInvalidException, JobExecutionAlreadyRunningException, JobInstanceAlreadyCompleteException, JobRestartException {
         LOGGER.info("Starting job");
+        String excelFilePath = String.format("%s/employee.xlsx", processingDir);
 
         JobParameters params = new JobParametersBuilder()
                 .addLong("jobId",System.currentTimeMillis())

src/main/java/com/springbatch/excel/tutorial/batch/listeners/JobCompletionListener.java

@@ -18,6 +18,7 @@ public class JobCompletionListener extends JobExecutionListenerSupport {
     public void afterJob(JobExecution jobExecution) {
 
         String jobId = jobExecution.getJobParameters().getString("jobId");
+        String excelFilePath = jobExecution.getJobParameters().getString("excelPath");
 
          // get job's start time
         Date start = jobExecution.getCreateTime();
@@ -26,11 +27,12 @@ public void afterJob(JobExecution jobExecution) {
 
         if(jobExecution.getStatus() == BatchStatus.COMPLETED) {
 
-            LOGGER.info("===========================JOB FINISHED================================================");
+            LOGGER.info("==========JOB FINISHED=======");
             LOGGER.info("JobId      : {}",jobId);
+            LOGGER.info("excel Path      : {}",excelFilePath);
             LOGGER.info("Start Date: {}", start);
             LOGGER.info("End Date: {}", end);
-            LOGGER.info("=======================================================================================");
+            LOGGER.info("==============================");
         }
 
     }

src/main/java/com/springbatch/excel/tutorial/batch/processors/EmployeeItemProcessor.java

@@ -1,31 +1,124 @@
 package com.springbatch.excel.tutorial.batch.processors;
 
 import com.springbatch.excel.tutorial.domain.Employee;
+import com.springbatch.excel.tutorial.repository.EmployeeRepository;
+import com.springbatch.excel.tutorial.support.poi.AbstractExcelPoi;
+import lombok.SneakyThrows;
+import org.apache.poi.ss.usermodel.*;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.batch.core.BatchStatus;
+import org.springframework.batch.core.ExitStatus;
+import org.springframework.batch.core.StepExecution;
+import org.springframework.batch.core.StepExecutionListener;
 import org.springframework.batch.item.ItemProcessor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.util.CollectionUtils;
+
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
 
 
 /**
  * @author Eric KOUAME
  */
-public class EmployeeItemProcessor implements ItemProcessor<Employee, Employee> {
+public class EmployeeItemProcessor extends AbstractExcelPoi<Employee>  implements ItemProcessor<Employee, Employee>, StepExecutionListener {
 
     private static final Logger LOGGER = LoggerFactory.getLogger(EmployeeItemProcessor.class);
 
+    private static final List<Employee> existingEmployees = new ArrayList<>();
 
-    public EmployeeItemProcessor() {
-        super();
-    }
+    @Value("${employee.excel.resultsfolder}")
+    private String resultsfolder;
+
+    @Autowired
+    EmployeeRepository repository;
 
     /**
      * {@inheritDoc}
      */
     @Override
     public Employee process(Employee item) {
 
+        if(repository.findByNumber(item.getNumber()).isPresent()){
+            existingEmployees.add(item);
+
+            return null;
+        }
+
         return item;
     }
 
+    @Override
+    public void beforeStep(StepExecution stepExecution) {
+        /* Nothing to do before */
+    }
 
+    @SneakyThrows
+    @Override
+    public ExitStatus afterStep(StepExecution stepExecution) {
+        String jobId = stepExecution.getJobParameters().getString("jobId");
+
+        if(stepExecution.getStatus() == BatchStatus.COMPLETED && !CollectionUtils.isEmpty(existingEmployees)) {
+            // create log file. if completed
+            String path = new ClassPathResource(resultsfolder).getFile().getPath();
+            write(String.format("%s/employe-result-%s.xlsx", path, jobId),existingEmployees);
+        }
+        return null;
+    }
+    /**
+     * Create and write results file. employee duplicate entries
+     * @param filePath file path
+     * @param aList    list of employee
+     */
+    @Override
+    public void write(String filePath , List<Employee> aList) {
+        try (XSSFWorkbook workbook = new XSSFWorkbook()) {
+            // Create a Sheet
+            Sheet sheet = workbook.createSheet("RESULTS");
+
+            // creation header
+            CellStyle cellStyle = sheet.getWorkbook().createCellStyle();
+            Font font = sheet.getWorkbook().createFont();
+            font.setBold(true);
+            font.setFontHeightInPoints((short) 12);
+            cellStyle.setFont(font);
+
+            cellStyle.setAlignment(HorizontalAlignment.CENTER);
+            List<String> headers = Arrays.asList("FirstName",
+                    "LastName",
+                    "Email",
+                    "Number"
+            );
+            createHeaderRow(sheet, headers,cellStyle);
+
+            // Create Other rows and cells with BeneficiaryExtraction data
+            int rowNum = 1;
+            for(Employee employee: aList) {
+                Row row = sheet.createRow(rowNum++);
+
+                createCell(row, 0, employee.getFirstName(), null);
+                createCell(row, 1, employee.getLastName(), null);
+                createCell(row, 2, employee.getEmail(), null);
+                createCell(row, 3, employee.getNumber(), null);
+            }
+
+            //lets write the excel data to file now
+            try (FileOutputStream outputStream = new FileOutputStream(filePath)) {
+                workbook.write(outputStream);
+                LOGGER.info("Results File written successfully");
+            }
+        }
+        catch (IOException e) {
+            LOGGER.warn("Cannot write results file: {}", e.getMessage());
+        }finally {
+            existingEmployees.clear();
+        }
+    }
 }

src/main/java/com/springbatch/excel/tutorial/repository/EmployeeRepository.java

@@ -4,6 +4,10 @@
 import org.springframework.data.mongodb.repository.MongoRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.Optional;
+
 @Repository
 public interface EmployeeRepository extends MongoRepository<Employee, String> {
+
+    Optional<Employee> findByNumber(String number);
 }

src/main/java/com/springbatch/excel/tutorial/service/EmployeeService.java

@@ -4,15 +4,13 @@
 
 import com.springbatch.excel.tutorial.domain.Employee;
 
-import java.util.List;
 import java.util.Optional;
 
 public interface EmployeeService {
-    List<Employee> getAll();
 
-    Employee save(Employee employee);
-
-    Optional<Employee> getById(String id);
-
-    void deleteById(String id);
+    /**
+     * @param number employee nuumber
+     * @return Optional employee
+     */
+    Optional<Employee> getByNumber(String number);
 }

src/main/java/com/springbatch/excel/tutorial/service/EmployeeServiceImpl.java

@@ -2,38 +2,24 @@
 
 import com.springbatch.excel.tutorial.domain.Employee;
 import com.springbatch.excel.tutorial.repository.EmployeeRepository;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
-import java.util.List;
 import java.util.Optional;
 
 @Service
 public class EmployeeServiceImpl implements EmployeeService {
 
-    @Autowired
-    private EmployeeRepository employeeRepository;
+    private final EmployeeRepository employeeRepository;
 
-    @Override
-    public List<Employee> getAll() {
-        return employeeRepository.findAll();
-    }
-
-    @Override
-    public Employee save(Employee employee) {
-        return this.employeeRepository.save(employee);
-    }
-
-    @Override
-    public Optional<Employee> getById(String id) {
-        return employeeRepository.findById(id);
+    public EmployeeServiceImpl(EmployeeRepository employeeRepository) {
+        this.employeeRepository = employeeRepository;
     }
 
+    /**
+     * {@inheritDoc}
+     */
     @Override
-    public void deleteById(String id) {
-        this.employeeRepository.deleteById(id);
+    public Optional<Employee> getByNumber(String number) {
+        return employeeRepository.findByNumber(number);
     }
-
-
-
 }

src/main/java/com/springbatch/excel/tutorial/support/poi/AbstractExcelPoi.java

@@ -78,6 +78,8 @@ private Workbook getWorkbook(String filePathStr) {
     }
 
 
+
+
     /**
      * Create header of file
      * @param sheet excel sheet
@@ -89,18 +91,41 @@ public void createHeaderRow(Sheet sheet, List<String> headers, CellStyle rowStyl
         if(!headers.isEmpty()){
             // Create a Row
             Row row = sheet.createRow(0);
-
             // Create cells
             for (int i = 0; i < headers.size(); i++) {
-                Cell cellId = row.createCell(i);
-                if(rowStyle != null){
-                    cellId.setCellStyle(rowStyle);
-                }
-                cellId.setCellValue(headers.get(i));
-
+                createCell(row, i, headers.get(i), rowStyle);
                 sheet.autoSizeColumn(i);
             }
         }
 
     }
+
+    /**
+     * @param row file row
+     * @param columnCount index of column
+     * @param value the value to will set
+     * @param style cel row
+     * @return Cell value
+     */
+    public Cell createCell(Row row, int columnCount, Object value, CellStyle style) {
+        Cell cell = row.createCell(columnCount);
+        if (value instanceof Integer) {
+            cell.setCellValue((Integer) value);
+        } else if (value instanceof Boolean) {
+            cell.setCellValue((Boolean) value);
+        }else if (value instanceof Double) {
+            cell.setCellValue((Double) value);
+        } else {
+            cell.setCellValue((String) value);
+        }
+
+
+        if(style != null){
+            cell.setCellStyle(style);
+        }
+
+        return cell;
+    }
+
+
 }

src/main/resources/application.yml

@@ -21,5 +21,5 @@ logging:
 
 employee:
   excel:
-    processing-dir: data/processing/
-    results-dir: data/results/
+    processingfolder: data/processing/
+    resultsfolder: data/results/