Added abstraction level

Author: daneeldeveloper

src/main/java/org/fugerit/java/query/export/facade/QueryExportConfig.java

@@ -5,21 +5,28 @@
 
 public class QueryExportConfig {
 
+	public static final boolean DEFAULT_EXPORT_HEADER = true;
+	
 	public static QueryExportConfig newConfigCSV( OutputStream output, Connection conn, String query ) {
-		return new QueryExportConfig( QueryExportFacade.FORMAT_CSV , QueryExportFacade.CSV_SEPARATOR_DEF, output, conn, query );
+		return new QueryExportConfig( QueryExportFacade.FORMAT_CSV , QueryExportFacade.CSV_SEPARATOR_DEF, output, conn, query, DEFAULT_EXPORT_HEADER );
 	}
 
 	public static QueryExportConfig newConfigCSV( OutputStream output, Connection conn, String query, char separator ) {
-		return new QueryExportConfig( QueryExportFacade.FORMAT_CSV , separator, output, conn, query );
+		return new QueryExportConfig( QueryExportFacade.FORMAT_CSV , separator, output, conn, query, DEFAULT_EXPORT_HEADER );
 	}
 
 	public QueryExportConfig(String format, char separator, OutputStream output, Connection conn, String query) {
+		this( format, separator, output, conn, query, true );
+	}
+	
+	public QueryExportConfig(String format, char separator, OutputStream output, Connection conn, String query, boolean exportHeader) {
 		super();
 		this.format = format;
 		this.separator = separator;
 		this.output = output;
 		this.conn = conn;
 		this.query = query;
+		this.exportHeader = exportHeader;
 	}
 
 	private String format;
@@ -31,6 +38,8 @@ public QueryExportConfig(String format, char separator, OutputStream output, Con
 	private Connection conn;
 
 	private String query;
+	
+	private boolean exportHeader;
 
 	public String getFormat() {
 		return format;
@@ -69,5 +78,13 @@ public OutputStream getOutput() {
 	public void setOutput(OutputStream output) {
 		this.output = output;
 	}
+
+	public boolean isExportHeader() {
+		return exportHeader;
+	}
+
+	public void setExportHeader(boolean exportHeader) {
+		this.exportHeader = exportHeader;
+	}
 
 }

src/main/java/org/fugerit/java/query/export/facade/QueryExportFacade.java

@@ -6,7 +6,13 @@
 import java.sql.ResultSet;
 import java.sql.ResultSetMetaData;
 import java.sql.Statement;
+import java.util.Iterator;
 
+import org.fugerit.java.query.export.meta.BasicMetaRSE;
+import org.fugerit.java.query.export.meta.BasicMetaResult;
+import org.fugerit.java.query.export.meta.MetaField;
+import org.fugerit.java.query.export.meta.MetaRecord;
+import org.fugerit.java.query.export.meta.MetaResult;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -25,8 +31,27 @@ public static int export( QueryExportConfig config ) throws Exception {
 		Statement stm = config.getConn().createStatement();
 		logger.info( "sql : "+config.getQuery() );
 		ResultSet rs = stm.executeQuery( config.getQuery() );
-		int count = 0;
-		ResultSetMetaData rsmd = rs.getMetaData();
+		MetaResult meta = new BasicMetaResult( BasicMetaRSE.newInstanceAllToString( rs.getMetaData() ) , rs );
+		export( config, meta );
+		int count = meta.close();
+		stm.close();
+		logger.info( "record count "+count );
+		return res;
+	}
+	
+	private static void writeRecordCSV( String[] line, Iterator<MetaField> itField, CSVWriter csvwriter ) throws Exception {
+		int k = 0;
+		while ( itField.hasNext() ) {
+			MetaField field = itField.next();
+			line[k] = field.getStringValue();
+			k++;
+		}
+		csvwriter.writeNext( line );
+	}
+	
+	public static int export( QueryExportConfig config, MetaResult meta ) throws Exception {
+		int res = 0;
+		ResultSetMetaData rsmd = meta.getMetaData();
 		if ( FORMAT_CSV.equalsIgnoreCase( config.getFormat() ) ) {
 			char separator = config.getSeparator();
 			BufferedWriter writer =  new BufferedWriter( new OutputStreamWriter( config.getOutput() , Charset.forName( "UTF-8" ) ) );
@@ -36,23 +61,18 @@ public static int export( QueryExportConfig config ) throws Exception {
 					CSVWriter.NO_ESCAPE_CHARACTER,
 					CSVWriter.DEFAULT_LINE_END);
 			String[] line = new String[ rsmd.getColumnCount() ];
-			for ( int k=0; k<rsmd.getColumnCount(); k++ ) {
-				line[k] = rsmd.getColumnLabel( k+1 );
+			if ( meta.hasHeader() ) {
+				writeRecordCSV(line, meta.headerIterator(), csvwriter);	
 			}
-			csvwriter.writeNext( line );
-			while ( rs.next() ) {
-				for ( int k=0; k<rsmd.getColumnCount(); k++ ) {
-					line[k] = rs.getString( k+1 );
-				} 
-				csvwriter.writeNext( line );
-				count++;
+			Iterator<MetaRecord> itRec = meta.recordIterator();
+			while ( itRec.hasNext() ) {
+				MetaRecord record = itRec.next();
+				Iterator<MetaField> itField = record.fieldIterator();
+				writeRecordCSV(line, itField, csvwriter);
 			}
 			csvwriter.flush();
 			csvwriter.close();
 		}
-		rs.close();
-		stm.close();
-		logger.info( "record count "+count );
 		return res;
 	}
 

src/main/java/org/fugerit/java/query/export/meta/BasicMetaField.java

@@ -0,0 +1,20 @@
+package org.fugerit.java.query.export.meta;
+
+public class BasicMetaField implements MetaField {
+
+	public BasicMetaField(String stringValue) {
+		super();
+		this.stringValue = stringValue;
+	}
+
+	private String stringValue;
+
+	public String getStringValue() {
+		return stringValue;
+	}
+
+	public void setStringValue(String stringValue) {
+		this.stringValue = stringValue;
+	}
+
+}

src/main/java/org/fugerit/java/query/export/meta/BasicMetaRSE.java

@@ -0,0 +1,56 @@
+package org.fugerit.java.query.export.meta;
+
+import java.sql.ResultSet;
+import java.sql.ResultSetMetaData;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.fugerit.java.core.db.dao.RSExtractor;
+
+public abstract class BasicMetaRSE implements RSExtractor<MetaRecord> {
+
+	private ResultSetMetaData rsmd;
+	
+	@Override
+	public abstract MetaRecord extractNext(ResultSet rs) throws SQLException;
+
+	public void init( ResultSetMetaData rsmd ) throws Exception {
+		this.rsmd = rsmd;
+	}
+	
+	public void destroy( ) throws Exception {
+		this.rsmd = null;
+	}
+	
+	protected ResultSetMetaData getRsmd() {
+		return rsmd;
+	}
+
+	public static BasicMetaRSE newInstanceAllToString( ResultSetMetaData rsmd ) throws Exception {
+		BasicMetaRSEAllToString rse = new BasicMetaRSEAllToString();
+		rse.init( rsmd );
+		return rse;
+		
+	}
+	
+}
+
+class BasicMetaRSEAllToString extends BasicMetaRSE {
+	
+	@Override
+	public MetaRecord extractNext(ResultSet rs) throws SQLException {
+		List<MetaField> fields = new ArrayList<MetaField>();
+		for ( int k=0; k<this.getRsmd().getColumnCount(); k++ ) {
+			Object current = rs.getObject( k+1 );
+			String value = null;
+			if ( current != null ) {
+				value = String.valueOf( current );
+			}
+			MetaField field = new BasicMetaField( value );
+			fields.add( field );
+		}
+		return new BasicMetaRecord( fields );
+	}
+	
+}

src/main/java/org/fugerit/java/query/export/meta/BasicMetaRecord.java

@@ -0,0 +1,20 @@
+package org.fugerit.java.query.export.meta;
+
+import java.util.Collection;
+import java.util.Iterator;
+
+public class BasicMetaRecord implements MetaRecord {
+
+	private Collection<MetaField> fields;
+	
+	@Override
+	public Iterator<MetaField> fieldIterator() {
+		return this.fields.iterator();
+	}
+
+	public BasicMetaRecord(Collection<MetaField> fields) {
+		super();
+		this.fields = fields;
+	}
+
+}

src/main/java/org/fugerit/java/query/export/meta/BasicMetaResult.java

@@ -0,0 +1,80 @@
+package org.fugerit.java.query.export.meta;
+
+import java.sql.ResultSet;
+import java.sql.ResultSetMetaData;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+public class BasicMetaResult implements MetaResult {
+
+	public BasicMetaResult( BasicMetaRSE rse, ResultSet rs) {
+		super();
+		this.rse = rse;
+		this.rs = rs;
+	}
+
+	private BasicMetaRSE rse;
+	
+	private ResultSet rs;
+	
+	private int count;
+	
+	@Override
+	public boolean hasHeader() throws Exception {
+		return true;
+	}
+
+	@Override
+	public Iterator<MetaField> headerIterator() throws Exception {
+		List<MetaField> list = new ArrayList<MetaField>();
+		ResultSetMetaData rsmd = this.rs.getMetaData();
+		for ( int k=0; k<rsmd.getColumnCount(); k++ ) {
+			list.add( new BasicMetaField( rsmd.getColumnLabel( k+1 ) ) );
+		}
+		return list.iterator();
+	}
+
+	@Override
+	public ResultSetMetaData getMetaData() throws SQLException {
+		return this.rs.getMetaData();
+	}
+
+	@Override
+	public Iterator<MetaRecord> recordIterator() throws Exception {
+		return new Iterator<MetaRecord>() {
+			@Override
+			public boolean hasNext() {
+				boolean next = false;
+				try {
+					next = rs.next();
+				} catch (SQLException e) {
+					throw new RuntimeException( e );
+				}
+				return next;
+			}
+			@Override
+			public MetaRecord next() {
+				MetaRecord record = null;
+				try {
+					record = rse.extractNext( rs );
+					count++;
+				} catch (SQLException e) {
+					throw new RuntimeException( e );
+				}
+				return record;
+			}
+		};
+	}
+
+	@Override
+	public int close() throws Exception {
+		this.rs.close();
+		this.rs = null;
+		this.rse.destroy();
+		this.rse = null;
+		return this.count;
+	}
+	
+}

src/main/java/org/fugerit/java/query/export/meta/MetaField.java

@@ -0,0 +1,7 @@
+package org.fugerit.java.query.export.meta;
+
+public interface MetaField {
+
+	public String getStringValue();
+	
+}

src/main/java/org/fugerit/java/query/export/meta/MetaRecord.java

@@ -0,0 +1,10 @@
+package org.fugerit.java.query.export.meta;
+
+import java.util.Iterator;
+
+public interface MetaRecord {
+
+	public Iterator<MetaField> fieldIterator();
+
+	
+}

src/main/java/org/fugerit/java/query/export/meta/MetaResult.java

@@ -0,0 +1,19 @@
+package org.fugerit.java.query.export.meta;
+
+import java.sql.ResultSetMetaData;
+import java.sql.SQLException;
+import java.util.Iterator;
+
+public interface MetaResult {
+	
+	public Iterator<MetaRecord> recordIterator() throws Exception;
+
+	public ResultSetMetaData getMetaData() throws SQLException;
+	
+	public boolean hasHeader() throws Exception;
+	
+	public Iterator<MetaField> headerIterator() throws Exception;
+	
+	public int close() throws Exception;
+	
+}