43fc77ba99
* Add Int4 and Int8 integer expression to distinguish int4range and int8range types * Add range functions to range expressions to avoid go template appearing in sql * Compact range integration tests and add range update tests
156 lines
4.9 KiB
Go
156 lines
4.9 KiB
Go
package jet
|
|
|
|
// IntegerExpression interface
|
|
type IntegerExpression interface {
|
|
Expression
|
|
numericExpression
|
|
|
|
EQ(rhs IntegerExpression) BoolExpression
|
|
NOT_EQ(rhs IntegerExpression) BoolExpression
|
|
IS_DISTINCT_FROM(rhs IntegerExpression) BoolExpression
|
|
IS_NOT_DISTINCT_FROM(rhs IntegerExpression) BoolExpression
|
|
|
|
LT(rhs IntegerExpression) BoolExpression
|
|
LT_EQ(rhs IntegerExpression) BoolExpression
|
|
GT(rhs IntegerExpression) BoolExpression
|
|
GT_EQ(rhs IntegerExpression) BoolExpression
|
|
BETWEEN(min, max IntegerExpression) BoolExpression
|
|
NOT_BETWEEN(min, max IntegerExpression) BoolExpression
|
|
|
|
ADD(rhs IntegerExpression) IntegerExpression
|
|
SUB(rhs IntegerExpression) IntegerExpression
|
|
MUL(rhs IntegerExpression) IntegerExpression
|
|
DIV(rhs IntegerExpression) IntegerExpression
|
|
MOD(rhs IntegerExpression) IntegerExpression
|
|
POW(rhs IntegerExpression) IntegerExpression
|
|
|
|
BIT_AND(rhs IntegerExpression) IntegerExpression
|
|
BIT_OR(rhs IntegerExpression) IntegerExpression
|
|
BIT_XOR(rhs IntegerExpression) IntegerExpression
|
|
BIT_SHIFT_LEFT(shift IntegerExpression) IntegerExpression
|
|
BIT_SHIFT_RIGHT(shift IntegerExpression) IntegerExpression
|
|
}
|
|
|
|
// additional integer expression subtypes, used in range expressions.
|
|
type (
|
|
Int4Expression IntegerExpression
|
|
Int8Expression IntegerExpression
|
|
)
|
|
|
|
type integerInterfaceImpl struct {
|
|
numericExpressionImpl
|
|
parent IntegerExpression
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) EQ(rhs IntegerExpression) BoolExpression {
|
|
return Eq(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) NOT_EQ(rhs IntegerExpression) BoolExpression {
|
|
return NotEq(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) IS_DISTINCT_FROM(rhs IntegerExpression) BoolExpression {
|
|
return IsDistinctFrom(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) IS_NOT_DISTINCT_FROM(rhs IntegerExpression) BoolExpression {
|
|
return IsNotDistinctFrom(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) GT(rhs IntegerExpression) BoolExpression {
|
|
return Gt(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) GT_EQ(rhs IntegerExpression) BoolExpression {
|
|
return GtEq(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) LT(rhs IntegerExpression) BoolExpression {
|
|
return Lt(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) LT_EQ(rhs IntegerExpression) BoolExpression {
|
|
return LtEq(i.parent, rhs)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BETWEEN(min, max IntegerExpression) BoolExpression {
|
|
return NewBetweenOperatorExpression(i.parent, min, max, false)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) NOT_BETWEEN(min, max IntegerExpression) BoolExpression {
|
|
return NewBetweenOperatorExpression(i.parent, min, max, true)
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) ADD(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(Add(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) SUB(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(Sub(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) MUL(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(Mul(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) DIV(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(Div(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) MOD(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(Mod(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) POW(rhs IntegerExpression) IntegerExpression {
|
|
return IntExp(POW(i.parent, rhs))
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BIT_AND(rhs IntegerExpression) IntegerExpression {
|
|
return newBinaryIntegerOperatorExpression(i.parent, rhs, "&")
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BIT_OR(rhs IntegerExpression) IntegerExpression {
|
|
return newBinaryIntegerOperatorExpression(i.parent, rhs, "|")
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BIT_XOR(rhs IntegerExpression) IntegerExpression {
|
|
return newBinaryIntegerOperatorExpression(i.parent, rhs, "#")
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BIT_SHIFT_LEFT(intExpression IntegerExpression) IntegerExpression {
|
|
return newBinaryIntegerOperatorExpression(i.parent, intExpression, "<<")
|
|
}
|
|
|
|
func (i *integerInterfaceImpl) BIT_SHIFT_RIGHT(intExpression IntegerExpression) IntegerExpression {
|
|
return newBinaryIntegerOperatorExpression(i.parent, intExpression, ">>")
|
|
}
|
|
|
|
func newBinaryIntegerOperatorExpression(lhs, rhs IntegerExpression, operator string) IntegerExpression {
|
|
return IntExp(NewBinaryOperatorExpression(lhs, rhs, operator))
|
|
}
|
|
|
|
func newPrefixIntegerOperatorExpression(expression IntegerExpression, operator string) IntegerExpression {
|
|
return IntExp(newPrefixOperatorExpression(expression, operator))
|
|
}
|
|
|
|
type integerExpressionWrapper struct {
|
|
integerInterfaceImpl
|
|
|
|
Expression
|
|
}
|
|
|
|
func newIntExpressionWrap(expression Expression) IntegerExpression {
|
|
intExpressionWrap := integerExpressionWrapper{Expression: expression}
|
|
|
|
intExpressionWrap.integerInterfaceImpl.parent = &intExpressionWrap
|
|
|
|
return &intExpressionWrap
|
|
}
|
|
|
|
// IntExp is int expression wrapper around arbitrary expression.
|
|
// Allows go compiler to see any expression as int expression.
|
|
// Does not add sql cast to generated sql builder output.
|
|
func IntExp(expression Expression) IntegerExpression {
|
|
return newIntExpressionWrap(expression)
|
|
}
|