Add LOCK table support.

sqlbuilder/lock_statement_test.go

@@ -0,0 +1,24 @@
+package sqlbuilder
+
+import (
+	"gotest.tools/assert"
+	"testing"
+)
+
+func TestLockSingleTable(t *testing.T) {
+	lock := table1.LOCK().IN(LOCK_ROW_SHARE)
+
+	queryStr, _, err := lock.Sql()
+
+	assert.NilError(t, err)
+	assert.Equal(t, queryStr, `LOCK TABLE db.table1 IN ROW SHARE MODE`)
+}
+
+func TestLockMultipleTable(t *testing.T) {
+	lock := LOCK(table2, table1).IN(LOCK_ACCESS_EXCLUSIVE).NOWAIT()
+
+	queryStr, _, err := lock.Sql()
+
+	assert.NilError(t, err)
+	assert.Equal(t, queryStr, `LOCK TABLE db.table2, db.table1 IN ACCESS EXCLUSIVE MODE NOWAIT`)
+}