io.questdb.griffin.engine.functions.StrFunction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of questdb Show documentation
QuestDB is high performance SQL time series database
There is a newer version: 8.3.2
/*******************************************************************************
 *     ___                  _   ____  ____
 *    / _ \ _   _  ___  ___| |_|  _ \| __ )
 *   | | | | | | |/ _ \/ __| __| | | |  _ \
 *   | |_| | |_| |  __/\__ \ |_| |_| | |_) |
 *    \__\_\\__,_|\___||___/\__|____/|____/
 *
 *  Copyright (c) 2014-2019 Appsicle
 *  Copyright (c) 2019-2024 QuestDB
 *
 *  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.
 *
 ******************************************************************************/

package io.questdb.griffin.engine.functions;


import io.questdb.cairo.ColumnType;
import io.questdb.cairo.TableUtils;
import io.questdb.cairo.sql.Record;
import io.questdb.cairo.sql.RecordCursorFactory;
import io.questdb.cairo.sql.ScalarFunction;
import io.questdb.griffin.SqlUtil;
import io.questdb.std.BinarySequence;
import io.questdb.std.Long256;
import io.questdb.std.str.CharSink;
import io.questdb.std.str.Utf8Sequence;
import io.questdb.std.str.Utf8StringSink;

public abstract class StrFunction implements ScalarFunction {
    private final Utf8StringSink utf8SinkA = new Utf8StringSink();
    private final Utf8StringSink utf8SinkB = new Utf8StringSink();

    @Override
    public final BinarySequence getBin(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public long getBinLen(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final boolean getBool(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final byte getByte(Record rec) {
        return SqlUtil.implicitCastStrAsByte(getStrA(rec));
    }

    @Override
    public final char getChar(Record rec) {
        return SqlUtil.implicitCastStrAsChar(getStrA(rec));
    }

    @Override
    public final long getDate(Record rec) {
        return SqlUtil.implicitCastStrAsDate(getStrA(rec));
    }

    @Override
    public final double getDouble(Record rec) {
        return SqlUtil.implicitCastStrAsDouble(getStrA(rec));
    }

    @Override
    public final float getFloat(Record rec) {
        return SqlUtil.implicitCastStrAsFloat(getStrA(rec));
    }

    @Override
    public byte getGeoByte(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public int getGeoInt(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public long getGeoLong(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public short getGeoShort(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final int getIPv4(Record rec) {
        return SqlUtil.implicitCastStrAsIPv4(getStrA(rec));
    }

    @Override
    public final int getInt(Record rec) {
        return SqlUtil.implicitCastStrAsInt(getStrA(rec));
    }

    @Override
    public final long getLong(Record rec) {
        return SqlUtil.implicitCastStrAsLong(getStrA(rec));
    }

    @Override
    public long getLong128Hi(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public long getLong128Lo(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final void getLong256(Record rec, CharSink sink) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final Long256 getLong256A(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final Long256 getLong256B(Record rec) {
        throw new UnsupportedOperationException();
    }

    @Override
    public final RecordCursorFactory getRecordCursorFactory() {
        throw new UnsupportedOperationException();
    }

    @Override
    public final short getShort(Record rec) {
        return SqlUtil.implicitCastStrAsShort(getStrA(rec));
    }

    @Override
    public int getStrLen(Record rec) {
        return TableUtils.lengthOf(getStrA(rec));
    }

    @Override
    public final CharSequence getSymbol(Record rec) {
        return getStrA(rec);
    }

    @Override
    public final CharSequence getSymbolB(Record rec) {
        return getStrB(rec);
    }

    @Override
    public final long getTimestamp(Record rec) {
        return SqlUtil.implicitCastStrAsTimestamp(getStrA(rec));
    }

    @Override
    public final int getType() {
        return ColumnType.STRING;
    }

    @Override
    public Utf8Sequence getVarcharA(Record rec) {
        final CharSequence cs = getStrA(rec);
        if (cs != null) {
            utf8SinkA.clear();
            utf8SinkA.put(cs);
            return utf8SinkA;
        }
        return null;
    }

    @Override
    public Utf8Sequence getVarcharB(Record rec) {
        final CharSequence cs = getStrB(rec);
        if (cs != null) {
            utf8SinkB.clear();
            utf8SinkB.put(cs);
            return utf8SinkB;
        }
        return null;
    }

    @Override
    public int getVarcharSize(Record rec) {
        CharSequence utf16Value = getStrA(rec);
        if (utf16Value == null) {
            return TableUtils.NULL_LEN;
        }
        utf8SinkA.clear();
        utf8SinkA.put(utf16Value);
        return utf8SinkA.size();
    }
}