support ReferTable

Author: wjhuang2016

expression/column.go

@@ -123,6 +123,11 @@ func (col *CorrelatedColumn) IsCorrelated() bool {
 	return true
 }
 
+// ReferTable implements Expression interface.
+func (col *CorrelatedColumn) ReferTable() bool {
+	return true
+}
+
 // Decorrelate implements Expression interface.
 func (col *CorrelatedColumn) Decorrelate(schema *Schema) Expression {
 	if !schema.Contains(&col.Column) {
@@ -297,6 +302,11 @@ func (col *Column) IsCorrelated() bool {
 	return false
 }
 
+// ReferTable implements Expression interface.
+func (col *Column) ReferTable() bool {
+	return true
+}
+
 // Decorrelate implements Expression interface.
 func (col *Column) Decorrelate(_ *Schema) Expression {
 	return col

expression/column_test.go

@@ -43,6 +43,7 @@ func (s *testEvaluatorSuite) TestColumn(c *C) {
 	c.Assert(corCol.Equal(nil, corCol), IsTrue)
 	c.Assert(corCol.Equal(nil, invalidCorCol), IsFalse)
 	c.Assert(corCol.IsCorrelated(), IsTrue)
+	c.Assert(corCol.ReferTable(), IsTrue)
 	c.Assert(corCol.Decorrelate(schema).Equal(nil, col), IsTrue)
 	c.Assert(invalidCorCol.Decorrelate(schema).Equal(nil, invalidCorCol), IsTrue)
 

expression/constant.go

@@ -311,6 +311,11 @@ func (c *Constant) IsCorrelated() bool {
 	return false
 }
 
+// ReferTable implements Expression interface.
+func (c *Constant) ReferTable() bool {
+	return false
+}
+
 // Decorrelate implements Expression interface.
 func (c *Constant) Decorrelate(_ *Schema) Expression {
 	return c

expression/expression.go

@@ -79,6 +79,9 @@ type Expression interface {
 	// IsCorrelated checks if this expression has correlated key.
 	IsCorrelated() bool
 
+	// ReferTable checks if this expression refers to a table.
+	ReferTable() bool
+
 	// Decorrelate try to decorrelate the expression by schema.
 	Decorrelate(schema *Schema) Expression
 

expression/expression_test.go

@@ -61,6 +61,7 @@ func (s *testEvaluatorSuite) TestConstant(c *C) {
 
 	sc := &stmtctx.StatementContext{TimeZone: time.Local}
 	c.Assert(Zero.IsCorrelated(), IsFalse)
+	c.Assert(Zero.ReferTable(), IsFalse)
 	c.Assert(Zero.Decorrelate(nil).Equal(s.ctx, Zero), IsTrue)
 	c.Assert(Zero.HashCode(sc), DeepEquals, []byte{0x0, 0x8, 0x0})
 	c.Assert(Zero.Equal(s.ctx, One), IsFalse)

expression/scalar_function.go

@@ -165,6 +165,16 @@ func (sf *ScalarFunction) IsCorrelated() bool {
 	return false
 }
 
+// ReferTable implements Expression interface.
+func (sf *ScalarFunction) ReferTable() bool {
+	for _, arg := range sf.GetArgs() {
+		if arg.ReferTable() {
+			return true
+		}
+	}
+	return false
+}
+
 // Decorrelate implements Expression interface.
 func (sf *ScalarFunction) Decorrelate(schema *Schema) Expression {
 	for i, arg := range sf.GetArgs() {

expression/scalar_function_test.go

@@ -40,6 +40,7 @@ func (s *testEvaluatorSuite) TestScalarFunction(c *C) {
 	c.Assert(err, IsNil)
 	c.Assert(res, DeepEquals, []byte{0x22, 0x6c, 0x74, 0x28, 0x66, 0x65, 0x69, 0x2e, 0x68, 0x61, 0x6e, 0x2c, 0x20, 0x31, 0x29, 0x22})
 	c.Assert(sf.IsCorrelated(), IsFalse)
+	c.Assert(sf.ReferTable(), IsTrue)
 	c.Assert(sf.Decorrelate(nil).Equal(s.ctx, sf), IsTrue)
 	c.Assert(sf.HashCode(sc), DeepEquals, []byte{0x3, 0x4, 0x6c, 0x74, 0x1, 0x80, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x1, 0x0, 0x5, 0xbf, 0xf0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0})