org.jsoftware.impl.extension.TkExtensionAndStrategy Maven / Gradle / Ivy

Go to download
package org.jsoftware.impl.extension;

import org.jsoftware.config.AbstractPatch;
import org.jsoftware.config.ApplyStrategy;
import org.jsoftware.config.Patch;
import org.jsoftware.config.RollbackPatch;
import org.jsoftware.config.dialect.PatchExecutionResult;
import org.jsoftware.impl.PatchStatement;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

public class TkExtensionAndStrategy implements Extension, ApplyStrategy {
	
	private static int patchLevel(Patch p) {
		String str = p.getFile().getName();
		try {
			str = str.substring(5, 5+4).trim();
			return Integer.parseInt(str);
		} catch (Exception e) {
			throw new IllegalArgumentException("Can not determinate patchLeave for patch " + p.getName());
		}
	}
	
	public List filter(Connection con, List patches) {
		try {	
			String tabName = detectTkTableName(con);
			ResultSet rs = con.createStatement().executeQuery("SELECT patch_level FROM " + tabName);
			rs.next();
			int currentPatchLevel = rs.getInt(1);
			rs.close();
			LinkedList patchesToApply = new LinkedList();
			for(Patch p : patches) {
				if (patchLevel(p) > currentPatchLevel) {
					patchesToApply.add(p);
				}
			}
			return patchesToApply;
		} catch (Exception e) {
			throw new RuntimeException("Can not apply strategy.", e);
		}
	}

	private String detectTkTableName(Connection con) {
		String tabName = null;
		Set qm = new HashSet();
		try {
			ResultSet rs = con.getMetaData().getTables(null,null,null,null);
			while(rs.next()) {
				String tb = rs.getString(3);
				if ("patches".equalsIgnoreCase(tb) || "tk_patches".equalsIgnoreCase(tb)) qm.add(tb);
			}
			rs.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		if (qm.size() == 1) {
			tabName = qm.iterator().next();
		}
		if (tabName == null) throw new RuntimeException("No tk table available. " + qm);
		return tabName;
	}

	public void beforePatching(Connection connection) {
	}

	public void afterPatching(Connection connection) {
		// TODO release TK lock
	}

	public void beforePatch(Connection connection, Patch patch) {
	}

	public void afterPatch(Connection connection, Patch patch, Exception ex) throws SQLException {
		if (ex == null) {
			connection.createStatement().executeUpdate("UPDATE " + detectTkTableName(connection) + " SET patch_level=" + patchLevel(patch));
		}
	}

    @Override
    public void beforePatchStatement(Connection connection, AbstractPatch patch, PatchStatement statement) {

    }

    @Override
    public void afterPatchStatement(Connection connection, AbstractPatch patch, PatchExecutionResult result) {

    }

    @Override
    public void beforeRollbackPatch(Connection connection, RollbackPatch patch) {

    }

    @Override
    public void afterRollbackPatch(Connection connection, RollbackPatch patch, Exception ex) throws SQLException {

    }


}