de.akquinet.jbosscc.guttenbase.export.plain.PlainGzipImporter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of GuttenBase Show documentation
Copying done right...
There is a newer version: 2.1.0
package de.akquinet.jbosscc.guttenbase.export.plain;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.Serializable;
import java.util.Map;
import java.util.zip.GZIPInputStream;

import de.akquinet.jbosscc.guttenbase.export.ExportTableHeader;
import de.akquinet.jbosscc.guttenbase.export.ImportDumpConnectionInfo;
import de.akquinet.jbosscc.guttenbase.export.ImportDumpExtraInformation;
import de.akquinet.jbosscc.guttenbase.export.Importer;
import de.akquinet.jbosscc.guttenbase.meta.DatabaseMetaData;
import de.akquinet.jbosscc.guttenbase.meta.TableMetaData;
import de.akquinet.jbosscc.guttenbase.repository.ConnectorRepository;

/**
 * Import schema information and data from gzipped {@link ObjectInputStream} file with serialized data.
 * 
 * 
 * © 2012 akquinet tech@spree
 * 
 * 
 * @author M. Dahm
 */
public class PlainGzipImporter implements Importer {
	private ObjectInputStream _objectInputStream;
	private ConnectorRepository _connectorRepository;
	private String _connectorId;
	// Ensure that table data has been read when seeking the extra informations
	private boolean _tableDataRead = false;

	@Override
	public void initializeImport(final ConnectorRepository connectorRepository, final String connectorId,
			final ImportDumpConnectionInfo importDumpConnectionInfo) throws IOException {
		assert importDumpConnectionInfo != null : "importDumpConnectionInfo != null";
		assert connectorId != null : "connectorId != null";
		assert connectorRepository != null : "connectorRepository != null";

		_connectorRepository = connectorRepository;
		_connectorId = connectorId;
		final File file = new File(importDumpConnectionInfo.getPath());
		final FileInputStream fileInputStream = new FileInputStream(file);
		final GZIPInputStream gzipInputStream = new GZIPInputStream(fileInputStream);
		_objectInputStream = new ObjectInputStream(gzipInputStream);
	}

	@Override
	public void finishImport() throws Exception {
		if (_tableDataRead) { // At end
			readExtraInformation();
		}

		_objectInputStream.close();
		_objectInputStream = null;
	}

	@Override
	public DatabaseMetaData readDatabaseMetaData() throws Exception {
		return (DatabaseMetaData) _objectInputStream.readObject();
	}

	@Override
	public void seekTableHeader(final TableMetaData tableMetaData) throws Exception {
		_tableDataRead = true;
		ExportTableHeader exportTableHeader;

		do {
			exportTableHeader = seekNextTableHeader();
		} while (!tableMetaData.getTableName().equalsIgnoreCase(exportTableHeader.getTableName()));
	}

	@Override
	public Object readObject() throws Exception {
		return _objectInputStream.readObject();
	}

	private ExportTableHeader seekNextTableHeader() throws Exception {
		Object value;

		do {
			value = _objectInputStream.readObject();
		} while (!(value instanceof ExportTableHeader));

		return (ExportTableHeader) value;
	}

	private void readExtraInformation() throws Exception {
		final ImportDumpExtraInformation importDumpExtraInformation = _connectorRepository.getConnectorHint(_connectorId,
				ImportDumpExtraInformation.class).getValue();

		@SuppressWarnings("unchecked")
		final Map extraInformation = (Map) readObject();

		importDumpExtraInformation.processExtraInformation(extraInformation);
	}
}