Add and implement QueryBuilder.InsertStatement()

Author: jrauh01

database/query_builder.go

@@ -1,11 +1,18 @@
 package database
 
 import (
+	"fmt"
 	"github.com/icinga/icinga-go-library/strcase"
 	"github.com/jmoiron/sqlx/reflectx"
+	"slices"
+	"strings"
 )
 
-type QueryBuilder interface{}
+type QueryBuilder interface {
+	InsertStatement(stmt InsertStatement) string
+
+	BuildColumns(entity Entity, columns []string, excludedColumns []string) []string
+}
 
 func NewQueryBuilder(driver string) QueryBuilder {
 	return &queryBuilder{
@@ -18,3 +25,38 @@ type queryBuilder struct {
 	driver    string
 	columnMap ColumnMap
 }
+
+func (qb *queryBuilder) InsertStatement(stmt InsertStatement) string {
+	columns := qb.BuildColumns(stmt.Entity(), stmt.Columns(), stmt.ExcludedColumns())
+	into := stmt.Table()
+	if into == "" {
+		into = TableName(stmt.Entity())
+	}
+
+	return fmt.Sprintf(
+		`INSERT INTO "%s" ("%s") VALUES (%s)`,
+		into,
+		strings.Join(columns, `", "`),
+		fmt.Sprintf(":%s", strings.Join(columns, ", :")),
+	)
+}
+
+func (qb *queryBuilder) BuildColumns(entity Entity, columns []string, excludedColumns []string) []string {
+	var deltaColumns []string
+	if len(columns) > 0 {
+		deltaColumns = columns
+	} else {
+		deltaColumns = qb.columnMap.Columns(entity)
+	}
+
+	if len(excludedColumns) > 0 {
+		deltaColumns = slices.DeleteFunc(
+			deltaColumns,
+			func(column string) bool {
+				return slices.Contains(excludedColumns, column)
+			},
+		)
+	}
+
+	return deltaColumns[:len(deltaColumns):len(deltaColumns)]
+}