Add StatementBuilder.Where method

Author: lann

statement.go

@@ -41,6 +41,13 @@ func (b StatementBuilderType) RunWith(runner BaseRunner) StatementBuilderType {
 	return setRunWith(b, runner).(StatementBuilderType)
 }
 
+// Where adds WHERE expressions to the query.
+//
+// See SelectBuilder.Where for more information.
+func (b StatementBuilderType) Where(pred interface{}, args ...interface{}) StatementBuilderType {
+	return builder.Append(b, "WhereParts", newWherePart(pred, args...)).(StatementBuilderType)
+}
+
 // StatementBuilder is a parent builder for other builders, e.g. SelectBuilder.
 var StatementBuilder = StatementBuilderType(builder.EmptyBuilder).PlaceholderFormat(Question)
 

statement_test.go

@@ -66,3 +66,16 @@ func TestRunWithBaseRunnerQueryRowError(t *testing.T) {
 	assert.Error(t, RunnerNotQueryRunner, sb.Select("test").QueryRow().Scan(nil))
 
 }
+
+func TestStatementBuilderWhere(t *testing.T) {
+	sb := StatementBuilder.Where("x = ?", 1)
+
+	sql, args, err := sb.Select("test").Where("y = ?", 2).ToSql()
+	assert.NoError(t, err)
+
+	expectedSql := "SELECT test WHERE x = ? AND y = ?"
+	assert.Equal(t, expectedSql, sql)
+
+	expectedArgs := []interface{}{1, 2}
+	assert.Equal(t, expectedArgs, args)
+}