net.sf.hajdbc.sql.AbstractDatabase Maven / Gradle / Ivy
/*
* HA-JDBC: High-Availability JDBC
* Copyright (C) 2012 Paul Ferraro
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see
*/
@XmlType(propOrder = { "user", "password", "xmlProperties" })
public abstract class AbstractDatabase implements Database
{
static final Logger logger = LoggerFactory.getLogger(AbstractDatabase.class);
@XmlAttribute(name = "id", required = true)
private String id;
@XmlAttribute(name = "location", required = true)
private String location;
@XmlElement(name = "user")
private String user;
@XmlElement(name = "password")
private String password;
@XmlAttribute(name = "weight")
private volatile Integer weight = 1;
@XmlAttribute(name = "local")
private Boolean local = false;
private Map properties = new HashMap();
private boolean dirty = false;
private volatile boolean active = false;
private volatile Connection detectConnection;
private volatile String ip=null;
@XmlElement(name = "property")
private Property[] getXmlProperties()
{
List properties = new ArrayList(this.properties.size());
for (Map.Entry entry: this.properties.entrySet())
{
Property property = new Property();
property.setName(entry.getKey());
property.setValue(entry.getValue());
properties.add(property);
}
return properties.toArray(new Property[properties.size()]);
}
@SuppressWarnings("unused")
private void setXmlProperties(Property[] properties)
{
for (Property property: properties)
{
this.properties.put(property.getName(), property.getValue());
}
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#getId()
*/
@ManagedAttribute
@Description("Uniquely identifies this database in the cluster")
@Override
public String getId()
{
return this.id;
}
public void setId(String id)
{
if (id.length() > ID_MAX_SIZE)
{
throw new IllegalArgumentException(String.format("Must be less than %d", ID_MAX_SIZE));
}
this.id = id;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#getLocation()
*/
@ManagedAttribute
@Description("Identifies the location of this database")
@Override
public String getLocation()
{
if(this.isLocal()){
String url = getLocalUrl(this.location);
return url;
}
return this.location;
}
@Description("type for this database")
@Override
public String getDbType(){
return getDbType(this.location);
}
private String getDbType(String url){
String type= "h2";
if((url!=null)&&(url.length()>5)
&&(url.substring(0,5).equalsIgnoreCase("jdbc:"))){
int index = url.indexOf(":", 5);
if(index>0){
type = url.substring(5,index).toLowerCase();
}
}
return type;
}
public static String getLocalUrl(String location) {
int index = location.indexOf("://")+3;
int end = location.indexOf(":", index);
if(end<0){
end = location.indexOf("/", index);
}
StringBuilder urlBuilder = new StringBuilder();
urlBuilder.append(location.substring(0, index));
urlBuilder.append("127.0.0.1");
urlBuilder.append(location.substring(end));
String url = urlBuilder.toString();
return url;
}
@ManagedAttribute
public void setLocation(String location)
{
this.assertInactive();
this.checkDirty(this.location, location);
this.location = location;
}
@ManagedAttribute
@Description("User ID for administrative connection authentication")
public String getUser()
{
return this.user;
}
@ManagedAttribute
public void setUser(String user)
{
this.assertInactive();
this.checkDirty(this.user, user);
this.user = user;
}
@ManagedAttribute
@Description("Password for administrative connection authentication")
public String getPassword()
{
return this.password;
}
@ManagedAttribute
public void setPassword(String password)
{
this.assertInactive();
this.checkDirty(this.password, password);
this.password = password;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#decodePassword(net.sf.hajdbc.codec.Decoder)
*/
@Override
public String decodePassword(Decoder decoder) throws SQLException
{
return (this.password != null) ? decoder.decode(this.password) : null;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#getWeight()
*/
@ManagedAttribute
@Description("Weight used in read request balancing")
@Override
public int getWeight()
{
return this.weight;
}
@ManagedAttribute
public void setWeight(int weight)
{
if (weight < 0)
{
throw new IllegalArgumentException();
}
this.checkDirty(this.weight, weight);
this.weight = weight;
}
/**
* @see java.lang.Object#hashCode()
*/
@Override
public int hashCode()
{
return this.id.hashCode();
}
/**
* @see java.lang.Object#equals(java.lang.Object)
*/
@Override
public boolean equals(Object object)
{
if ((object == null) || !(object instanceof Database)) {
return false;
}
String id = ((Database) object).getId();
return (id != null) && id.equals(this.id);
}
/**
* @see java.lang.Object#toString()
*/
@Override
public String toString()
{
return this.id;
}
/**
* @see java.lang.Comparable#compareTo(Object)
*/
@Override
public int compareTo(Database database)
{
return this.id.compareTo(database.getId());
}
@ManagedAttribute
@Description("Connection properties")
public Map getProperties()
{
return this.properties;
}
@ManagedOperation
@Description("Removes the specified connection property")
public void removeProperty(String name)
{
this.assertInactive();
String value = this.properties.remove(name);
this.dirty |= (value != null);
}
@ManagedOperation
@Description("Creates/updates the specified connection property")
public void setProperty(String name, String value)
{
this.assertInactive();
if ((name == null) || (value == null))
{
throw new IllegalArgumentException();
}
String old = this.properties.put(name, value);
this.checkDirty(old, value);
}
@ManagedAttribute
public void setLocal(boolean local)
{
this.assertInactive();
this.checkDirty(this.local, local);
this.local = local;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#isLocal()
*/
@ManagedAttribute
@Description("Indicates whether this database is local to this JVM")
@Override
public boolean isLocal()
{
return this.local;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#clean()
*/
@Override
public void clean()
{
this.dirty = false;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#isDirty()
*/
@Override
public boolean isDirty()
{
return this.dirty;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#isActive()
*/
@ManagedAttribute
@Description("Indicates whether or not this database is active")
@Override
public boolean isActive()
{
return this.active;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.Database#setActive(boolean)
*/
@Override
public void setActive(boolean active)
{
this.active = active;
if(!active&&detectConnection!=null){
Resources.close(detectConnection);
detectConnection = null;
}
}
@Override
public String getIp() {
return ip;
}
public void setIp(String ip) {
this.ip = ip;
}
/**
* Set the dirty flag if the new value differs from the old value.
* @param oldValue
* @param newValue
*/
protected void checkDirty(Object oldValue, Object newValue)
{
this.dirty |= ((oldValue != null) && (newValue != null)) ? !oldValue.equals(newValue) : (oldValue != newValue);
}
/**
* Helper method to determine whether the connect() method requires authentication.
* @return true, if authentication is required, false otherwise
*/
protected boolean requiresAuthentication()
{
return this.user != null;
}
protected void assertInactive()
{
if (this.active)
{
throw new IllegalStateException();
}
}
public Connection getDetectConnection(Decoder decoder) throws SQLException{
if(detectConnection==null){
detectConnection = this.connect(this.getConnectionSource(),this.decodePassword(decoder));
}
return detectConnection;
}
}