#9: Add lockForUpdate

builder.go

@@ -11,18 +11,19 @@ import (
 
 // inner type to build qualified sql
 type builder struct {
-	where      string
-	whereNamed map[string]interface{}
-	table      string
-	from       string
-	join       []string
-	orderBy    map[string]string
-	groupBy    string
-	having     string
-	columns    []string
-	union      []string
-	offset     int64
-	limit      int64
+	where         string
+	whereNamed    map[string]interface{}
+	table         string
+	from          string
+	join          []string
+	orderBy       map[string]string
+	groupBy       string
+	having        string
+	columns       []string
+	union         []string
+	offset        int64
+	limit         int64
+	lockForUpdate *string
 }
 
 // DB is an entity that composite builder and Conn types
@@ -353,6 +354,13 @@ func (r *DB) From(fromTbl string) *DB {
 	return r
 }
 
+func (r *DB) LockForUpdate() *DB {
+	str := " FOR UPDATE"
+	r.Builder.lockForUpdate = &str
+
+	return r
+}
+
 // Dump prints raw sql to stdout
 func (r *DB) Dump() {
 	log.SetOutput(os.Stdout)

pgsql_builder.go

@@ -106,6 +106,10 @@ func (r *builder) buildSelect() string {
 		query += " OFFSET " + strconv.FormatInt(r.offset, 10)
 	}
 
+	if r.lockForUpdate != nil {
+		query += *r.lockForUpdate
+	}
+
 	return query
 }