Adding Verb method to InsertBuilder

Fix #2

Co-authored-by: Henrique Vicente <[email protected]>
Co-authored-by: neokami <[email protected]>

insert.go

@@ -12,7 +12,7 @@ import (
 // InsertBuilder builds SQL INSERT statements.
 type InsertBuilder struct {
 	prefixes      []SQLizer
-	replace       bool
+	verb          string
 	into          string
 	columns       []string
 	values        [][]any
@@ -21,6 +21,12 @@ type InsertBuilder struct {
 	selectBuilder *SelectBuilder
 }
 
+// Verb to be used for the operation (default: INSERT).
+func (b InsertBuilder) Verb(v string) InsertBuilder {
+	b.verb = v
+	return b
+}
+
 // SQL builds the query into a SQL string and bound args.
 func (b InsertBuilder) SQL() (sqlStr string, args []any, err error) {
 	if b.into == "" {
@@ -43,10 +49,10 @@ func (b InsertBuilder) SQL() (sqlStr string, args []any, err error) {
 		sql.WriteString(" ")
 	}
 
-	if !b.replace {
-		sql.WriteString("INSERT ")
+	if b.verb != "" {
+		sql.WriteString(b.verb + " ")
 	} else {
-		sql.WriteString("REPLACE ")
+		sql.WriteString("INSERT ")
 	}
 
 	sql.WriteString("INTO ")

insert_test.go

@@ -227,3 +227,19 @@ func TestInsertBuilderReplace(t *testing.T) {
 		t.Errorf("expected SQL to be %q, got %q instead", want, sql)
 	}
 }
+
+func TestInsertBuilderVerb(t *testing.T) {
+	t.Parallel()
+	b := Insert("table").Verb("REPLACE").Values(1)
+
+	want := "REPLACE INTO table VALUES ($1)"
+
+	sql, _, err := b.SQL()
+	if err != nil {
+		t.Errorf("unexpected error: %v", err)
+	}
+
+	if want != sql {
+		t.Errorf("expected SQL to be %q, got %q instead", want, sql)
+	}
+}

statement.go

@@ -59,7 +59,7 @@ func Insert(into string) InsertBuilder {
 // See InsertBuilder.Into.
 func Replace(into string) InsertBuilder {
 	builder := InsertBuilder{}
-	builder.replace = true
+	builder.verb = "REPLACE"
 	return builder.Into(into)
 }