org.fugerit.java.tool.fixed.ExtractFixedConfigHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fj-tool Show documentation
Fugerit Java Tool Library for JDK 1.8 and up
There is a newer version: 8.6.5
package org.fugerit.java.tool.fixed;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.fugerit.java.core.fixed.parser.FixedFieldDescriptor;
import org.fugerit.java.core.xml.TransformerXML;
import org.fugerit.java.tool.RunToolException;
import org.fugerit.java.tool.ToolHandlerHelper;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class ExtractFixedConfigHandler extends ToolHandlerHelper {

	/**
	 * Arg for input file
	 */
	public static final String PARAM_INPUT_FILE = "input-file";
	
	/**
	 * Arg output xml
	 */
	public static final String PARAM_OUTPUT_XML = "output-xml";
	
	/**
	 * Arg output bean
	 */
	public static final String PARAM_OUTPUT_BEAN = "output-bean";
	
	private int handleRow( int start, String length, String name, List listFields, int rowCount ) {
		Integer len = Integer.valueOf( length );
		FixedFieldDescriptor ffd = new FixedFieldDescriptor( name, start, len );
		logger.info( "field descriptor : {}, rowCount : {}", ffd, rowCount );
		listFields.add( ffd );
		return (start+len);
	}
	
	@Override
	public int handleWorker(Properties params) throws RunToolException {
		int exit = EXIT_OK;
		String inputFilePath = params.getProperty( PARAM_INPUT_FILE );
		File inputFile = new File( inputFilePath );
		try ( InputStream is = new FileInputStream( inputFile );
				XSSFWorkbook workbook = new XSSFWorkbook( is ) ) {
			Sheet sheet = workbook.getSheetAt( 0 );
			Iterator rows = sheet.rowIterator();
			List listFields = new ArrayList<>();
			int rowCount = 0;
			int start = 1;
			while ( rows.hasNext() ) {
				Row row = rows.next();
				String name = row.getCell( 0 ).getStringCellValue();
				Cell lengthCell = row.getCell( 1 );
				String length = null;
				CellType lengthCellType = lengthCell.getCellType();
				if ( CellType.STRING.equals(lengthCellType  ) ) {
					length = lengthCell.getStringCellValue();
				} else if ( CellType.NUMERIC.equals(lengthCellType  ) ) {
					length = String.valueOf( (long)lengthCell.getNumericCellValue() );
				} 
				log.debug( "Check length : {}", Integer.parseInt( length ) );
				start = this.handleRow(start, length, name, listFields, rowCount);
				rowCount++;
			}
			// generating xml output
			String outputXmlPath = params.getProperty( PARAM_OUTPUT_XML );
			if ( outputXmlPath != null ) {
				DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
				DocumentBuilder parser = dbf.newDocumentBuilder();
				Document doc = parser.newDocument();
				Element root = doc.createElement( "field-list" );
				Iterator itFields = listFields.iterator();
				while ( itFields.hasNext() ) {
					FixedFieldDescriptor ffd = itFields.next();
					Element fieldTag = doc.createElement( "field" );
					fieldTag.setAttribute( "id" , ffd.getNormalizedName() );
					fieldTag.setAttribute( "start" , String.valueOf( ffd.getStart() ) );
					fieldTag.setAttribute( "length" , String.valueOf( ffd.getLength() ) );
					fieldTag.setAttribute( "end" , String.valueOf( ffd.getEnd() ) );
					fieldTag.setAttribute( "description" , ffd.getName() );
					root.appendChild( fieldTag );
				}
				TransformerFactory tFactory = TransformerXML.newSafeTransformerFactory();
				Transformer transformer = tFactory.newTransformer();
				transformer.setOutputProperty( OutputKeys.INDENT, "yes" );
				transformer.setOutputProperty( OutputKeys.OMIT_XML_DECLARATION, "yes" );
				DOMSource source = new DOMSource( root );
				FileOutputStream fos = new FileOutputStream( new File( outputXmlPath ) );
				StreamResult result = new StreamResult( fos );
				transformer.transform(source, result);
				fos.close();
			}
		} catch (Exception e) {
			throw RunToolException.convertEx( e );
		}
		return exit;
	}

}