org.jsoftware.impl.ResultDisplay Maven / Gradle / Ivy

Go to download
package org.jsoftware.impl;

import org.jsoftware.config.AbstractPatch;
import org.jsoftware.config.Patch;
import org.jsoftware.config.RollbackPatch;
import org.jsoftware.config.dialect.PatchExecutionResult;
import org.jsoftware.impl.extension.Extension;

import javax.swing.*;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLWarning;

/**
 * Bottom bar with patching results.
 * @author szalik
 */
public class ResultDisplay extends JEditorPane implements Extension {
	private static final long serialVersionUID = -6937849129583028112L;
	private final StringBuilder buf;
	
	public ResultDisplay() {
		super("text/html", null);
		setEditable(false);
		setAutoscrolls(true);
		buf = new StringBuilder("");
	}
	
	// DbPatch Extension
	public void afterPatching(Connection connection) {	}
	public void beforePatch(Connection connection, Patch patch) {
		addInfo("com.patching.start", new Object[] {patch.getName(), patch.getFile().getName()});	
	}
	public void beforePatching(Connection connection) {	}
	public void afterPatch(Connection connection, Patch patch, Exception ex) {
		if (ex == null) {
			addInfo("com.patching.done.ok", new Object[] {patch.getName(), patch.getFile().getName()});
		} else {
			addText(Messages.msg("com.patching.done.error", patch.getName(), patch.getFile().getName()), "#ff0000");
		}
		addText("
", null);
	}

    @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 {
        if (ex == null) {
            addInfo("com.rollbacking.done.ok", new Object[] {patch.getName(), patch.getFile().getName()});
        } else {
            addText(Messages.msg("com.rollbacking.done.error", patch.getName(), patch.getFile().getName()), "#ff0000");
        }
        addText("
", null);

    }

    public void afterPatchStatement(Connection connection, Patch patch, PatchExecutionResult result) {
		addText(result.getPatchStatement().getCode() + "
", null);
		if (result.isFailure()) {
			addText("" + result.getCause().getLocalizedMessage() + "", "#ff0000");
			addText("
", null);
		} else {
			SQLWarning warn = result.getSqlWarnings();
			while(warn != null) {
				addText(warn.getLocalizedMessage(), "#f4850d");
				addText("
", null);
				warn = warn.getNextWarning();
			}
			if (result.getDmlType() != null) {
				addText(Messages.msg("com.patch." + result.getDmlType().name().toLowerCase(), result.getDmlCount()), "#18f40d");
			} else {
				addText(Messages.msg("com.patch.otherType"), "#18f40d");
			}
			addText("
", null);
		}
	}
	
	private synchronized void addText(String text, String color) {
		text = text.trim();
		if (text.length() > 0) {
			if (color != null) {
				buf.append("");
				buf.append(text);
				buf.append("");
			} else {
				buf.append(text);
			}
			setText(buf.toString() + "");
		}
	}
	
	public void addInfo(String key, Object[] args) {
		addText(Messages.msg(key, args), "#130df4");
		addText("
", null);
	}
}