/**
* Copyright (c) 2009-2014, Data Geekery GmbH (http://www.datageekery.com)
* All rights reserved.
*
* This work is dual-licensed
* - under the Apache Software License 2.0 (the "ASL")
* - under the jOOQ License and Maintenance Agreement (the "jOOQ License")
* =============================================================================
* You may choose which license applies to you:
*
* - If you're using this work with Open Source databases, you may choose
* either ASL or jOOQ License.
* - If you're using this work with at least one commercial database, you must
* choose jOOQ License
*
* For more information, please visit http://www.jooq.org/licenses
*
* Apache Software License 2.0:
* -----------------------------------------------------------------------------
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* jOOQ License and Maintenance Agreement:
* -----------------------------------------------------------------------------
* Data Geekery grants the Customer the non-exclusive, timely limited and
* non-transferable license to install and use the Software under the terms of
* the jOOQ License and Maintenance Agreement.
*
* This library is distributed with a LIMITED WARRANTY. See the jOOQ License
* and Maintenance Agreement for more details: http://www.jooq.org/licensing
*/
package org.jooq.impl;
import static org.jooq.Clause.SEQUENCE;
import static org.jooq.Clause.SEQUENCE_REFERENCE;
import static org.jooq.SQLDialect.CUBRID;
import static org.jooq.SQLDialect.FIREBIRD;
// ...
import static org.jooq.impl.DSL.field;
import static org.jooq.impl.DSL.inline;
import static org.jooq.impl.DSL.select;
import org.jooq.Clause;
import org.jooq.Configuration;
import org.jooq.Context;
import org.jooq.DataType;
import org.jooq.Field;
import org.jooq.RenderContext;
import org.jooq.SQLDialect;
import org.jooq.Schema;
import org.jooq.Sequence;
import org.jooq.exception.SQLDialectNotSupportedException;
/**
* A common base class for sequences
* <p>
* This type is for JOOQ INTERNAL USE only. Do not reference directly
*
* @author Lukas Eder
*/
public class SequenceImpl<T extends Number> extends AbstractQueryPart implements Sequence<T> {
/**
* Generated UID
*/
private static final long serialVersionUID = 6224349401603636427L;
private static final Clause[] CLAUSES = { SEQUENCE, SEQUENCE_REFERENCE };
final String name;
final boolean nameIsPlainSQL;
final Schema schema;
final DataType<T> type;
public SequenceImpl(String name, Schema schema, DataType<T> type) {
this(name, schema, type, false);
}
SequenceImpl(String name, Schema schema, DataType<T> type, boolean nameIsPlainSQL) {
this.name = name;
this.schema = schema;
this.type = type;
this.nameIsPlainSQL = nameIsPlainSQL;
}
@Override
public final String getName() {
return name;
}
@Override
public final Schema getSchema() {
return schema;
}
@Override
public final DataType<T> getDataType() {
return type;
}
@Override
public final Field<T> currval() {
return new SequenceFunction("currval");
}
@Override
public final Field<T> nextval() {
return new SequenceFunction("nextval");
}
private class SequenceFunction extends AbstractFunction<T> {
/**
* Generated UID
*/
private static final long serialVersionUID = 2292275568395094887L;
private final String method;
SequenceFunction(String method) {
super(method, type);
this.method = method;
}
@Override
final Field<T> getFunction0(Configuration configuration) {
SQLDialect family = configuration.dialect().family();
switch (family) {
/* [pro] xx
xxxx xxxx
xxxx xxxxxxx
xxxx xxxxxxx
xxxx xxxxxxx x
xxxxxx xxxxx x xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx x xxx x xxxxxxx
xxxxxx xxxxxxxxxxxx xxxxxxxxxxxxxxx
x
xx [/pro] */
case POSTGRES: {
String field = method + "('" + getQualifiedName(configuration) + "')";
return field(field, getDataType());
}
case H2: {
String field = method + "(" + getQualifiedName(configuration, true) + ")";
return field(field, getDataType());
}
/* [pro] xx
xxxx xxxxxxxxxx
xx [/pro] */
case FIREBIRD:
case DERBY:
case HSQLDB: {
if ("nextval".equals(method)) {
String field = "next value for " + getQualifiedName(configuration);
return field(field, getDataType());
}
else if (family == FIREBIRD) {
return field("gen_id(" + getQualifiedName(configuration) + ", 0)", getDataType());
}
/* [pro] xx
xxxx xx xxxxxxx xx xxxxxxxxxx x
xxxxxx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
xxxxxxxxxxxxxxxxxxxx xxxx
xxxxxxxxxxxxxxxxxx xxxx
xxxxxxxxxxxxxxxxx x xxxxxxxxxxxxxx
xxxxxxxxxxxxxxx x xxx xxxxx
xxxxx
xxxxxx xx xxxx x xx x xx x xxxxxxxx x xxx
xxxxxx xx xxxx x xxx xxxxxxxxx x xxx xxxxxxxx x xxxxxxxxxxxxxxxx xx
xxxxxxxxxx
xxxxxxxxxxxx
x
xx [/pro] */
else {
throw new SQLDialectNotSupportedException("The sequence's current value functionality is not supported for the " + family + " dialect.");
}
}
case CUBRID: {
String field = getQualifiedName(configuration) + ".";
if ("nextval".equals(method)) {
field += "next_value";
}
else {
field += "current_value";
}
return field(field, getDataType());
}
// Default is needed for hashCode() and toString()
default: {
String field = getQualifiedName(configuration) + "." + method;
return field(field, getDataType());
}
}
}
private final String getQualifiedName(Configuration configuration) {
return getQualifiedName(configuration, false);
}
private final String getQualifiedName(Configuration configuration, boolean asStringLiterals) {
RenderContext local = create(configuration).renderContext();
accept0(local, asStringLiterals);
return local.render();
}
}
// ------------------------------------------------------------------------
// XXX: QueryPart API
// ------------------------------------------------------------------------
@Override
public final void accept(Context<?> ctx) {
accept0(ctx, false);
}
private final void accept0(Context<?> ctx, boolean asStringLiterals) {
Schema mappedSchema = Utils.getMappedSchema(ctx.configuration(), schema);
if (mappedSchema != null && ctx.family() != CUBRID) {
if (asStringLiterals) {
ctx.visit(inline(mappedSchema.getName()))
.sql(", ");
}
else {
ctx.visit(mappedSchema)
.sql(".");
}
}
if (asStringLiterals)
ctx.visit(inline(name));
else if (nameIsPlainSQL)
ctx.sql(name);
else
ctx.literal(name);
}
@Override
public final Clause[] clauses(Context<?> ctx) {
return CLAUSES;
}
// ------------------------------------------------------------------------
// XXX: Object API
// ------------------------------------------------------------------------
@Override
public int hashCode() {
// [#1938] This is a much more efficient hashCode() implementation
// compared to that of standard QueryParts
return name.hashCode();
}
}