package play.modules.siena;
import java.io.File;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.DriverPropertyInfo;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.sql.DataSource;
import jregex.Matcher;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.dbcp.BasicDataSource;
import com.mchange.v2.c3p0.ComboPooledDataSource;
import com.mchange.v2.c3p0.DataSources;
import play.Logger;
import play.Play;
import play.PlayPlugin;
import play.db.DB;
import play.exceptions.DatabaseException;
import play.mvc.Http;
import play.mvc.Http.Request;
import play.mvc.Http.Response;
/**
* The DB plugin
*/
public class GoogleSqlDBPlugin extends PlayPlugin {
public static String url = "";
org.h2.tools.Server h2Server;
@Override
public boolean rawInvocation(Request request, Response response) throws Exception {
if (Play.mode.isDev() && request.path.equals("/@db")) {
response.status = Http.StatusCode.MOVED;
// For H2 embeded database, we'll also start the Web console
if (h2Server != null) {
h2Server.stop();
}
h2Server = org.h2.tools.Server.createWebServer();
h2Server.start();
response.setHeader("Location", "http://localhost:8082/");
return true;
}
return false;
}
@Override
public void onLoad() {
// if not in dev mode and in google mode, disables classic dbplugin
if (Play.mode.isProd() && SienaPlugin.dbType().contains("google")) {
// FIRST DISABLES DBPlugin
Play.pluginCollection.disablePlugin(play.db.DBPlugin.class);
}
}
@Override
public void onApplicationStart() {
if (Play.mode.isProd() && SienaPlugin.dbType().contains("google")) {
try {
Properties p = Play.configuration;
if (DB.datasource != null) {
DB.destroy();
}
if (p.getProperty("db", "").startsWith("java:")) {
Context ctx = new InitialContext();
DB.datasource = (DataSource) ctx.lookup(p.getProperty("db"));
} else {
// Try the driver
String driver = p.getProperty("db.driver");
try {
Driver d = (Driver) Class.forName(driver, true, Play.classloader).newInstance();
DriverManager.registerDriver(new ProxyDriver(d));
} catch (Exception e) {
throw new Exception("Driver not found (" + driver + ")");
}
// Try the connection
Connection fake = null;
try {
if (p.getProperty("db.user") == null) {
fake = DriverManager.getConnection(p.getProperty("db.url"));
} else {
fake = DriverManager.getConnection(p.getProperty("db.url"), p.getProperty("db.user"), p.getProperty("db.pass"));
}
} finally {
if (fake != null) {
fake.close();
}
}
// apache basic DS
BasicDataSource ds = new BasicDataSource();
ds.setDriverClassName(p.getProperty("db.driver"));
ds.setUsername(p.getProperty("db.user"));
ds.setPassword(p.getProperty("db.pass"));
ds.setUrl(p.getProperty("db.url"));
DB.datasource = ds;
url = ds.getUrl();
Connection c = null;
try {
c = ds.getConnection();
} finally {
if (c != null) {
c.close();
}
}
//Logger.info("Connected to %s", ds.getJdbcUrl());
Logger.info("Connected to %s", url);
}
DB.destroyMethod = p.getProperty("db.destroyMethod", "");
} catch (Exception e) {
DB.datasource = null;
Logger.error(e, "Cannot connected to the database : %s", e.getMessage());
if (e.getCause() instanceof InterruptedException) {
throw new DatabaseException("Cannot connected to the database. Check the configuration.", e);
}
throw new DatabaseException("Cannot connected to the database, " + e.getMessage(), e);
}
}
}
@Override
public void onApplicationStop() {
if (Play.mode.isProd()) {
DB.destroy();
}
}
@Override
public String getStatus() {
StringWriter sw = new StringWriter();
PrintWriter out = new PrintWriter(sw);
if (DB.datasource == null || !(DB.datasource instanceof ComboPooledDataSource)) {
out.println("Datasource:");
out.println("~~~~~~~~~~~");
out.println("(not yet connected)");
return sw.toString();
}
ComboPooledDataSource datasource = (ComboPooledDataSource) DB.datasource;
out.println("Datasource:");
out.println("~~~~~~~~~~~");
out.println("Jdbc url: " + datasource.getJdbcUrl());
out.println("Jdbc driver: " + datasource.getDriverClass());
out.println("Jdbc user: " + datasource.getUser());
out.println("Jdbc password: " + datasource.getPassword());
out.println("Min pool size: " + datasource.getMinPoolSize());
out.println("Max pool size: " + datasource.getMaxPoolSize());
out.println("Initial pool size: " + datasource.getInitialPoolSize());
out.println("Checkout timeout: " + datasource.getCheckoutTimeout());
return sw.toString();
}
@Override
public void invocationFinally() {
DB.close();
}
private static void check(Properties p, String mode, String property) {
if (!StringUtils.isEmpty(p.getProperty(property))) {
Logger.warn("Ignoring " + property + " because running the in " + mode + " db.");
}
}
private static boolean changed() {
Properties p = Play.configuration;
if ("mem".equals(p.getProperty("db")) && p.getProperty("db.url") == null) {
p.put("db.driver", "org.h2.Driver");
p.put("db.url", "jdbc:h2:mem:play;MODE=MYSQL");
p.put("db.user", "sa");
p.put("db.pass", "");
}
if ("fs".equals(p.getProperty("db")) && p.getProperty("db.url") == null) {
p.put("db.driver", "org.h2.Driver");
p.put("db.url", "jdbc:h2:" + (new File(Play.applicationPath, "db/h2/play").getAbsolutePath()) + ";MODE=MYSQL");
p.put("db.user", "sa");
p.put("db.pass", "");
}
if (p.getProperty("db", "").startsWith("java:") && p.getProperty("db.url") == null) {
if (DB.datasource == null) {
return true;
}
} else {
// Internal pool is c3p0, we should call the close() method to destroy it.
check(p, "internal pool", "db.destroyMethod");
p.put("db.destroyMethod", "close");
}
Matcher m = new jregex.Pattern("^mysql:(//)?(({user}[a-zA-Z0-9_]+)(:({pwd}[^@]+))?@)?(({host}[^/]+)/)?({name}[^\\s]+)$").matcher(p.getProperty("db", ""));
if (m.matches()) {
String user = m.group("user");
String password = m.group("pwd");
String name = m.group("name");
String host = m.group("host");
p.put("db.driver", "com.mysql.jdbc.Driver");
p.put("db.url", "jdbc:mysql://" + (host == null ? "localhost" : host) + "/" + name + "?useUnicode=yes&characterEncoding=UTF-8&connectionCollation=utf8_general_ci");
if (user != null) {
p.put("db.user", user);
}
if (password != null) {
p.put("db.pass", password);
}
}
m = new jregex.Pattern("^postgres:(//)?(({user}[a-zA-Z0-9_]+)(:({pwd}[^@]+))?@)?(({host}[^/]+)/)?({name}[^\\s]+)$").matcher(p.getProperty("db", ""));
if (m.matches()) {
String user = m.group("user");
String password = m.group("pwd");
String name = m.group("name");
String host = m.group("host");
p.put("db.driver", "org.postgresql.Driver");
p.put("db.url", "jdbc:postgresql://" + (host == null ? "localhost" : host) + "/" + name);
if (user != null) {
p.put("db.user", user);
}
if (password != null) {
p.put("db.pass", password);
}
}
if(p.getProperty("db.url") != null && p.getProperty("db.url").startsWith("jdbc:h2:mem:")) {
p.put("db.driver", "org.h2.Driver");
p.put("db.user", "sa");
p.put("db.pass", "");
}
if ((p.getProperty("db.driver") == null) || (p.getProperty("db.url") == null)) {
return false;
}
if (DB.datasource == null) {
return true;
} else {
ComboPooledDataSource ds = (ComboPooledDataSource) DB.datasource;
if (!p.getProperty("db.driver").equals(ds.getDriverClass())) {
return true;
}
if (!p.getProperty("db.url").equals(ds.getJdbcUrl())) {
return true;
}
if (!p.getProperty("db.user", "").equals(ds.getUser())) {
return true;
}
if (!p.getProperty("db.pass", "").equals(ds.getPassword())) {
return true;
}
}
if (!p.getProperty("db.destroyMethod", "").equals(DB.destroyMethod)) {
return true;
}
return false;
}
/**
* Needed because DriverManager will not load a driver ouside of the system classloader
*/
public static class ProxyDriver implements Driver {
private Driver driver;
ProxyDriver(Driver d) {
this.driver = d;
}
public boolean acceptsURL(String u) throws SQLException {
return this.driver.acceptsURL(u);
}
public Connection connect(String u, Properties p) throws SQLException {
return this.driver.connect(u, p);
}
public int getMajorVersion() {
return this.driver.getMajorVersion();
}
public int getMinorVersion() {
return this.driver.getMinorVersion();
}
public DriverPropertyInfo[] getPropertyInfo(String u, Properties p) throws SQLException {
return this.driver.getPropertyInfo(u, p);
}
public boolean jdbcCompliant() {
return this.driver.jdbcCompliant();
}
}
}