feat: implement add/sub col expression

xwjdsh · Jul 20, 2022 · e6d7c96 · e6d7c96
1 parent 56b9edf
commit e6d7c96
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/field/export_test.go b/field/export_test.go
@@ -98,6 +98,20 @@ func TestExpr_Build(t *testing.T) {
 			Result:       "(UNIX_TIMESTAMP(?))*?",
 			ExpectedVars: []interface{}{"2005-03-27 03:00:00", uint64(99)},
 		},
+		{
+			Expr:   field.NewInt("t1", "id").AddCol(field.NewInt("t2", "num")),
+			Result: "`t1`.`id` + `t2`.`num`",
+		},
+		{
+			Expr:         field.NewInt("t1", "id").AddCol(field.NewInt("t2", "num").Add(1)),
+			Result:       "`t1`.`id` + `t2`.`num`+?",
+			ExpectedVars: []interface{}{int(1)},
+		},
+		{
+			Expr:         field.NewInt("t1", "id").EqCol(field.NewInt("t1", "id").AddCol(field.NewInt("t2", "num").Add(1))),
+			Result:       "`t1`.`id` = `t1`.`id` + `t2`.`num`+?",
+			ExpectedVars: []interface{}{int(1)},
+		},
 		// ======================== integer ========================
 		{
 			Expr:   field.NewUint("", "id"),

diff --git a/field/expr.go b/field/expr.go
@@ -214,6 +214,15 @@ func (e expr) SetCol(col Expr) AssignExpr {
 	return e.setE(clause.Eq{Column: e.col.Name, Value: col.RawExpr()})
 }
 
+// ======================== operate columns ========================
+func (e expr) AddCol(col Expr) Expr {
+	return e.setE(clause.Expr{SQL: "? + ?", Vars: []interface{}{e.RawExpr(), col.RawExpr()}})
+}
+
+func (e expr) SubCol(col Expr) Expr {
+	return e.setE(clause.Expr{SQL: "? - ?", Vars: []interface{}{e.RawExpr(), col.RawExpr()}})
+}
+
 // ======================== keyword ========================
 func (e expr) As(alias string) Expr {
 	if e.e != nil {