package org.hibernate.tool.schema.internal.exec;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.URL;
import org.hibernate.tool.schema.spi.SchemaManagementException;
import org.hibernate.tool.schema.spi.ScriptSourceInput;
import org.jboss.logging.Logger;
public class ScriptSourceInputFromUrl extends AbstractScriptSourceInput implements ScriptSourceInput {
private static final Logger log = Logger.getLogger( ScriptSourceInputFromFile.class );
private final URL url;
private final String charsetName;
private Reader reader;
public ScriptSourceInputFromUrl(URL url, String charsetName) {
this.url = url;
this.charsetName = charsetName;
}
@Override
protected Reader reader() {
if ( reader == null ) {
throw new SchemaManagementException( "Illegal state - reader is null - not prepared" );
}
return reader;
}
@Override
public void prepare() {
super.prepare();
try {
this.reader = charsetName != null ?
new InputStreamReader( url.openStream(), charsetName ) :
new InputStreamReader( url.openStream() );
}
catch (IOException e) {
throw new SchemaManagementException(
"Unable to open specified script source url [" + url + "] for reading"
);
}
}
@Override
public void release() {
try {
reader().close();
}
catch (IOException e) {
log.warn( "Unable to close file reader for generation script source" );
}
}
@Override
public String toString() {
return "ScriptSourceInputFromUrl(" + url.toExternalForm() + ")";
}
}