Add support "SELECT FOR UPDATE" to orm. Resolve issue #2157

2016-09-12 20:07:30 +00:00
parent dd0f05b1f1
commit 11247d41a7
6 changed files with 36 additions and 5 deletions
--- a/orm/qb_tidb.go
+++ b/orm/qb_tidb.go
@@ -31,6 +31,12 @@ func (qb *TiDBQueryBuilder) Select(fields ...string) QueryBuilder {
 	return qb
 }

+// ForUpdate add the FOR UPDATE clause
+func (qb *TiDBQueryBuilder) ForUpdate() QueryBuilder {
+	qb.Tokens = append(qb.Tokens, "FOR UPDATE")
+	return qb
+}
+
 // From join the tables
 func (qb *TiDBQueryBuilder) From(tables ...string) QueryBuilder {
 	qb.Tokens = append(qb.Tokens, "FROM", strings.Join(tables, CommaSpace))