166 lines
4.3 KiB
Java
166 lines
4.3 KiB
Java
/*
|
|
* 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.
|
|
*
|
|
* Other licenses:
|
|
* -----------------------------------------------------------------------------
|
|
* Commercial licenses for this work are available. These replace the above
|
|
* ASL 2.0 and offer limited warranties, support, maintenance, and commercial
|
|
* database integrations.
|
|
*
|
|
* For more information, please visit: http://www.jooq.org/licenses
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*
|
|
*/
|
|
package org.jooq.impl;
|
|
|
|
import static org.jooq.impl.DSL.*;
|
|
import static org.jooq.impl.Internal.*;
|
|
import static org.jooq.impl.Keywords.*;
|
|
import static org.jooq.impl.Names.*;
|
|
import static org.jooq.impl.SQLDataType.*;
|
|
import static org.jooq.impl.Tools.*;
|
|
import static org.jooq.impl.Tools.BooleanDataKey.*;
|
|
import static org.jooq.impl.Tools.DataKey.*;
|
|
import static org.jooq.SQLDialect.*;
|
|
|
|
import org.jooq.*;
|
|
import org.jooq.conf.*;
|
|
import org.jooq.impl.*;
|
|
import org.jooq.tools.*;
|
|
|
|
import java.util.*;
|
|
|
|
|
|
/**
|
|
* The <code>LPAD</code> statement.
|
|
*/
|
|
@SuppressWarnings({ "rawtypes", "unchecked", "unused" })
|
|
final class Lpad
|
|
extends
|
|
AbstractField<String>
|
|
{
|
|
|
|
private static final long serialVersionUID = 1L;
|
|
|
|
private final Field<String> string;
|
|
private final Field<? extends Number> length;
|
|
private final Field<String> character;
|
|
|
|
Lpad(
|
|
Field<String> string,
|
|
Field<? extends Number> length
|
|
) {
|
|
super(
|
|
N_LPAD,
|
|
allNotNull(VARCHAR, string, length)
|
|
);
|
|
|
|
this.string = nullSafeNotNull(string, VARCHAR);
|
|
this.length = nullSafeNotNull(length, INTEGER);
|
|
this.character = null;
|
|
}
|
|
|
|
Lpad(
|
|
Field<String> string,
|
|
Field<? extends Number> length,
|
|
Field<String> character
|
|
) {
|
|
super(
|
|
N_LPAD,
|
|
allNotNull(VARCHAR, string, length, character)
|
|
);
|
|
|
|
this.string = nullSafeNotNull(string, VARCHAR);
|
|
this.length = nullSafeNotNull(length, INTEGER);
|
|
this.character = nullSafeNotNull(character, VARCHAR);
|
|
}
|
|
|
|
// -------------------------------------------------------------------------
|
|
// XXX: QueryPart API
|
|
// -------------------------------------------------------------------------
|
|
|
|
|
|
|
|
private final Field<String> character() {
|
|
return character == null ? inline(" ") : character;
|
|
}
|
|
|
|
@Override
|
|
public final void accept(Context<?> ctx) {
|
|
switch (ctx.family()) {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// This beautiful expression was contributed by "Ludo", here:
|
|
// http://stackoverflow.com/questions/6576343/how-to-simulate-lpad-rpad-with-sqlite
|
|
case SQLITE:
|
|
ctx.visit(N_SUBSTR).sql('(')
|
|
.visit(N_REPLACE).sql('(')
|
|
.visit(N_HEX).sql('(')
|
|
.visit(N_ZEROBLOB).sql('(')
|
|
.visit(length)
|
|
.sql(")), '00', ").visit(character())
|
|
.sql("), 1, ").visit(length).sql(" - ").visit(N_LENGTH).sql('(').visit(string)
|
|
.sql(")) || ").visit(string);
|
|
break;
|
|
|
|
default:
|
|
ctx.visit(N_LPAD).sql('(').visit(string).sql(", ").visit(length).sql(", ").visit(character()).sql(')');
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
// -------------------------------------------------------------------------
|
|
// The Object API
|
|
// -------------------------------------------------------------------------
|
|
|
|
@Override
|
|
public boolean equals(Object that) {
|
|
if (that instanceof Lpad) {
|
|
return
|
|
StringUtils.equals(string, ((Lpad) that).string) &&
|
|
StringUtils.equals(length, ((Lpad) that).length) &&
|
|
StringUtils.equals(character, ((Lpad) that).character)
|
|
;
|
|
}
|
|
else
|
|
return super.equals(that);
|
|
}
|
|
}
|