Skip to content

Commit

Permalink
Add PlannerPipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
dlurton committed Apr 30, 2022
1 parent 405ae74 commit 53f6bda
Show file tree
Hide file tree
Showing 4 changed files with 586 additions and 6 deletions.
14 changes: 8 additions & 6 deletions lang/src/org/partiql/lang/SqlException.kt
Original file line number Diff line number Diff line change
Expand Up @@ -34,12 +34,10 @@ import org.partiql.lang.util.propertyValueMapOf
*
* @param message the message for this exception
* @param errorCode the error code for this exception
* @param propertyValueMap context for this error
* @param errorContext context for this error, includes details like line & character offsets, among others.
* @param internal True to indicate that this exception a bug in ParitQL itself. False to indicate it was caused by
* incorrect usage of APIs or invalid end-user queries.
* @param cause for this exception
*
* @constructor a custom error [message], the [errorCode], error context as a [propertyValueMap] and optional [cause] creates an
* [SqlException]. This is the constructor for the second configuration explained above.
*
*/
open class SqlException(
override var message: String,
Expand Down Expand Up @@ -83,7 +81,7 @@ open class SqlException(
*
* * ErrorCategory is one of `Lexer Error`, `Parser Error`, `Runtime Error`
* * ErrorLocation is the line and column where the error occurred
* * Errormessatge is the **generated** error message
* * ErrorMessage is the **generated** error message
*
*
* TODO: Prepend to the auto-generated message the file name.
Expand All @@ -92,6 +90,10 @@ open class SqlException(
fun generateMessage(): String =
"${errorCategory(errorCode)}: ${errorLocation(errorContext)}: ${errorMessage(errorCode, errorContext)}"

/** Same as [generateMessage] but without the location. */
fun generateMessageNoLocation(): String =
"${errorCategory(errorCode)}: ${errorMessage(errorCode, errorContext)}"

private fun errorMessage(errorCode: ErrorCode?, propertyValueMap: PropertyValueMap?): String =
errorCode?.getErrorMessage(propertyValueMap) ?: UNKNOWN

Expand Down
85 changes: 85 additions & 0 deletions lang/src/org/partiql/lang/planner/EvaluatorOptions.kt
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
package org.partiql.lang.planner

import org.partiql.lang.eval.ProjectionIterationBehavior
import org.partiql.lang.eval.ThunkOptions
import org.partiql.lang.eval.TypedOpBehavior
import org.partiql.lang.eval.TypingMode
import java.time.ZoneOffset

/*
Differences between CompilerOptions and PlannerOptions:
- There is no EvaluatorOptions equivalent for CompileOptions.visitorTransformMode since the planner always runs some basic
normalization and variable resolution passes *before* the customer can inject their own transforms.
- There is no EvaluatorOptions equivalent for CompileOptions.thunkReturnTypeAssertions since PlannerPipeline does not
support the static type inferencer (yet).
- PlannerOptions.allowUndefinedVariables is new.
- PlannerOptions has no equivalent for CompileOptions.undefinedVariableBehavior -- this was added for backward
compatibility on behalf of a customer we don't have anymore. Internal bug number is IONSQL-134.
*/

/**
* Specifies options that effect the behavior of the PartiQL physical plan evaluator.
*
* @param defaultTimezoneOffset Default timezone offset to be used when TIME WITH TIME ZONE does not explicitly
* specify the time zone. Defaults to [ZoneOffset.UTC].
*/
@Suppress("DataClassPrivateConstructor")
data class EvaluatorOptions private constructor (
val projectionIteration: ProjectionIterationBehavior = ProjectionIterationBehavior.FILTER_MISSING,
val thunkOptions: ThunkOptions = ThunkOptions.standard(),
val typingMode: TypingMode = TypingMode.LEGACY,
val typedOpBehavior: TypedOpBehavior = TypedOpBehavior.LEGACY,
val defaultTimezoneOffset: ZoneOffset = ZoneOffset.UTC
) {
companion object {

/**
* Creates a java style builder that will choose the default values for any unspecified options.
*/
@JvmStatic
fun builder() = Builder()

/**
* Creates a java style builder that will clone the [EvaluatorOptions] passed to the constructor.
*/
@JvmStatic
fun builder(options: EvaluatorOptions) = Builder(options)

/**
* Kotlin style builder that will choose the default values for any unspecified options.
*/
fun build(block: Builder.() -> Unit) = Builder().apply(block).build()

/**
* Kotlin style builder that will clone the [EvaluatorOptions] passed to the constructor.
*/
fun build(options: EvaluatorOptions, block: Builder.() -> Unit) = Builder(options).apply(block).build()

/**
* Creates a [EvaluatorOptions] instance with the standard values for use by the legacy AST compiler.
*/
@JvmStatic
fun standard() = Builder().build()
}

/**
* Builds a [EvaluatorOptions] instance.
*/
class Builder(private var options: EvaluatorOptions = EvaluatorOptions()) {

fun projectionIteration(value: ProjectionIterationBehavior) = set { copy(projectionIteration = value) }
fun typingMode(value: TypingMode) = set { copy(typingMode = value) }
fun typedOpBehavior(value: TypedOpBehavior) = set { copy(typedOpBehavior = value) }
fun thunkOptions(value: ThunkOptions) = set { copy(thunkOptions = value) }
fun defaultTimezoneOffset(value: ZoneOffset) = set { copy(defaultTimezoneOffset = value) }

private inline fun set(block: EvaluatorOptions.() -> EvaluatorOptions): Builder {
options = block(options)
return this
}

fun build() = options
}
}
Loading

0 comments on commit 53f6bda

Please sign in to comment.