org.apache.torque.mojo.IdentifyInvalidDataFiles Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of maven-impex-plugin Show documentation
Maven plugin for converting database agnostic XML files into platform specific SQL files and for examining proprietary databases via JDBC to generate database agnostic XML files
There is a newer version: 1.1.13
Show newest version
package org.apache.torque.mojo;

import java.io.File;
import java.util.List;
import java.util.Properties;
import java.util.Set;
import java.util.TreeSet;

import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.tools.ant.DirectoryScanner;
import org.apache.torque.engine.database.model.Database;
import org.apache.torque.engine.database.model.Table;
import org.kuali.core.db.torque.SetUtils;
import org.kuali.core.db.torque.Utils;

/**
 * This mojo identifies data files that are present on the file system but are not present in schema.xml. This can
 * happen if a table is removed from the schema.
 * 
 * It sets a project property called "impex.data.invalid". This property is a comma delimited list of filenames that
 * have no match in the db schema.
 * 
 * @goal id-invalid-data-files
 */
public class IdentifyInvalidDataFiles extends BaseMojo {
	private static final String FS = System.getProperty("file.separator");

	/**
	 * @parameter expression="${extension}" default-value=".xml"
	 * @required
	 */
	private String extension;

	/**
	 * @parameter expression="${dataDir}" default-value="${project.basedir}/src/main/impex"
	 * @required
	 */
	private File dataDir;

	/**
	 * @parameter expression="${dataDirIncludes}" default-value="*.xml"
	 */
	private String dataDirIncludes;

	/**
	 * @parameter expression="${dataDirExcludes}" default-value="schema.xml"
	 */
	private String dataDirExcludes;

	/**
	 * @parameter expression="${schemaXMLFile}" default-value="src/main/impex/schema.xml"
	 */
	private String schemaXMLFile;

	/**
	 * @parameter expression="${targetDatabase}" default-value="oracle"
	 */
	private String targetDatabase;

	@Override
	protected void executeMojo() throws MojoExecutionException, MojoFailureException {
		Utils utils = new Utils();
		try {
			getLog().info("Examining " + dataDir.getAbsolutePath());
			Database db = utils.getDatabase(schemaXMLFile, targetDatabase);
			DirectoryScanner ds = getDirectoryScanner();
			Set existing = getExistingFiles(ds);
			Set allowed = getDatabaseFiles(db);
			Set invalid = SetUtils.difference(existing, allowed);
			getLog().info(existing.size() + " data files currently exist");
			getLog().info(invalid.size() + " of those are invalid");
			StringBuilder sb = new StringBuilder();
			int count = 0;
			for (File file : invalid) {
				if (count != 0) {
					sb.append(",");
				}
				sb.append("**/src/main/impex/" + file.getName());
				getLog().info("Marked for removal: " + file.getName());
				count++;
			}
			Properties properties = getProject().getProperties();
			properties.setProperty("impex.data.invalid", sb.toString());
		} catch (Exception e) {
			throw new MojoExecutionException("Error executing mojo", e);
		}
	}

	protected Set getDatabaseFiles(Database db) {
		List tables = db.getTables();
		Set files = new TreeSet();
		for (Object object : tables) {
			Table table = (Table) object;
			String tableName = table.getName();
			String filename = dataDir.getAbsolutePath() + FS + tableName + extension;
			File file = new File(filename);
			files.add(file);
		}
		return files;
	}

	protected Set getExistingFiles(DirectoryScanner ds) {
		ds.scan();
		String[] relativeFilenames = ds.getIncludedFiles();
		Set files = new TreeSet();
		for (int i = 0; i < relativeFilenames.length; i++) {
			String filename = ds.getBasedir().getAbsolutePath() + FS + relativeFilenames[i];
			File file = new File(filename);
			files.add(file);
		}
		return files;
	}

	protected DirectoryScanner getDirectoryScanner() {
		DirectoryScanner ds = new DirectoryScanner();
		ds.setBasedir(dataDir);
		ds.setIncludes(new String[] { dataDirIncludes });
		ds.setExcludes(new String[] { dataDirExcludes });
		return ds;
	}

	protected File getFile(Table table) {
		String tableName = table.getName();
		String filename = dataDir.getAbsolutePath() + FS + tableName + extension;
		File file = new File(filename);
		return file;
	}

	public File getDataDir() {
		return dataDir;
	}

	public void setDataDir(File dataDir) {
		this.dataDir = dataDir;
	}

	public String getDataDirIncludes() {
		return dataDirIncludes;
	}

	public void setDataDirIncludes(String dataDirIncludes) {
		this.dataDirIncludes = dataDirIncludes;
	}

	public String getDataDirExcludes() {
		return dataDirExcludes;
	}

	public void setDataDirExcludes(String dataDirExcludes) {
		this.dataDirExcludes = dataDirExcludes;
	}

	public String getSchemaXMLFile() {
		return schemaXMLFile;
	}

	public void setSchemaXMLFile(String schemaXMLFile) {
		this.schemaXMLFile = schemaXMLFile;
	}

	public String getExtension() {
		return extension;
	}

	public void setExtension(String extension) {
		this.extension = extension;
	}

	public String getTargetDatabase() {
		return targetDatabase;
	}

	public void setTargetDatabase(String targetDatabase) {
		this.targetDatabase = targetDatabase;
	}

}