Formats contributions

partiql · Jan 5, 2024 · 481dd89 · 481dd89
1 parent ec149e2
commit 481dd89
Show file tree

Hide file tree

Showing 10 changed files with 82 additions and 26 deletions.
diff --git a/...s/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/BinaryOpHelpers.kt b/...s/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/BinaryOpHelpers.kt
@@ -29,27 +29,31 @@ import java.math.BigDecimal
 import java.math.BigInteger
 
 @OptIn(PartiQLValueExperimental::class)
+@JvmName("binaryOpInt8ByteToInt")
 internal inline fun binaryOpInt8(lhs: PartiQLValue, rhs: PartiQLValue, op: (Byte, Byte) -> Int): Int8Value {
     val lhsValue = lhs.check<Int8Value>().value!!
     val rhsValue = rhs.check<Int8Value>().value!!
     return int8Value((op(lhsValue, rhsValue)).toByte())
 }
 
 @OptIn(PartiQLValueExperimental::class)
+@JvmName("binaryOpInt8ByteToByte")
 internal inline fun binaryOpInt8(lhs: PartiQLValue, rhs: PartiQLValue, op: (Byte, Byte) -> Byte): Int8Value {
     val lhsValue = lhs.check<Int8Value>().value!!
     val rhsValue = rhs.check<Int8Value>().value!!
     return int8Value((op(lhsValue, rhsValue)))
 }
 
 @OptIn(PartiQLValueExperimental::class)
+@JvmName("binaryOpInt16ShortToInt")
 internal inline fun binaryOpInt16(lhs: PartiQLValue, rhs: PartiQLValue, op: (Short, Short) -> Int): Int16Value {
     val lhsValue = lhs.check<Int16Value>().value!!
     val rhsValue = rhs.check<Int16Value>().value!!
     return int16Value((op(lhsValue, rhsValue)).toShort())
 }
 
 @OptIn(PartiQLValueExperimental::class)
+@JvmName("binaryOpInt16ShortToShort")
 internal inline fun binaryOpInt16(lhs: PartiQLValue, rhs: PartiQLValue, op: (Short, Short) -> Short): Int16Value {
     val lhsValue = lhs.check<Int16Value>().value!!
     val rhsValue = rhs.check<Int16Value>().value!!

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnAnd.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnAnd.kt
@@ -7,12 +7,13 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.BoolValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.BOOL
 import org.partiql.value.PartiQLValueType.MISSING
-import org.partiql.value.*
-import org.partiql.value.PartiQLValueType.*
+import org.partiql.value.boolValue
+import org.partiql.value.check
 
 @OptIn(PartiQLValueExperimental::class, PartiQLFunctionExperimental::class)
 internal object Fn_AND__BOOL_BOOL__BOOL : PartiQLFunction.Scalar {

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnBitwiseAnd.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnBitwiseAnd.kt
@@ -7,14 +7,18 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.INT
 import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
 import java.math.BigInteger
 import kotlin.experimental.and
 

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnConcat.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnConcat.kt
@@ -7,12 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.ClobValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.CLOB
 import org.partiql.value.PartiQLValueType.STRING
 import org.partiql.value.PartiQLValueType.SYMBOL
-import org.partiql.value.*
+import org.partiql.value.StringValue
+import org.partiql.value.SymbolValue
 
 @OptIn(PartiQLValueExperimental::class, PartiQLFunctionExperimental::class)
 internal object Fn_CONCAT__STRING_STRING__STRING : PartiQLFunction.Scalar {

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnDivide.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnDivide.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,7 +25,6 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
 import java.math.BigDecimal
 import java.math.BigInteger
 

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnMinus.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnMinus.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,8 +25,6 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
-import org.partiql.value.PartiQLValueType.*
 import java.math.BigDecimal
 import java.math.BigInteger
 

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnModulo.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnModulo.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,8 +25,6 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
-import org.partiql.value.PartiQLValueType.*
 import java.math.BigDecimal
 import java.math.BigInteger
 

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnNeg.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnNeg.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,9 +25,15 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
-import java.math.BigDecimal
-import java.math.BigInteger
+import org.partiql.value.check
+import org.partiql.value.decimalValue
+import org.partiql.value.float32Value
+import org.partiql.value.float64Value
+import org.partiql.value.int16Value
+import org.partiql.value.int32Value
+import org.partiql.value.int64Value
+import org.partiql.value.int8Value
+import org.partiql.value.intValue
 
 @OptIn(PartiQLValueExperimental::class, PartiQLFunctionExperimental::class)
 internal object Fn_NEG__INT8__INT8 : PartiQLFunction.Scalar {
@@ -32,7 +46,7 @@ internal object Fn_NEG__INT8__INT8 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Int8Value {
         val value = args[0].check<Int8Value>().value!!
         return int8Value(value.times(-1).toByte())
     }
@@ -49,7 +63,7 @@ internal object Fn_NEG__INT16__INT16 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Int16Value {
         val value = args[0].check<Int16Value>().value!!
         return int16Value(value.times(-1).toShort())
     }
@@ -66,7 +80,7 @@ internal object Fn_NEG__INT32__INT32 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Int32Value {
         val value = args[0].check<Int32Value>().value!!
         return int32Value(value.times(-1))
     }
@@ -83,7 +97,7 @@ internal object Fn_NEG__INT64__INT64 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Int64Value {
         val value = args[0].check<Int64Value>().value!!
         return int64Value(value.times(-1L))
     }
@@ -100,9 +114,9 @@ internal object Fn_NEG__INT__INT : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): IntValue {
         val value = args[0].check<IntValue>().value!!
-        return intValue(value.times(-BigInteger.ONE))
+        return intValue(value.negate())
     }
 }
 
@@ -117,9 +131,9 @@ internal object Fn_NEG__DECIMAL_ARBITRARY__DECIMAL_ARBITRARY : PartiQLFunction.S
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): DecimalValue {
         val value = args[0].check<DecimalValue>().value!!
-        return decimalValue(value.times(-BigDecimal.ONE))
+        return decimalValue(value.negate())
     }
 }
 
@@ -134,7 +148,7 @@ internal object Fn_NEG__FLOAT32__FLOAT32 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Float32Value {
         val value = args[0].check<Float32Value>().value!!
         return float32Value(value.times(-1))
     }
@@ -151,7 +165,7 @@ internal object Fn_NEG__FLOAT64__FLOAT64 : PartiQLFunction.Scalar {
         isNullable = false,
     )
 
-    override fun invoke(args: Array<PartiQLValue>): PartiQLValue {
+    override fun invoke(args: Array<PartiQLValue>): Float64Value {
         val value = args[0].check<Float64Value>().value!!
         return float64Value(value.times(-1))
     }

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnPlus.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnPlus.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,8 +25,6 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
-import org.partiql.value.PartiQLValueType.*
 import java.math.BigDecimal
 import java.math.BigInteger
 

diff --git a/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnTimes.kt b/plugins/partiql-plugin/src/main/kotlin/org/partiql/plugin/internal/fn/scalar/FnTimes.kt
@@ -7,6 +7,14 @@ import org.partiql.spi.function.PartiQLFunction
 import org.partiql.spi.function.PartiQLFunctionExperimental
 import org.partiql.types.function.FunctionParameter
 import org.partiql.types.function.FunctionSignature
+import org.partiql.value.DecimalValue
+import org.partiql.value.Float32Value
+import org.partiql.value.Float64Value
+import org.partiql.value.Int16Value
+import org.partiql.value.Int32Value
+import org.partiql.value.Int64Value
+import org.partiql.value.Int8Value
+import org.partiql.value.IntValue
 import org.partiql.value.PartiQLValue
 import org.partiql.value.PartiQLValueExperimental
 import org.partiql.value.PartiQLValueType.DECIMAL_ARBITRARY
@@ -17,8 +25,6 @@ import org.partiql.value.PartiQLValueType.INT16
 import org.partiql.value.PartiQLValueType.INT32
 import org.partiql.value.PartiQLValueType.INT64
 import org.partiql.value.PartiQLValueType.INT8
-import org.partiql.value.*
-import org.partiql.value.PartiQLValueType.*
 import java.math.BigDecimal
 import java.math.BigInteger