/* (c) 2014 Open Source Geospatial Foundation - all rights reserved * (c) 2001 - 2013 OpenPlans * This code is licensed under the GPL 2.0 license, available at the root * application directory. */ package org.geoserver.monitor.hib; import java.sql.SQLException; import java.sql.Types; import org.hibernate.Hibernate; import org.hibernate.cfg.Environment; import org.hibernate.dialect.Dialect; import org.hibernate.dialect.function.NoArgSQLFunction; import org.hibernate.dialect.function.StandardSQLFunction; import org.hibernate.dialect.function.VarArgsSQLFunction; import org.hibernate.exception.TemplatedViolatedConstraintNameExtracter; import org.hibernate.exception.ViolatedConstraintNameExtracter; import org.hibernate.util.ReflectHelper; /** * A dialect compatible with the H2 database. * * @author Thomas Mueller * */ public class H2Dialect extends Dialect { private String querySequenceString; public H2Dialect() { super(); querySequenceString = "select sequence_name from information_schema.sequences"; try { Class constants = ReflectHelper.classForName("org.h2.engine.Constants"); Integer build = (Integer) constants.getDeclaredField("BUILD_ID").get(null); int buildId = build.intValue(); if (buildId < 32) { querySequenceString = "select name from information_schema.sequences"; } } catch (Throwable e) { // ignore (probably H2 not in the classpath) } registerColumnType(Types.BOOLEAN, "boolean"); registerColumnType(Types.BIGINT, "bigint"); registerColumnType(Types.BINARY, "binary"); registerColumnType(Types.BIT, "boolean"); registerColumnType(Types.CHAR, "varchar($l)"); registerColumnType(Types.DATE, "date"); registerColumnType(Types.DECIMAL, "decimal($p,$s)"); registerColumnType(Types.DOUBLE, "double"); registerColumnType(Types.FLOAT, "float"); registerColumnType(Types.INTEGER, "integer"); registerColumnType(Types.LONGVARBINARY, "longvarbinary"); registerColumnType(Types.LONGVARCHAR, "longvarchar"); registerColumnType(Types.REAL, "real"); registerColumnType(Types.SMALLINT, "smallint"); registerColumnType(Types.TINYINT, "tinyint"); registerColumnType(Types.TIME, "time"); registerColumnType(Types.TIMESTAMP, "timestamp"); registerColumnType(Types.VARCHAR, "varchar($l)"); registerColumnType(Types.VARBINARY, "binary($l)"); registerColumnType(Types.NUMERIC, "numeric"); registerColumnType(Types.BLOB, "blob"); registerColumnType(Types.CLOB, "clob"); // select topic, syntax from information_schema.help // where section like 'Function%' order by section, topic // registerFunction("abs", new StandardSQLFunction("abs")); registerFunction("acos", new StandardSQLFunction("acos", Hibernate.DOUBLE)); registerFunction("asin", new StandardSQLFunction("asin", Hibernate.DOUBLE)); registerFunction("atan", new StandardSQLFunction("atan", Hibernate.DOUBLE)); registerFunction("atan2", new StandardSQLFunction("atan2", Hibernate.DOUBLE)); registerFunction("bitand", new StandardSQLFunction("bitand", Hibernate.INTEGER)); registerFunction("bitor", new StandardSQLFunction("bitor", Hibernate.INTEGER)); registerFunction("bitxor", new StandardSQLFunction("bitxor", Hibernate.INTEGER)); registerFunction("ceiling", new StandardSQLFunction("ceiling", Hibernate.DOUBLE)); registerFunction("cos", new StandardSQLFunction("cos", Hibernate.DOUBLE)); registerFunction("cot", new StandardSQLFunction("cot", Hibernate.DOUBLE)); registerFunction("degrees", new StandardSQLFunction("degrees", Hibernate.DOUBLE)); registerFunction("exp", new StandardSQLFunction("exp", Hibernate.DOUBLE)); registerFunction("floor", new StandardSQLFunction("floor", Hibernate.DOUBLE)); registerFunction("log", new StandardSQLFunction("log", Hibernate.DOUBLE)); registerFunction("log10", new StandardSQLFunction("log10", Hibernate.DOUBLE)); // registerFunction("mod", new StandardSQLFunction("mod", Hibernate.INTEGER)); registerFunction("pi", new NoArgSQLFunction("pi", Hibernate.DOUBLE)); registerFunction("power", new StandardSQLFunction("power", Hibernate.DOUBLE)); registerFunction("radians", new StandardSQLFunction("radians", Hibernate.DOUBLE)); registerFunction("rand", new NoArgSQLFunction("rand", Hibernate.DOUBLE)); registerFunction("round", new StandardSQLFunction("round", Hibernate.DOUBLE)); registerFunction("roundmagic", new StandardSQLFunction("roundmagic", Hibernate.DOUBLE)); registerFunction("sign", new StandardSQLFunction("sign", Hibernate.INTEGER)); registerFunction("sin", new StandardSQLFunction("sin", Hibernate.DOUBLE)); // registerFunction("sqrt", new StandardSQLFunction("sqrt", Hibernate.DOUBLE)); registerFunction("tan", new StandardSQLFunction("tan", Hibernate.DOUBLE)); registerFunction("truncate", new StandardSQLFunction("truncate", Hibernate.DOUBLE)); registerFunction("compress", new StandardSQLFunction("compress", Hibernate.BINARY)); registerFunction("expand", new StandardSQLFunction("compress", Hibernate.BINARY)); registerFunction("decrypt", new StandardSQLFunction("decrypt", Hibernate.BINARY)); registerFunction("encrypt", new StandardSQLFunction("encrypt", Hibernate.BINARY)); registerFunction("hash", new StandardSQLFunction("hash", Hibernate.BINARY)); registerFunction("ascii", new StandardSQLFunction("ascii", Hibernate.INTEGER)); // registerFunction("bit_length", new StandardSQLFunction("bit_length", Hibernate.INTEGER)); registerFunction("char", new StandardSQLFunction("char", Hibernate.CHARACTER)); registerFunction("concat", new VarArgsSQLFunction(Hibernate.STRING, "(", "||", ")")); registerFunction("difference", new StandardSQLFunction("difference", Hibernate.INTEGER)); registerFunction("hextoraw", new StandardSQLFunction("hextoraw", Hibernate.STRING)); registerFunction("lower", new StandardSQLFunction("lower", Hibernate.STRING)); registerFunction("insert", new StandardSQLFunction("lower", Hibernate.STRING)); registerFunction("left", new StandardSQLFunction("left", Hibernate.STRING)); // registerFunction("length", new StandardSQLFunction("length", Hibernate.INTEGER)); // registerFunction("locate", new StandardSQLFunction("locate", Hibernate.INTEGER)); // registerFunction("lower", new StandardSQLFunction("lower", Hibernate.STRING)); registerFunction("lcase", new StandardSQLFunction("lcase", Hibernate.STRING)); registerFunction("ltrim", new StandardSQLFunction("ltrim", Hibernate.STRING)); registerFunction("octet_length", new StandardSQLFunction("octet_length", Hibernate.INTEGER)); registerFunction("position", new StandardSQLFunction("position", Hibernate.INTEGER)); registerFunction("rawtohex", new StandardSQLFunction("rawtohex", Hibernate.STRING)); registerFunction("repeat", new StandardSQLFunction("repeat", Hibernate.STRING)); registerFunction("replace", new StandardSQLFunction("replace", Hibernate.STRING)); registerFunction("right", new StandardSQLFunction("right", Hibernate.STRING)); registerFunction("rtrim", new StandardSQLFunction("rtrim", Hibernate.STRING)); registerFunction("soundex", new StandardSQLFunction("soundex", Hibernate.STRING)); registerFunction("space", new StandardSQLFunction("space", Hibernate.STRING)); registerFunction("stringencode", new StandardSQLFunction("stringencode", Hibernate.STRING)); registerFunction("stringdecode", new StandardSQLFunction("stringdecode", Hibernate.STRING)); // registerFunction("substring", new StandardSQLFunction("substring", Hibernate.STRING)); // registerFunction("upper", new StandardSQLFunction("upper", Hibernate.STRING)); registerFunction("ucase", new StandardSQLFunction("ucase", Hibernate.STRING)); registerFunction("stringtoutf8", new StandardSQLFunction("stringtoutf8", Hibernate.BINARY)); registerFunction("utf8tostring", new StandardSQLFunction("utf8tostring", Hibernate.STRING)); registerFunction("current_date", new NoArgSQLFunction("current_date", Hibernate.DATE)); registerFunction("current_time", new NoArgSQLFunction("current_time", Hibernate.TIME)); registerFunction("current_timestamp", new NoArgSQLFunction("current_timestamp", Hibernate.TIMESTAMP)); registerFunction("datediff", new NoArgSQLFunction("datediff", Hibernate.INTEGER)); registerFunction("dayname", new StandardSQLFunction("dayname", Hibernate.STRING)); registerFunction("dayofmonth", new StandardSQLFunction("dayofmonth", Hibernate.INTEGER)); registerFunction("dayofweek", new StandardSQLFunction("dayofweek", Hibernate.INTEGER)); registerFunction("dayofyear", new StandardSQLFunction("dayofyear", Hibernate.INTEGER)); // registerFunction("hour", new StandardSQLFunction("hour", Hibernate.INTEGER)); // registerFunction("minute", new StandardSQLFunction("minute", Hibernate.INTEGER)); // registerFunction("month", new StandardSQLFunction("month", Hibernate.INTEGER)); registerFunction("monthname", new StandardSQLFunction("monthname", Hibernate.STRING)); registerFunction("quarter", new StandardSQLFunction("quarter", Hibernate.INTEGER)); // registerFunction("second", new StandardSQLFunction("second", Hibernate.INTEGER)); registerFunction("week", new StandardSQLFunction("week", Hibernate.INTEGER)); // registerFunction("year", new StandardSQLFunction("year", Hibernate.INTEGER)); registerFunction("curdate", new NoArgSQLFunction("curdate", Hibernate.DATE)); registerFunction("curtime", new NoArgSQLFunction("curtime", Hibernate.TIME)); registerFunction("curtimestamp", new NoArgSQLFunction("curtimestamp", Hibernate.TIME)); registerFunction("now", new NoArgSQLFunction("now", Hibernate.TIMESTAMP)); registerFunction("database", new NoArgSQLFunction("database", Hibernate.STRING)); registerFunction("user", new NoArgSQLFunction("user", Hibernate.STRING)); getDefaultProperties().setProperty(Environment.STATEMENT_BATCH_SIZE, DEFAULT_BATCH_SIZE); } public String getAddColumnString() { return "add column"; } public boolean supportsIdentityColumns() { return true; } public String getIdentityColumnString() { return "generated by default as identity"; // not null is implicit } public String getIdentitySelectString() { return "call identity()"; } public String getIdentityInsertString() { return "null"; } public String getForUpdateString() { return " for update"; } public boolean supportsUnique() { return true; } public boolean supportsLimit() { return true; } public String getLimitString(String sql, boolean hasOffset) { return new StringBuffer(sql.length() + 20).append(sql).append( hasOffset ? " limit ? offset ?" : " limit ?").toString(); } public boolean bindLimitParametersInReverseOrder() { return true; } public boolean bindLimitParametersFirst() { return false; } public boolean supportsIfExistsAfterTableName() { return true; } public String[] getCreateSequenceStrings(String sequenceName) { return new String[] { "create sequence " + sequenceName }; } public String[] getDropSequenceStrings(String sequenceName) { return new String[] { "drop sequence " + sequenceName }; } public String getSelectSequenceNextValString(String sequenceName) { return "next value for " + sequenceName; } public String getSequenceNextValString(String sequenceName) { return "call next value for " + sequenceName; } public String getQuerySequencesString() { return querySequenceString; } public boolean supportsSequences() { return true; } public ViolatedConstraintNameExtracter getViolatedConstraintNameExtracter() { return EXTRACTER; } private static ViolatedConstraintNameExtracter EXTRACTER = new TemplatedViolatedConstraintNameExtracter() { /** * Extract the name of the violated constraint from the given SQLException. * * @param e * The exception that was the result of the constraint violation. * @return The extracted constraint name. */ public String extractConstraintName(SQLException e) { String constraintName = null; // 23000: Check constraint violation: {0} // 23001: Unique index or primary key violation: {0} if (e.getSQLState().startsWith("23")) { String message = e.getMessage(); int idx = message.indexOf("violation: "); if (idx > 0) { constraintName = message.substring(idx + "violation: ".length()); } } return constraintName; } }; public boolean supportsTemporaryTables() { return true; } public String getCreateTemporaryTableString() { return "create temporary table if not exists"; } public boolean supportsCurrentTimestampSelection() { return true; } public boolean isCurrentTimestampSelectStringCallable() { return false; } public String getCurrentTimestampSelectString() { return "call current_timestamp()"; } public boolean supportsUnionAll() { return true; } }