expr

package
v0.4.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Aug 2, 2025 License: Apache-2.0, MIT Imports: 16 Imported by: 0

Documentation

Overview

Package expr provides expression evaluation for DataFrame operations

Index

Constants

View Source 
const (
	AggNameSum   = "sum"
	AggNameCount = "count"
	AggNameMean  = "mean"
	AggNameMin   = "min"
	AggNameMax   = "max"
)

Aggregation function name constants.

Variables

This section is empty.

Functions

func ExpressionToColumnName added in v0.3.0 

func ExpressionToColumnName(expr Expr) string

ExpressionToColumnName converts an aggregation expression to its standardized column name This is used to generate consistent column names for aggregated results.

func ValidateExpressionContext added in v0.3.0 

func ValidateExpressionContext(expr Expr, ctx EvaluationContext) error

ValidateExpressionContext validates that an expression can be used in the given context.

Types

type AggregationContext added in v0.3.0 

type AggregationContext struct {
	// contains filtered or unexported fields
}

AggregationContext manages mapping between aggregated column references and their actual column names in GROUP BY results.

When HAVING expressions reference aggregation results like SUM(sales), the context maps these to the actual column names in the aggregated DataFrame (e.g., "sum_sales").

func BuildContextFromAggregations added in v0.3.0 

func BuildContextFromAggregations(aggregations []*AggregationExpr) *AggregationContext

BuildContextFromAggregations creates an AggregationContext from a list of aggregation expressions.

func NewAggregationContext added in v0.3.0 

func NewAggregationContext() *AggregationContext

NewAggregationContext creates a new AggregationContext.

func (*AggregationContext) AddMapping added in v0.3.0 

func (ac *AggregationContext) AddMapping(exprStr, columnName string)

AddMapping adds a mapping between an aggregation expression and its column name.

func (*AggregationContext) AllMappings added in v0.3.0 

func (ac *AggregationContext) AllMappings() map[string]string

AllMappings returns all column mappings.

func (*AggregationContext) Clear added in v0.3.0 

func (ac *AggregationContext) Clear()

Clear removes all mappings.

func (*AggregationContext) GetColumnName added in v0.3.0 

func (ac *AggregationContext) GetColumnName(exprStr string) (string, bool)

GetColumnName returns the column name for the given expression string.

func (*AggregationContext) GetExpression added in v0.3.0 

func (ac *AggregationContext) GetExpression(columnName string) (string, bool)

GetExpression returns the expression string for the given column name.

func (*AggregationContext) HasMapping added in v0.3.0 

func (ac *AggregationContext) HasMapping(exprStr string) bool

HasMapping checks if a mapping exists for the given expression string.

func (*AggregationContext) String added in v0.3.0 

func (ac *AggregationContext) String() string

String returns a string representation of the context.

type AggregationExpr 

type AggregationExpr struct {
	// contains filtered or unexported fields
}

AggregationExpr represents an aggregation function over a column.

func Count 

func Count(column Expr) *AggregationExpr

Count creates a count aggregation expression.

func Max 

func Max(column Expr) *AggregationExpr

Max creates a max aggregation expression.

func Mean 

func Mean(column Expr) *AggregationExpr

Mean creates a mean aggregation expression.

func Min 

func Min(column Expr) *AggregationExpr

Min creates a min aggregation expression.

func Sum 

func Sum(column Expr) *AggregationExpr

Sum creates a sum aggregation expression.

func (*AggregationExpr) Abs added in v0.3.0 

func (a *AggregationExpr) Abs() *FunctionExpr

Abs creates an absolute value function expression.

func (*AggregationExpr) Add added in v0.3.0 

func (a *AggregationExpr) Add(other Expr) *BinaryExpr

Add creates an addition expression.

func (*AggregationExpr) AggType 

func (a *AggregationExpr) AggType() AggregationType

func (*AggregationExpr) Alias 

func (a *AggregationExpr) Alias() string

func (*AggregationExpr) And added in v0.3.0 

func (a *AggregationExpr) And(other Expr) *BinaryExpr

And creates a logical AND expression.

func (*AggregationExpr) As 

func (a *AggregationExpr) As(alias string) *AggregationExpr

As sets an alias for the aggregation expression.

func (*AggregationExpr) Ceil added in v0.3.0 

func (a *AggregationExpr) Ceil() *FunctionExpr

Ceil creates a ceil function expression.

func (*AggregationExpr) Column 

func (a *AggregationExpr) Column() Expr

func (*AggregationExpr) Div added in v0.3.0 

func (a *AggregationExpr) Div(other Expr) *BinaryExpr

Div creates a division expression.

func (*AggregationExpr) Eq added in v0.3.0 

func (a *AggregationExpr) Eq(other Expr) *BinaryExpr

Eq creates an equality expression.

func (*AggregationExpr) Floor added in v0.3.0 

func (a *AggregationExpr) Floor() *FunctionExpr

Floor creates a floor function expression.

func (*AggregationExpr) Ge added in v0.3.0 

func (a *AggregationExpr) Ge(other Expr) *BinaryExpr

Ge creates a greater-than-or-equal expression.

func (*AggregationExpr) Gt added in v0.3.0 

func (a *AggregationExpr) Gt(other Expr) *BinaryExpr

Gt creates a greater-than expression.

func (*AggregationExpr) Le added in v0.3.0 

func (a *AggregationExpr) Le(other Expr) *BinaryExpr

Le creates a less-than-or-equal expression.

func (*AggregationExpr) Log added in v0.3.0 

func (a *AggregationExpr) Log() *FunctionExpr

Log creates a natural logarithm function expression.

func (*AggregationExpr) Lt added in v0.3.0 

func (a *AggregationExpr) Lt(other Expr) *BinaryExpr

Lt creates a less-than expression.

func (*AggregationExpr) Mul added in v0.3.0 

func (a *AggregationExpr) Mul(other Expr) *BinaryExpr

Mul creates a multiplication expression.

func (*AggregationExpr) Ne added in v0.3.0 

func (a *AggregationExpr) Ne(other Expr) *BinaryExpr

Ne creates a not-equal expression.

func (*AggregationExpr) Neg added in v0.3.0 

func (a *AggregationExpr) Neg() *UnaryExpr

Neg creates a negation (unary minus) expression.

func (*AggregationExpr) Not added in v0.3.0 

func (a *AggregationExpr) Not() *UnaryExpr

Not creates a logical NOT expression.

func (*AggregationExpr) Or added in v0.3.0 

func (a *AggregationExpr) Or(other Expr) *BinaryExpr

Or creates a logical OR expression.

func (*AggregationExpr) Over added in v0.2.0 

func (a *AggregationExpr) Over(window *WindowSpec) *WindowExpr

Over creates a window expression with the specified window for aggregation functions.

func (*AggregationExpr) Round added in v0.3.0 

func (a *AggregationExpr) Round() *FunctionExpr

Round creates a round function expression.

func (*AggregationExpr) RoundTo added in v0.3.0 

func (a *AggregationExpr) RoundTo(precision Expr) *FunctionExpr

RoundTo creates a round function expression with precision.

func (*AggregationExpr) Sqrt added in v0.3.0 

func (a *AggregationExpr) Sqrt() *FunctionExpr

Sqrt creates a square root function expression.

func (*AggregationExpr) String 

func (a *AggregationExpr) String() string

func (*AggregationExpr) Sub added in v0.3.0 

func (a *AggregationExpr) Sub(other Expr) *BinaryExpr

Sub creates a subtraction expression.

func (*AggregationExpr) SupportsContext added in v0.3.0 

func (a *AggregationExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context AggregationExpr can only be evaluated in GroupContext.

func (*AggregationExpr) Type 

func (a *AggregationExpr) Type() Type

type AggregationType 

type AggregationType int

AggregationType represents the type of aggregation function. 

const (
	AggSum AggregationType = iota
	AggCount
	AggMean
	AggMin
	AggMax
)

type AliasResolver added in v0.3.0 

type AliasResolver struct {
	// contains filtered or unexported fields
}

AliasResolver handles comprehensive alias resolution for HAVING clauses. It supports user-defined aliases, default aggregation names, and GROUP BY columns, providing a unified interface for column name resolution.

func BuildAliasResolver added in v0.3.0 

func BuildAliasResolver(
	groupByColumns []string,
	aggregations []*AggregationExpr,
	caseInsensitive bool,
) (*AliasResolver, error)

BuildAliasResolver creates an AliasResolver from GROUP BY columns and aggregation expressions. This is a convenience function for common setup scenarios.

func NewAliasResolver added in v0.3.0 

func NewAliasResolver(caseInsensitive bool) *AliasResolver

NewAliasResolver creates a new AliasResolver with the specified configuration.

func (*AliasResolver) AddAggregation added in v0.3.0 

func (ar *AliasResolver) AddAggregation(aggExpr *AggregationExpr) error

AddAggregation adds an aggregation expression with its optional user-defined alias. This handles both user aliases and default names for aggregation expressions.

func (*AliasResolver) AddGroupByColumn added in v0.3.0 

func (ar *AliasResolver) AddGroupByColumn(columnName string)

AddGroupByColumn adds a GROUP BY column to the resolver. GROUP BY columns can be referenced by their original names in HAVING.

func (*AliasResolver) GenerateDefaultName added in v0.3.0 

func (ar *AliasResolver) GenerateDefaultName(aggExpr *AggregationExpr) string

GenerateDefaultName creates a default column name for an aggregation expression.

func (*AliasResolver) GetAllAvailableAliases added in v0.3.0 

func (ar *AliasResolver) GetAllAvailableAliases() []string

GetAllAvailableAliases returns all aliases that can be used in HAVING clauses. This includes GROUP BY columns, user-defined aliases, and default aggregation names.

func (*AliasResolver) GetAvailableAliases added in v0.3.0 

func (ar *AliasResolver) GetAvailableAliases(columnName string) []string

GetAvailableAliases returns all valid aliases for a given column name. This is useful for error messages and user assistance.

func (*AliasResolver) GetColumnNameFromExpression added in v0.3.0 

func (ar *AliasResolver) GetColumnNameFromExpression(exprStr string) (string, bool)

GetColumnNameFromExpression resolves a column name from an aggregation expression string. This is used when validating direct aggregation references in HAVING.

func (*AliasResolver) ResolveAlias added in v0.3.0 

func (ar *AliasResolver) ResolveAlias(alias string) (string, bool)

ResolveAlias attempts to resolve an alias to its actual column name. It checks user aliases, default names, and GROUP BY columns in that order.

func (*AliasResolver) ValidateAlias added in v0.3.0 

func (ar *AliasResolver) ValidateAlias(alias string) error

ValidateAlias checks if the given alias is valid and unambiguous.

type ArithmeticArray added in v0.3.1 

type ArithmeticArray[T any] interface {
	Len() int
	IsNull(int) bool
	Value(int) T
}

ArithmeticArray defines interface for arrays supporting arithmetic operations.

type ArithmeticBuilder added in v0.3.1 

type ArithmeticBuilder[T any] interface {
	Release()
	AppendNull()
	Append(T)
	NewArray() arrow.Array
}

type BinaryExpr 

type BinaryExpr struct {
	// contains filtered or unexported fields
}

BinaryExpr represents a binary operation.

func NewBinaryExpr added in v0.3.0 

func NewBinaryExpr(left Expr, op BinaryOp, right Expr) *BinaryExpr

NewBinaryExpr creates a new binary expression.

func (*BinaryExpr) Abs 

func (b *BinaryExpr) Abs() *FunctionExpr

Abs creates an absolute value function expression.

func (*BinaryExpr) Add 

func (b *BinaryExpr) Add(other Expr) *BinaryExpr

Add creates an addition expression.

func (*BinaryExpr) And 

func (b *BinaryExpr) And(other Expr) *BinaryExpr

And creates a logical AND expression.

func (*BinaryExpr) Ceil 

func (b *BinaryExpr) Ceil() *FunctionExpr

Ceil creates a ceil function expression.

func (*BinaryExpr) Cos 

func (b *BinaryExpr) Cos() *FunctionExpr

Cos creates a cosine function expression.

func (*BinaryExpr) Div 

func (b *BinaryExpr) Div(other Expr) *BinaryExpr

Div creates a division expression.

func (*BinaryExpr) Eq added in v0.3.0 

func (b *BinaryExpr) Eq(other Expr) *BinaryExpr

Eq creates an equality expression.

func (*BinaryExpr) Floor 

func (b *BinaryExpr) Floor() *FunctionExpr

Floor creates a floor function expression.

func (*BinaryExpr) Ge added in v0.3.0 

func (b *BinaryExpr) Ge(other Expr) *BinaryExpr

Ge creates a greater-than-or-equal expression.

func (*BinaryExpr) Gt added in v0.3.0 

func (b *BinaryExpr) Gt(other Expr) *BinaryExpr

Gt creates a greater-than expression.

func (*BinaryExpr) Le added in v0.3.0 

func (b *BinaryExpr) Le(other Expr) *BinaryExpr

Le creates a less-than-or-equal expression.

func (*BinaryExpr) Left 

func (b *BinaryExpr) Left() Expr

func (*BinaryExpr) Log 

func (b *BinaryExpr) Log() *FunctionExpr

Log creates a natural logarithm function expression.

func (*BinaryExpr) Lt added in v0.3.0 

func (b *BinaryExpr) Lt(other Expr) *BinaryExpr

Lt creates a less-than expression.

func (*BinaryExpr) Mul 

func (b *BinaryExpr) Mul(other Expr) *BinaryExpr

Mul creates a multiplication expression.

func (*BinaryExpr) Ne added in v0.3.0 

func (b *BinaryExpr) Ne(other Expr) *BinaryExpr

Ne creates a not-equal expression.

func (*BinaryExpr) Op 

func (b *BinaryExpr) Op() BinaryOp

func (*BinaryExpr) Or 

func (b *BinaryExpr) Or(other Expr) *BinaryExpr

Or creates a logical OR expression.

func (*BinaryExpr) Right 

func (b *BinaryExpr) Right() Expr

func (*BinaryExpr) Round 

func (b *BinaryExpr) Round() *FunctionExpr

Round creates a round function expression.

func (*BinaryExpr) RoundTo 

func (b *BinaryExpr) RoundTo(precision Expr) *FunctionExpr

RoundTo creates a round function expression with precision.

func (*BinaryExpr) Sin 

func (b *BinaryExpr) Sin() *FunctionExpr

Sin creates a sine function expression.

func (*BinaryExpr) Sqrt 

func (b *BinaryExpr) Sqrt() *FunctionExpr

Sqrt creates a square root function expression.

func (*BinaryExpr) String 

func (b *BinaryExpr) String() string

func (*BinaryExpr) Sub 

func (b *BinaryExpr) Sub(other Expr) *BinaryExpr

Sub creates a subtraction expression.

func (*BinaryExpr) SupportsContext added in v0.3.0 

func (b *BinaryExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context BinaryExpr context support depends on operands.

func (*BinaryExpr) Type 

func (b *BinaryExpr) Type() Type

type BinaryOp 

type BinaryOp int

BinaryOp represents binary operations. 

const (
	OpAdd BinaryOp = iota
	OpSub
	OpMul
	OpDiv
	OpEq
	OpNe
	OpLt
	OpLe
	OpGt
	OpGe
	OpAnd
	OpOr
)

type BooleanComparator added in v0.3.1 

type BooleanComparator struct {
	// contains filtered or unexported fields
}

BooleanComparator provides optimized boolean comparison.

func (*BooleanComparator) Compare added in v0.3.1 

func (c *BooleanComparator) Compare(arr arrow.Array, i, j int) int

func (*BooleanComparator) CompareValues added in v0.3.1 

func (c *BooleanComparator) CompareValues(v1, v2 interface{}) int

type BoundaryType added in v0.2.0 

type BoundaryType int

BoundaryType represents the type of frame boundary. 

const (
	BoundaryUnboundedPreceding BoundaryType = iota
	BoundaryPreceding
	BoundaryCurrentRow
	BoundaryFollowing
	BoundaryUnboundedFollowing
)

type CaseExpr 

type CaseExpr struct {
	// contains filtered or unexported fields
}

CaseExpr represents a CASE expression with multiple WHEN clauses.

func Case 

func Case() *CaseExpr

Case creates a new CASE expression.

func (*CaseExpr) Else 

func (c *CaseExpr) Else(value Expr) *CaseExpr

Else sets the default value for the case expression.

func (*CaseExpr) ElseValue 

func (c *CaseExpr) ElseValue() Expr

func (*CaseExpr) String 

func (c *CaseExpr) String() string

func (*CaseExpr) SupportsContext added in v0.3.0 

func (c *CaseExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context CaseExpr context support depends on all its conditions and values.

func (*CaseExpr) Type 

func (c *CaseExpr) Type() Type

func (*CaseExpr) When 

func (c *CaseExpr) When(condition, value Expr) *CaseExpr

When adds a condition-value pair to the case expression.

func (*CaseExpr) Whens 

func (c *CaseExpr) Whens() []CaseWhen

type CaseWhen 

type CaseWhen struct {
	// contains filtered or unexported fields
}

CaseWhen represents a condition and value pair in CASE expression.

type ColumnExpr 

type ColumnExpr struct {
	// contains filtered or unexported fields
}

ColumnExpr represents a column reference.

func Col 

func Col(name string) *ColumnExpr

Col creates a column expression.

func (*ColumnExpr) Abs 

func (c *ColumnExpr) Abs() *FunctionExpr

Abs creates an absolute value function expression.

func (*ColumnExpr) Add 

func (c *ColumnExpr) Add(other Expr) *BinaryExpr

Add creates an addition expression.

func (*ColumnExpr) CastToBool 

func (c *ColumnExpr) CastToBool() *FunctionExpr

CastToBool creates a cast to bool function expression.

func (*ColumnExpr) CastToFloat64 

func (c *ColumnExpr) CastToFloat64() *FunctionExpr

CastToFloat64 creates a cast to float64 function expression.

func (*ColumnExpr) CastToInt64 

func (c *ColumnExpr) CastToInt64() *FunctionExpr

CastToInt64 creates a cast to int64 function expression.

func (*ColumnExpr) CastToString 

func (c *ColumnExpr) CastToString() *FunctionExpr

CastToString creates a cast to string function expression.

func (*ColumnExpr) Ceil 

func (c *ColumnExpr) Ceil() *FunctionExpr

Ceil creates a ceil function expression.

func (*ColumnExpr) Cos 

func (c *ColumnExpr) Cos() *FunctionExpr

Cos creates a cosine function expression.

func (*ColumnExpr) Count 

func (c *ColumnExpr) Count() *AggregationExpr

Count creates a count aggregation of this column.

func (*ColumnExpr) DateAdd added in v0.3.0 

func (c *ColumnExpr) DateAdd(intervalExpr *IntervalExpr) *FunctionExpr

DateAdd adds an interval to a date/time column.

func (*ColumnExpr) DateSub added in v0.3.0 

func (c *ColumnExpr) DateSub(intervalExpr *IntervalExpr) *FunctionExpr

DateSub subtracts an interval from a date/time column.

func (*ColumnExpr) Day added in v0.2.0 

func (c *ColumnExpr) Day() *FunctionExpr

Day creates a DAY function expression to extract day from date/time.

func (*ColumnExpr) Div 

func (c *ColumnExpr) Div(other Expr) *BinaryExpr

Div creates a division expression.

func (*ColumnExpr) Eq 

func (c *ColumnExpr) Eq(other Expr) *BinaryExpr

Eq creates an equality expression.

func (*ColumnExpr) Floor 

func (c *ColumnExpr) Floor() *FunctionExpr

Floor creates a floor function expression.

func (*ColumnExpr) Ge 

func (c *ColumnExpr) Ge(other Expr) *BinaryExpr

Ge creates a greater-than-or-equal expression.

func (*ColumnExpr) Gt 

func (c *ColumnExpr) Gt(other Expr) *BinaryExpr

Gt creates a greater-than expression.

func (*ColumnExpr) Hour added in v0.2.0 

func (c *ColumnExpr) Hour() *FunctionExpr

Hour creates an HOUR function expression to extract hour from timestamp.

func (*ColumnExpr) Le 

func (c *ColumnExpr) Le(other Expr) *BinaryExpr

Le creates a less-than-or-equal expression.

func (*ColumnExpr) Length 

func (c *ColumnExpr) Length() *FunctionExpr

Length creates a LENGTH function expression.

func (*ColumnExpr) Log 

func (c *ColumnExpr) Log() *FunctionExpr

Log creates a natural logarithm function expression.

func (*ColumnExpr) Lower 

func (c *ColumnExpr) Lower() *FunctionExpr

Lower creates a LOWER function expression.

func (*ColumnExpr) Lt 

func (c *ColumnExpr) Lt(other Expr) *BinaryExpr

Lt creates a less-than expression.

func (*ColumnExpr) Max 

func (c *ColumnExpr) Max() *AggregationExpr

Max creates a max aggregation of this column.

func (*ColumnExpr) Mean 

func (c *ColumnExpr) Mean() *AggregationExpr

Mean creates a mean aggregation of this column.

func (*ColumnExpr) Min 

func (c *ColumnExpr) Min() *AggregationExpr

Min creates a min aggregation of this column.

func (*ColumnExpr) Minute added in v0.2.0 

func (c *ColumnExpr) Minute() *FunctionExpr

Minute creates a MINUTE function expression to extract minute from timestamp.

func (*ColumnExpr) Month added in v0.2.0 

func (c *ColumnExpr) Month() *FunctionExpr

Month creates a MONTH function expression to extract month from date/time.

func (*ColumnExpr) Mul 

func (c *ColumnExpr) Mul(other Expr) *BinaryExpr

Mul creates a multiplication expression.

func (*ColumnExpr) Name 

func (c *ColumnExpr) Name() string

func (*ColumnExpr) Ne 

func (c *ColumnExpr) Ne(other Expr) *BinaryExpr

Ne creates a not-equal expression.

func (*ColumnExpr) Neg 

func (c *ColumnExpr) Neg() *UnaryExpr

Neg creates a negation (unary minus) expression.

func (*ColumnExpr) Not 

func (c *ColumnExpr) Not() *UnaryExpr

Not creates a logical NOT expression.

func (*ColumnExpr) Round 

func (c *ColumnExpr) Round() *FunctionExpr

Round creates a round function expression.

func (*ColumnExpr) RoundTo 

func (c *ColumnExpr) RoundTo(precision Expr) *FunctionExpr

RoundTo creates a round function expression with precision.

func (*ColumnExpr) Second added in v0.2.0 

func (c *ColumnExpr) Second() *FunctionExpr

Second creates a SECOND function expression to extract second from timestamp.

func (*ColumnExpr) Sin 

func (c *ColumnExpr) Sin() *FunctionExpr

Sin creates a sine function expression.

func (*ColumnExpr) Sqrt 

func (c *ColumnExpr) Sqrt() *FunctionExpr

Sqrt creates a square root function expression.

func (*ColumnExpr) String 

func (c *ColumnExpr) String() string

func (*ColumnExpr) Sub 

func (c *ColumnExpr) Sub(other Expr) *BinaryExpr

Sub creates a subtraction expression.

func (*ColumnExpr) Substring 

func (c *ColumnExpr) Substring(start, length Expr) *FunctionExpr

Substring creates a SUBSTRING function expression.

func (*ColumnExpr) Sum 

func (c *ColumnExpr) Sum() *AggregationExpr

Sum creates a sum aggregation of this column.

func (*ColumnExpr) SupportsContext added in v0.3.0 

func (c *ColumnExpr) SupportsContext(_ EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context ColumnExpr can be evaluated in both contexts (but meaning differs).

func (*ColumnExpr) Trim 

func (c *ColumnExpr) Trim() *FunctionExpr

Trim creates a TRIM function expression.

func (*ColumnExpr) Type 

func (c *ColumnExpr) Type() Type

func (*ColumnExpr) Upper 

func (c *ColumnExpr) Upper() *FunctionExpr

Upper creates an UPPER function expression.

func (*ColumnExpr) Year added in v0.2.0 

func (c *ColumnExpr) Year() *FunctionExpr

Year creates a YEAR function expression to extract year from date/time.

type Comparator added in v0.3.1 

type Comparator interface {
	Compare(arr arrow.Array, i, j int) int
	CompareValues(v1, v2 interface{}) int
}

Comparator interface for type-specific comparison.

type ContextualExpr added in v0.3.0 

type ContextualExpr interface {
	Expr
	// SupportsContext returns whether this expression can be evaluated in the given context
	SupportsContext(ctx EvaluationContext) bool
}

ContextualExpr represents expressions that can be evaluated in different contexts.

type EvaluationContext added in v0.3.0 

type EvaluationContext int

EvaluationContext represents the context in which an expression is evaluated. 

const (
	// RowContext evaluates expressions against raw row data (for WHERE clauses).
	RowContext EvaluationContext = iota

	// GroupContext evaluates expressions against aggregated group data (for HAVING clauses).
	GroupContext
)

func (EvaluationContext) String added in v0.3.0 

func (ec EvaluationContext) String() string

type Evaluator 

type Evaluator struct {
	// contains filtered or unexported fields
}

Evaluator evaluates expressions against Arrow arrays.

func NewEvaluator 

func NewEvaluator(mem memory.Allocator) *Evaluator

NewEvaluator creates a new expression evaluator.

func (*Evaluator) Evaluate 

func (e *Evaluator) Evaluate(expr Expr, columns map[string]arrow.Array) (arrow.Array, error)

Evaluate evaluates an expression that returns a value array (numeric, string, etc.)

func (*Evaluator) EvaluateBoolean 

func (e *Evaluator) EvaluateBoolean(expr Expr, columns map[string]arrow.Array) (arrow.Array, error)

EvaluateBoolean evaluates an expression that should return a boolean array.

func (*Evaluator) EvaluateBooleanWithContext added in v0.3.0 

func (e *Evaluator) EvaluateBooleanWithContext(
	expr Expr,
	columns map[string]arrow.Array,
	ctx EvaluationContext,
) (arrow.Array, error)

EvaluateBooleanWithContext evaluates an expression that should return a boolean array with context.

func (*Evaluator) EvaluateWindow added in v0.2.0 

func (e *Evaluator) EvaluateWindow(expr *WindowExpr, columns map[string]arrow.Array) (arrow.Array, error)

EvaluateWindow evaluates a window expression.

func (*Evaluator) EvaluateWithContext added in v0.3.0 

func (e *Evaluator) EvaluateWithContext(
	expr Expr,
	columns map[string]arrow.Array,
	ctx EvaluationContext,
) (arrow.Array, error)

EvaluateWithContext evaluates an expression with a specific evaluation context.

type Expr 

type Expr interface {
	Type() Type
	String() string
}

Expr represents an expression that can be evaluated lazily.

type Float64Comparator added in v0.3.1 

type Float64Comparator struct {
	// contains filtered or unexported fields
}

Float64Comparator provides optimized float64 comparison.

func (*Float64Comparator) Compare added in v0.3.1 

func (c *Float64Comparator) Compare(arr arrow.Array, i, j int) int

func (*Float64Comparator) CompareValues added in v0.3.1 

func (c *Float64Comparator) CompareValues(v1, v2 interface{}) int

type FrameBoundary added in v0.2.0 

type FrameBoundary struct {
	// contains filtered or unexported fields
}

FrameBoundary represents a frame boundary.

func CurrentRow added in v0.2.0 

func CurrentRow() *FrameBoundary

CurrentRow creates a current row boundary.

func Following added in v0.2.0 

func Following(offset int) *FrameBoundary

Following creates a following boundary with offset.

func Preceding added in v0.2.0 

func Preceding(offset int) *FrameBoundary

Preceding creates a preceding boundary with offset.

func UnboundedFollowing added in v0.2.0 

func UnboundedFollowing() *FrameBoundary

UnboundedFollowing creates an unbounded following boundary.

func UnboundedPreceding added in v0.2.0 

func UnboundedPreceding() *FrameBoundary

UnboundedPreceding creates an unbounded preceding boundary.

func (*FrameBoundary) String added in v0.2.0 

func (b *FrameBoundary) String() string

String returns the string representation of the boundary.

type FrameType added in v0.2.0 

type FrameType int

FrameType represents the type of window frame. 

const (
	FrameTypeRows FrameType = iota
	FrameTypeRange
	FrameTypeGroups
)

type FunctionExpr 

type FunctionExpr struct {
	// contains filtered or unexported fields
}

FunctionExpr represents a function call expression.

func Coalesce 

func Coalesce(exprs ...Expr) *FunctionExpr

Coalesce creates a COALESCE function expression.

func Concat 

func Concat(exprs ...Expr) *FunctionExpr

Concat creates a CONCAT function expression.

func DateAdd added in v0.3.0 

func DateAdd(dateExpr Expr, intervalExpr *IntervalExpr) *FunctionExpr

DateAdd creates a DATE_ADD function expression to add interval to date/time.

func DateDiff added in v0.3.0 

func DateDiff(startDate, endDate Expr, unit string) *FunctionExpr

DateDiff creates a DATE_DIFF function expression to calculate difference between dates.

func DateSub added in v0.3.0 

func DateSub(dateExpr Expr, intervalExpr *IntervalExpr) *FunctionExpr

DateSub creates a DATE_SUB function expression to subtract interval from date/time.

func Day added in v0.2.0 

func Day(expr Expr) *FunctionExpr

Day creates a DAY function expression.

func Hour added in v0.2.0 

func Hour(expr Expr) *FunctionExpr

Hour creates an HOUR function expression.

func If 

func If(condition, thenValue, elseValue Expr) *FunctionExpr

If creates an IF function expression.

func Minute added in v0.2.0 

func Minute(expr Expr) *FunctionExpr

Minute creates a MINUTE function expression.

func Month added in v0.2.0 

func Month(expr Expr) *FunctionExpr

Month creates a MONTH function expression.

func NewFunction added in v0.3.0 

func NewFunction(name string, args ...Expr) *FunctionExpr

NewFunction creates a function expression.

func Second added in v0.2.0 

func Second(expr Expr) *FunctionExpr

Second creates a SECOND function expression.

func Year added in v0.2.0 

func Year(expr Expr) *FunctionExpr

Year creates a YEAR function expression.

func (*FunctionExpr) Abs 

func (f *FunctionExpr) Abs() *FunctionExpr

Abs creates an absolute value function expression.

func (*FunctionExpr) Add 

func (f *FunctionExpr) Add(other Expr) *BinaryExpr

Add creates an addition expression for function expressions.

func (*FunctionExpr) And 

func (f *FunctionExpr) And(other Expr) *BinaryExpr

func (*FunctionExpr) Args 

func (f *FunctionExpr) Args() []Expr

func (*FunctionExpr) CastToBool 

func (f *FunctionExpr) CastToBool() *FunctionExpr

func (*FunctionExpr) CastToFloat64 

func (f *FunctionExpr) CastToFloat64() *FunctionExpr

func (*FunctionExpr) CastToInt64 

func (f *FunctionExpr) CastToInt64() *FunctionExpr

func (*FunctionExpr) CastToString 

func (f *FunctionExpr) CastToString() *FunctionExpr

CastToString creates a string casting function expression.

func (*FunctionExpr) Ceil 

func (f *FunctionExpr) Ceil() *FunctionExpr

func (*FunctionExpr) Cos 

func (f *FunctionExpr) Cos() *FunctionExpr

func (*FunctionExpr) DateAdd added in v0.3.0 

func (f *FunctionExpr) DateAdd(intervalExpr *IntervalExpr) *FunctionExpr

DateAdd adds an interval to a date/time function result.

func (*FunctionExpr) DateSub added in v0.3.0 

func (f *FunctionExpr) DateSub(intervalExpr *IntervalExpr) *FunctionExpr

DateSub subtracts an interval from a date/time function result.

func (*FunctionExpr) Day added in v0.2.0 

func (f *FunctionExpr) Day() *FunctionExpr

Day creates a DAY function expression to extract day from date/time.

func (*FunctionExpr) Div 

func (f *FunctionExpr) Div(other Expr) *BinaryExpr

func (*FunctionExpr) Eq 

func (f *FunctionExpr) Eq(other Expr) *BinaryExpr

func (*FunctionExpr) Floor 

func (f *FunctionExpr) Floor() *FunctionExpr

func (*FunctionExpr) Ge 

func (f *FunctionExpr) Ge(other Expr) *BinaryExpr

func (*FunctionExpr) Gt 

func (f *FunctionExpr) Gt(other Expr) *BinaryExpr

func (*FunctionExpr) Hour added in v0.2.0 

func (f *FunctionExpr) Hour() *FunctionExpr

Hour creates an HOUR function expression to extract hour from timestamp.

func (*FunctionExpr) Le 

func (f *FunctionExpr) Le(other Expr) *BinaryExpr

func (*FunctionExpr) Length 

func (f *FunctionExpr) Length() *FunctionExpr

func (*FunctionExpr) Log 

func (f *FunctionExpr) Log() *FunctionExpr

func (*FunctionExpr) Lower 

func (f *FunctionExpr) Lower() *FunctionExpr

func (*FunctionExpr) Lt 

func (f *FunctionExpr) Lt(other Expr) *BinaryExpr

func (*FunctionExpr) Minute added in v0.2.0 

func (f *FunctionExpr) Minute() *FunctionExpr

Minute creates a MINUTE function expression to extract minute from timestamp.

func (*FunctionExpr) Month added in v0.2.0 

func (f *FunctionExpr) Month() *FunctionExpr

Month creates a MONTH function expression to extract month from date/time.

func (*FunctionExpr) Mul 

func (f *FunctionExpr) Mul(other Expr) *BinaryExpr

func (*FunctionExpr) Name 

func (f *FunctionExpr) Name() string

func (*FunctionExpr) Ne 

func (f *FunctionExpr) Ne(other Expr) *BinaryExpr

func (*FunctionExpr) Neg 

func (f *FunctionExpr) Neg() *UnaryExpr

Neg creates a negation expression for function expressions.

func (*FunctionExpr) Not 

func (f *FunctionExpr) Not() *UnaryExpr

func (*FunctionExpr) Or 

func (f *FunctionExpr) Or(other Expr) *BinaryExpr

func (*FunctionExpr) Round 

func (f *FunctionExpr) Round() *FunctionExpr

func (*FunctionExpr) RoundTo 

func (f *FunctionExpr) RoundTo(precision Expr) *FunctionExpr

func (*FunctionExpr) Second added in v0.2.0 

func (f *FunctionExpr) Second() *FunctionExpr

Second creates a SECOND function expression to extract second from timestamp.

func (*FunctionExpr) Sin 

func (f *FunctionExpr) Sin() *FunctionExpr

func (*FunctionExpr) Sqrt 

func (f *FunctionExpr) Sqrt() *FunctionExpr

func (*FunctionExpr) String 

func (f *FunctionExpr) String() string

func (*FunctionExpr) Sub 

func (f *FunctionExpr) Sub(other Expr) *BinaryExpr

func (*FunctionExpr) Substring 

func (f *FunctionExpr) Substring(start, length Expr) *FunctionExpr

func (*FunctionExpr) SupportsContext added in v0.3.0 

func (f *FunctionExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context FunctionExpr context support depends on its arguments.

func (*FunctionExpr) Trim 

func (f *FunctionExpr) Trim() *FunctionExpr

func (*FunctionExpr) Type 

func (f *FunctionExpr) Type() Type

func (*FunctionExpr) Upper 

func (f *FunctionExpr) Upper() *FunctionExpr

Upper creates an uppercase function expression.

func (*FunctionExpr) Year added in v0.2.0 

func (f *FunctionExpr) Year() *FunctionExpr

Year creates a YEAR function expression to extract year from date/time.

type HavingValidator added in v0.3.0 

type HavingValidator struct {
	// contains filtered or unexported fields
}

HavingValidator provides validation for HAVING clause expressions. It ensures that HAVING expressions only reference columns that are available in GROUP BY results (either GROUP BY columns or aggregated columns). It supports comprehensive alias resolution for user-defined and default names.

func BuildHavingValidatorWithAlias added in v0.3.0 

func BuildHavingValidatorWithAlias(
	groupByColumns []string,
	aggregations []*AggregationExpr,
	caseInsensitive bool,
) (*HavingValidator, error)

BuildHavingValidatorWithAlias creates a complete HAVING validator with alias resolution from GROUP BY columns and aggregation expressions. This is a convenience function that sets up both the AggregationContext and AliasResolver automatically.

func NewHavingValidator added in v0.3.0 

func NewHavingValidator(aggregationContext *AggregationContext, groupByColumns []string) *HavingValidator

NewHavingValidator creates a new HavingValidator with the given aggregation context and GROUP BY columns. For backward compatibility.

func NewHavingValidatorWithAlias added in v0.3.0 

func NewHavingValidatorWithAlias(
	aggregationContext *AggregationContext,
	groupByColumns []string,
	aliasResolver *AliasResolver,
) *HavingValidator

NewHavingValidatorWithAlias creates a new HavingValidator with enhanced alias resolution. This is the recommended constructor for full HAVING clause functionality.

func (*HavingValidator) GetAvailableAggregations added in v0.3.0 

func (hv *HavingValidator) GetAvailableAggregations() []string

GetAvailableAggregations returns all aggregation expressions available in the context.

func (*HavingValidator) GetAvailableColumns added in v0.3.0 

func (hv *HavingValidator) GetAvailableColumns() []string

GetAvailableColumns returns all columns available in HAVING context (GROUP BY columns + aggregated columns). Uses alias resolver when available.

func (*HavingValidator) ValidateExpression added in v0.3.0 

func (hv *HavingValidator) ValidateExpression(expr Expr) error

ValidateExpression recursively validates a HAVING expression to ensure all column references are valid (either GROUP BY columns or aggregated columns).

type Int64Comparator added in v0.3.1 

type Int64Comparator struct {
	// contains filtered or unexported fields
}

Int64Comparator provides optimized int64 comparison.

func (*Int64Comparator) Compare added in v0.3.1 

func (c *Int64Comparator) Compare(arr arrow.Array, i, j int) int

func (*Int64Comparator) CompareValues added in v0.3.1 

func (c *Int64Comparator) CompareValues(v1, v2 interface{}) int

type IntervalExpr added in v0.3.0 

type IntervalExpr struct {
	// contains filtered or unexported fields
}

IntervalExpr represents a time interval value and type.

func Days added in v0.3.0 

func Days(value int64) *IntervalExpr

Days creates an interval representing days.

func Hours added in v0.3.0 

func Hours(value int64) *IntervalExpr

Hours creates an interval representing hours.

func Minutes added in v0.3.0 

func Minutes(value int64) *IntervalExpr

Minutes creates an interval representing minutes.

func Months added in v0.3.0 

func Months(value int64) *IntervalExpr

Months creates an interval representing months.

func Years added in v0.3.0 

func Years(value int64) *IntervalExpr

Years creates an interval representing years.

func (*IntervalExpr) IntervalType added in v0.3.0 

func (i *IntervalExpr) IntervalType() IntervalType

func (*IntervalExpr) String added in v0.3.0 

func (i *IntervalExpr) String() string

func (*IntervalExpr) SupportsContext added in v0.3.0 

func (i *IntervalExpr) SupportsContext(_ EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context IntervalExpr can be evaluated in any context (it's a literal-like expression).

func (*IntervalExpr) Type added in v0.3.0 

func (i *IntervalExpr) Type() Type

func (*IntervalExpr) Value added in v0.3.0 

func (i *IntervalExpr) Value() int64

type IntervalType added in v0.3.0 

type IntervalType int

IntervalType represents different types of time intervals. 

const (
	IntervalDays IntervalType = iota
	IntervalHours
	IntervalMinutes
	IntervalMonths
	IntervalYears
)

type InvalidExpr 

type InvalidExpr struct {
	// contains filtered or unexported fields
}

InvalidExpr represents an invalid expression with an error message.

func Invalid 

func Invalid(message string) *InvalidExpr

Invalid creates an invalid expression with an error message.

func (*InvalidExpr) Message 

func (i *InvalidExpr) Message() string

func (*InvalidExpr) String 

func (i *InvalidExpr) String() string

func (*InvalidExpr) SupportsContext added in v0.3.0 

func (i *InvalidExpr) SupportsContext(_ EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context InvalidExpr cannot be evaluated in any context.

func (*InvalidExpr) Type 

func (i *InvalidExpr) Type() Type

type LiteralExpr 

type LiteralExpr struct {
	// contains filtered or unexported fields
}

LiteralExpr represents a literal value.

func Lit 

func Lit(value interface{}) *LiteralExpr

Lit creates a literal expression.

func (*LiteralExpr) String 

func (l *LiteralExpr) String() string

func (*LiteralExpr) SupportsContext added in v0.3.0 

func (l *LiteralExpr) SupportsContext(_ EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context LiteralExpr can be evaluated in any context.

func (*LiteralExpr) Type 

func (l *LiteralExpr) Type() Type

func (*LiteralExpr) Value 

func (l *LiteralExpr) Value() interface{}

type OrderByExpr added in v0.2.0 

type OrderByExpr struct {
	// contains filtered or unexported fields
}

OrderByExpr represents a column ordering specification.

type SortKey added in v0.3.1 

type SortKey struct {
	// contains filtered or unexported fields
}

SortKey represents pre-computed sort keys for a row.

type StringComparator added in v0.3.1 

type StringComparator struct {
	// contains filtered or unexported fields
}

StringComparator provides optimized string comparison.

func (*StringComparator) Compare added in v0.3.1 

func (c *StringComparator) Compare(arr arrow.Array, i, j int) int

func (*StringComparator) CompareValues added in v0.3.1 

func (c *StringComparator) CompareValues(v1, v2 interface{}) int

type Type added in v0.3.1 

type Type int

Type represents the type of expression. 

const (
	ExprColumn Type = iota
	ExprLiteral
	ExprBinary
	ExprUnary
	ExprFunction
	ExprAggregation
	ExprCase
	ExprInvalid
)
const (
	ExprWindow Type = iota + 100
	ExprWindowFunction
)

Add new expression types for window functions.

type UnaryExpr 

type UnaryExpr struct {
	// contains filtered or unexported fields
}

UnaryExpr represents a unary operation.

func (*UnaryExpr) Op 

func (u *UnaryExpr) Op() UnaryOp

func (*UnaryExpr) Operand 

func (u *UnaryExpr) Operand() Expr

func (*UnaryExpr) String 

func (u *UnaryExpr) String() string

func (*UnaryExpr) SupportsContext added in v0.3.0 

func (u *UnaryExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context UnaryExpr context support depends on its operand.

func (*UnaryExpr) Type 

func (u *UnaryExpr) Type() Type

type UnaryOp 

type UnaryOp int

UnaryOp represents unary operations. 

const (
	UnaryNeg UnaryOp = iota
	UnaryNot
)

type WindowExpr added in v0.2.0 

type WindowExpr struct {
	// contains filtered or unexported fields
}

WindowExpr represents a window function expression.

func (*WindowExpr) String added in v0.2.0 

func (w *WindowExpr) String() string

String returns the string representation.

func (*WindowExpr) SupportsContext added in v0.3.0 

func (w *WindowExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context WindowExpr typically requires RowContext as it operates on row-level data with window semantics.

func (*WindowExpr) Type added in v0.2.0 

func (w *WindowExpr) Type() Type

Type returns the expression type.

type WindowFrame added in v0.2.0 

type WindowFrame struct {
	// contains filtered or unexported fields
}

WindowFrame represents the frame specification for window functions.

func Between added in v0.2.0 

func Between(start, end *FrameBoundary) *WindowFrame

Between creates a window frame between two boundaries.

func (*WindowFrame) String added in v0.2.0 

func (f *WindowFrame) String() string

String returns the string representation of the frame.

type WindowFunctionExpr added in v0.2.0 

type WindowFunctionExpr struct {
	// contains filtered or unexported fields
}

WindowFunctionExpr represents a window-specific function (ROW_NUMBER, RANK, etc.)

func CumeDist added in v0.3.1 

func CumeDist() *WindowFunctionExpr

CumeDist creates a CUME_DIST() window function.

func DenseRank added in v0.2.0 

func DenseRank() *WindowFunctionExpr

DenseRank creates a DENSE_RANK() window function.

func FirstValue added in v0.2.0 

func FirstValue(expr Expr) *WindowFunctionExpr

FirstValue creates a FIRST_VALUE() window function.

func Lag added in v0.2.0 

func Lag(expr Expr, offset int) *WindowFunctionExpr

Lag creates a LAG() window function.

func LastValue added in v0.2.0 

func LastValue(expr Expr) *WindowFunctionExpr

LastValue creates a LAST_VALUE() window function.

func Lead added in v0.2.0 

func Lead(expr Expr, offset int) *WindowFunctionExpr

Lead creates a LEAD() window function.

func NthValue added in v0.3.1 

func NthValue(expr Expr, n int) *WindowFunctionExpr

NthValue creates a NTH_VALUE() window function.

func Ntile added in v0.3.1 

func Ntile(buckets int) *WindowFunctionExpr

Ntile creates a NTILE() window function.

func PercentRank added in v0.3.1 

func PercentRank() *WindowFunctionExpr

PercentRank creates a PERCENT_RANK() window function.

func Rank added in v0.2.0 

func Rank() *WindowFunctionExpr

Rank creates a RANK() window function.

func RowNumber added in v0.2.0 

func RowNumber() *WindowFunctionExpr

RowNumber creates a ROW_NUMBER() window function.

func (*WindowFunctionExpr) Over added in v0.2.0 

func (w *WindowFunctionExpr) Over(window *WindowSpec) *WindowExpr

Over creates a window expression with the specified window.

func (*WindowFunctionExpr) String added in v0.2.0 

func (w *WindowFunctionExpr) String() string

String returns the string representation.

func (*WindowFunctionExpr) SupportsContext added in v0.3.0 

func (w *WindowFunctionExpr) SupportsContext(ctx EvaluationContext) bool

SupportsContext returns whether this expression can be evaluated in the given context WindowFunctionExpr can only be evaluated in RowContext.

func (*WindowFunctionExpr) Type added in v0.2.0 

func (w *WindowFunctionExpr) Type() Type

Type returns the expression type.

type WindowParallelConfig added in v0.3.0 

type WindowParallelConfig struct {
	// Minimum number of partitions to trigger parallel execution
	MinPartitionsForParallel int
	// Minimum rows per partition to trigger parallel sorting
	MinRowsForParallelSort int
	// Maximum number of workers for window operations
	MaxWorkers int
	// Enable adaptive parallelization based on data characteristics
	AdaptiveParallelization bool
}

WindowParallelConfig holds configuration for parallel window function execution.

func DefaultWindowParallelConfig added in v0.3.0 

func DefaultWindowParallelConfig() *WindowParallelConfig

DefaultWindowParallelConfig returns the default configuration for parallel window execution.

type WindowSpec added in v0.2.0 

type WindowSpec struct {
	// contains filtered or unexported fields
}

WindowSpec represents a window specification for window functions.

func NewWindow added in v0.2.0 

func NewWindow() *WindowSpec

NewWindow creates a new window specification.

func (*WindowSpec) Groups added in v0.3.1 

func (w *WindowSpec) Groups(frame *WindowFrame) *WindowSpec

Groups sets a GROUPS frame for the window.

func (*WindowSpec) OrderBy added in v0.2.0 

func (w *WindowSpec) OrderBy(column string, ascending bool) *WindowSpec

OrderBy adds an ordering specification to the window.

func (*WindowSpec) PartitionBy added in v0.2.0 

func (w *WindowSpec) PartitionBy(columns ...string) *WindowSpec

PartitionBy sets the partition columns for the window.

func (*WindowSpec) Range added in v0.2.0 

func (w *WindowSpec) Range(frame *WindowFrame) *WindowSpec

Range sets a RANGE frame for the window.

func (*WindowSpec) Rows added in v0.2.0 

func (w *WindowSpec) Rows(frame *WindowFrame) *WindowSpec

Rows sets a ROWS frame for the window.

func (*WindowSpec) String added in v0.2.0 

func (w *WindowSpec) String() string

String returns the string representation of the window spec.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.