com.github.sourcegroove.batch.item.file.excel.ExcelRowTokenizer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-batch-file-layout Show documentation

Layout driven spring batch item readers and writers

There is a newer version: 1.3.21

Show newest version

package com.github.sourcegroove.batch.item.file.excel;

import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.batch.item.file.transform.DefaultFieldSetFactory;
import org.springframework.batch.item.file.transform.FieldSet;
import org.springframework.batch.item.file.transform.FieldSetFactory;

import java.util.List;


public class ExcelRowTokenizer {
    protected final Log log = LogFactory.getLog(getClass());
    private FieldSetFactory fieldSetFactory = new DefaultFieldSetFactory();
    private String[] names;

    public void setNames(String[] names) {
        this.names = names;
    }

    public FieldSet tokenize(List row, int rowNumber){
        if(CollectionUtils.isNotEmpty(row) && CollectionUtils.size(row) == names.length - 1){
            //add that last column the mapper missed.... (or we could make this 'ignore' extra columns...
            row.add("");
        }

        String[] values = row.toArray(new String[row.size()]);
        if(values != null && names != null && names.length > values.length){
            throw new RuntimeException("Error tokenizing row: " + rowNumber
                    + " name count " + names.length
                    + " and field value count " + values.length
                    + "  don't match ");
        } else if (names == null){
            return this.fieldSetFactory.create(values);
        } else {
            return this.fieldSetFactory.create(values, names);
        }
    }


    private void logFields(List values, String[] names){
        for(String name : names){
            log.info("Name: " + name);
        }
        for(String value : values){
            log.info("Value: " + value);
        }
    }
}