jet/sqlbuilder/expression.go

package sqlbuilder

import (
	"bytes"
	"github.com/dropbox/godropbox/database/sqltypes"
	"github.com/dropbox/godropbox/errors"
)

// An expression
type Expression interface {
	Clause

	As(alias string) Clause
	IsDistinct(expression Expression) BoolExpression
	IsNull(expression Expression) BoolExpression
}

type expressionInterfaceImpl struct {
	parent Expression
}

func (e *expressionInterfaceImpl) As(alias string) Clause {
	return NewAlias(e.parent, alias)
}

func (e *expressionInterfaceImpl) IsDistinct(expression Expression) BoolExpression {
	return nil
}

func (e *expressionInterfaceImpl) IsNull(expression Expression) BoolExpression {
	return nil
}

// Representation of binary operations (e.g. comparisons, arithmetic)
type binaryExpression struct {
	expressionInterfaceImpl
	lhs, rhs Expression
	operator []byte
}

func NewBinaryExpression(lhs, rhs Expression, operator []byte, parent ...Expression) *binaryExpression {
	binaryExpression := binaryExpression{
		lhs:      lhs,
		rhs:      rhs,
		operator: operator,
	}
	if len(parent) > 0 {
		binaryExpression.parent = parent[0]
	}

	return &binaryExpression
}

func (c *binaryExpression) SerializeSql(out *bytes.Buffer) (err error) {
	if c.lhs == nil {
		return errors.Newf("nil lhs.  Generated sql: %s", out.String())
	}
	if err = c.lhs.SerializeSql(out); err != nil {
		return
	}

	_, _ = out.Write(c.operator)

	if c.rhs == nil {
		return errors.Newf("nil rhs.  Generated sql: %s", out.String())
	}
	if err = c.rhs.SerializeSql(out); err != nil {
		return
	}

	return nil
}

// A not expression which negates a expression value
type prefixExpression struct {
	expressionInterfaceImpl

	expression Expression
	operator   []byte
}

func NewPrefixExpression(expression Expression, operator []byte, parent ...Expression) *prefixExpression {
	prefixExpression := prefixExpression{
		expression: expression,
		operator:   operator,
	}
	if len(parent) > 0 {
		prefixExpression.parent = parent[0]
	}

	return &prefixExpression
}

func (p *prefixExpression) SerializeSql(out *bytes.Buffer) (err error) {
	_, _ = out.Write(p.operator)

	if p.expression == nil {
		return errors.Newf("nil prefix expression.  Generated sql: %s", out.String())
	}
	if err = p.expression.SerializeSql(out); err != nil {
		return
	}

	return nil
}

// Representation of n-ary conjunctions (AND/OR)
type conjunctExpression struct {
	expressionInterfaceImpl
	expressions []BoolExpression
	conjunction []byte
}

func (conj *conjunctExpression) SerializeSql(out *bytes.Buffer) (err error) {
	if len(conj.expressions) == 0 {
		return errors.Newf(
			"Empty conjunction.  Generated sql: %s",
			out.String())
	}

	clauses := make([]Clause, len(conj.expressions), len(conj.expressions))
	for i, expr := range conj.expressions {
		clauses[i] = expr
	}

	useParentheses := len(clauses) > 1
	if useParentheses {
		_ = out.WriteByte('(')
	}

	if err = serializeClauses(clauses, conj.conjunction, out); err != nil {
		return
	}

	if useParentheses {
		_ = out.WriteByte(')')
	}

	return nil
}

//--------------------------------------------------------------

// Representation of an escaped literal
type literalExpression struct {
	expressionInterfaceImpl
	value sqltypes.Value
}

func NewLiteralExpression(value sqltypes.Value) *literalExpression {
	exp := literalExpression{value: value}
	exp.expressionInterfaceImpl.parent = &exp

	return &exp
}

func (c literalExpression) SerializeSql(out *bytes.Buffer) error {
	sqltypes.Value(c.value).EncodeSql(out)
	return nil
}