Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.trino.operator.scalar.timestamp;
import io.trino.spi.connector.ConnectorSession;
import io.trino.spi.function.LiteralParameters;
import io.trino.spi.function.ScalarOperator;
import io.trino.spi.function.SqlType;
import io.trino.spi.type.LongTimestamp;
import io.trino.spi.type.StandardTypes;
import io.trino.type.Constraint;
import org.joda.time.DateTimeField;
import org.joda.time.chrono.ISOChronology;
import static io.trino.spi.function.OperatorType.ADD;
import static io.trino.spi.function.OperatorType.SUBTRACT;
import static io.trino.spi.type.TimestampType.MAX_SHORT_PRECISION;
import static io.trino.type.DateTimes.MICROSECONDS_PER_MILLISECOND;
import static io.trino.type.DateTimes.getMicrosOfMilli;
import static io.trino.type.DateTimes.rescale;
import static io.trino.type.DateTimes.round;
import static io.trino.type.DateTimes.scaleEpochMicrosToMillis;
import static io.trino.type.DateTimes.scaleEpochMillisToMicros;
import static java.lang.Math.multiplyExact;
@SuppressWarnings("UtilityClassWithoutPrivateConstructor")
public final class TimestampOperators
{
private TimestampOperators() {}
@ScalarOperator(ADD)
public static final class TimestampPlusIntervalDayToSecond
{
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static long add(
@SqlType("timestamp(p)") long timestamp,
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval)
{
// scale to micros
interval = multiplyExact(interval, MICROSECONDS_PER_MILLISECOND);
return timestamp + interval;
}
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static LongTimestamp add(
@SqlType("timestamp(p)") LongTimestamp timestamp,
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval)
{
return new LongTimestamp(timestamp.getEpochMicros() + multiplyExact(interval, MICROSECONDS_PER_MILLISECOND), timestamp.getPicosOfMicro());
}
}
@ScalarOperator(ADD)
public static final class IntervalDayToSecondPlusTimestamp
{
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static long add(
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval,
@SqlType("timestamp(p)") long timestamp)
{
return TimestampPlusIntervalDayToSecond.add(timestamp, interval);
}
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static LongTimestamp add(
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval,
@SqlType("timestamp(p)") LongTimestamp timestamp)
{
return TimestampPlusIntervalDayToSecond.add(timestamp, interval);
}
}
@ScalarOperator(ADD)
public static final class TimestampPlusIntervalYearToMonth
{
private static final DateTimeField MONTH_OF_YEAR_UTC = ISOChronology.getInstanceUTC().monthOfYear();
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static long add(
@SqlType("timestamp(p)") long timestamp,
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval)
{
long fractionMicros = getMicrosOfMilli(timestamp);
long result = MONTH_OF_YEAR_UTC.add(scaleEpochMicrosToMillis(timestamp), interval);
return scaleEpochMillisToMicros(result) + fractionMicros;
}
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static LongTimestamp add(
@SqlType("timestamp(p)") LongTimestamp timestamp,
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval)
{
return new LongTimestamp(
add(timestamp.getEpochMicros(), interval),
timestamp.getPicosOfMicro());
}
}
@ScalarOperator(ADD)
public static final class IntervalYearToMonthPlusTimestamp
{
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static long add(
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval,
@SqlType("timestamp(p)") long timestamp)
{
return TimestampPlusIntervalYearToMonth.add(timestamp, interval);
}
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static LongTimestamp add(
ConnectorSession session,
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval,
@SqlType("timestamp(p)") LongTimestamp timestamp)
{
return TimestampPlusIntervalYearToMonth.add(timestamp, interval);
}
}
@ScalarOperator(SUBTRACT)
public static final class TimestampMinusIntervalYearToMonth
{
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static long subtract(
@SqlType("timestamp(p)") long timestamp,
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval)
{
return TimestampPlusIntervalYearToMonth.add(timestamp, -interval);
}
@LiteralParameters("p")
@SqlType("timestamp(p)")
public static LongTimestamp subtract(
ConnectorSession session,
@SqlType("timestamp(p)") LongTimestamp timestamp,
@SqlType(StandardTypes.INTERVAL_YEAR_TO_MONTH) long interval)
{
return TimestampPlusIntervalYearToMonth.add(timestamp, -interval);
}
}
@ScalarOperator(SUBTRACT)
public static final class TimestampMinusIntervalDayToSecond
{
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static long subtract(
@SqlType("timestamp(p)") long timestamp,
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval)
{
return TimestampPlusIntervalDayToSecond.add(timestamp, -interval);
}
@LiteralParameters({"p", "u"})
@SqlType("timestamp(u)")
@Constraint(variable = "u", expression = "max(3, p)") // Interval is currently p = 3, so the minimum result precision is 3.
public static LongTimestamp subtract(
@SqlType("timestamp(p)") LongTimestamp timestamp,
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND) long interval)
{
return TimestampPlusIntervalDayToSecond.add(timestamp, -interval);
}
}
@ScalarOperator(SUBTRACT)
public static final class TimestampMinusTimestamp
{
@LiteralParameters("p")
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND)
public static long subtract(
@SqlType("timestamp(p)") long left,
@SqlType("timestamp(p)") long right)
{
long interval = left - right;
interval = round(interval, 3);
interval = rescale(interval, MAX_SHORT_PRECISION, 3);
return interval;
}
@LiteralParameters("p")
@SqlType(StandardTypes.INTERVAL_DAY_TO_SECOND)
public static long subtract(
@SqlType("timestamp(p)") LongTimestamp left,
@SqlType("timestamp(p)") LongTimestamp right)
{
return subtract(left.getEpochMicros(), right.getEpochMicros());
}
}
}
