Skip to content

Commit

Permalink
more expression2
Browse files Browse the repository at this point in the history
  • Loading branch information
max-hoffman committed Mar 14, 2024
1 parent 7663ec0 commit dbe960c
Showing 1 changed file with 21 additions and 21 deletions.
42 changes: 21 additions & 21 deletions go/libraries/doltcore/sqle/dolt_diff_summary_table_function.go
Expand Up @@ -16,12 +16,12 @@ package sqle

import (
"fmt"
"github.com/dolthub/go-mysql-server/sql/expression"
"io"
"sort"
"strings"

"github.com/dolthub/go-mysql-server/sql"
expression2 "github.com/dolthub/go-mysql-server/sql/expression"
"github.com/dolthub/go-mysql-server/sql/types"

"github.com/dolthub/dolt/go/libraries/doltcore/diff"
Expand Down Expand Up @@ -199,12 +199,12 @@ func (ds *DiffSummaryTableFunction) Expressions() []sql.Expression {
}

// WithExpressions implements the sql.Expressioner interface.
func (ds *DiffSummaryTableFunction) WithExpressions(expression ...sql.Expression) (sql.Node, error) {
if len(expression) < 1 {
return nil, sql.ErrInvalidArgumentNumber.New(ds.Name(), "1 to 3", len(expression))
func (ds *DiffSummaryTableFunction) WithExpressions(exprs ...sql.Expression) (sql.Node, error) {
if len(exprs) < 1 {
return nil, sql.ErrInvalidArgumentNumber.New(ds.Name(), "1 to 3", len(exprs))
}

for _, expr := range expression {
for _, expr := range exprs {
if !expr.Resolved() {
return nil, ErrInvalidNonLiteralArgument.New(ds.Name(), expr.String())
}
Expand All @@ -215,41 +215,41 @@ func (ds *DiffSummaryTableFunction) WithExpressions(expression ...sql.Expression
}

newDstf := *ds
if strings.Contains(expression[0].String(), "..") {
if len(expression) < 1 || len(expression) > 2 {
return nil, sql.ErrInvalidArgumentNumber.New(newDstf.Name(), "1 or 2", len(expression))
if strings.Contains(exprs[0].String(), "..") {
if len(exprs) < 1 || len(exprs) > 2 {
return nil, sql.ErrInvalidArgumentNumber.New(newDstf.Name(), "1 or 2", len(exprs))
}
newDstf.dotCommitExpr = expression[0]
if len(expression) == 2 {
newDstf.tableNameExpr = expression[1]
newDstf.dotCommitExpr = exprs[0]
if len(exprs) == 2 {
newDstf.tableNameExpr = exprs[1]
}
} else {
if len(expression) < 2 || len(expression) > 3 {
return nil, sql.ErrInvalidArgumentNumber.New(newDstf.Name(), "2 or 3", len(expression))
if len(exprs) < 2 || len(exprs) > 3 {
return nil, sql.ErrInvalidArgumentNumber.New(newDstf.Name(), "2 or 3", len(exprs))
}
newDstf.fromCommitExpr = expression[0]
newDstf.toCommitExpr = expression[1]
if len(expression) == 3 {
newDstf.tableNameExpr = expression[2]
newDstf.fromCommitExpr = exprs[0]
newDstf.toCommitExpr = exprs[1]
if len(exprs) == 3 {
newDstf.tableNameExpr = exprs[2]
}
}

// validate the expressions
if newDstf.dotCommitExpr != nil {
if !types.IsText(newDstf.dotCommitExpr.Type()) && !expression2.IsBindVar(newDstf.dotCommitExpr) {
if !types.IsText(newDstf.dotCommitExpr.Type()) && !expression.IsBindVar(newDstf.dotCommitExpr) {
return nil, sql.ErrInvalidArgumentDetails.New(newDstf.Name(), newDstf.dotCommitExpr.String())
}
} else {
if !types.IsText(newDstf.fromCommitExpr.Type()) && !expression2.IsBindVar(newDstf.fromCommitExpr) {
if !types.IsText(newDstf.fromCommitExpr.Type()) && !expression.IsBindVar(newDstf.fromCommitExpr) {
return nil, sql.ErrInvalidArgumentDetails.New(newDstf.Name(), newDstf.fromCommitExpr.String())
}
if !types.IsText(newDstf.toCommitExpr.Type()) && !expression2.IsBindVar(newDstf.toCommitExpr) {
if !types.IsText(newDstf.toCommitExpr.Type()) && !expression.IsBindVar(newDstf.toCommitExpr) {
return nil, sql.ErrInvalidArgumentDetails.New(newDstf.Name(), newDstf.toCommitExpr.String())
}
}

if newDstf.tableNameExpr != nil {
if !types.IsText(newDstf.tableNameExpr.Type()) && !expression2.IsBindVar(newDstf.tableNameExpr) {
if !types.IsText(newDstf.tableNameExpr.Type()) && !expression.IsBindVar(newDstf.tableNameExpr) {
return nil, sql.ErrInvalidArgumentDetails.New(newDstf.Name(), newDstf.tableNameExpr.String())
}
}
Expand Down

0 comments on commit dbe960c

Please sign in to comment.