Postgres refactor.

postgres/insert_statement.go

@@ -0,0 +1,73 @@
+package postgres
+
+import "github.com/go-jet/jet/internal/jet"
+
+// InsertStatement is interface for SQL INSERT statements
+type InsertStatement interface {
+	jet.Statement
+
+	// Insert row of values
+	VALUES(value interface{}, values ...interface{}) InsertStatement
+	// Insert row of values, where value for each column is extracted from filed of structure data.
+	// If data is not struct or there is no field for every column selected, this method will panic.
+	MODEL(data interface{}) InsertStatement
+
+	MODELS(data interface{}) InsertStatement
+
+	QUERY(selectStatement SelectStatement) InsertStatement
+
+	RETURNING(projections ...jet.Projection) InsertStatement
+}
+
+func newInsertStatement(table WritableTable, columns []jet.IColumn) InsertStatement {
+	newInsert := &insertStatementImpl{}
+	newInsert.StatementImpl = jet.NewStatementImpl(Dialect, jet.DeleteStatementType, newInsert,
+		&newInsert.Insert, &newInsert.Values, &newInsert.Select, &newInsert.Returning)
+
+	newInsert.Insert.Table = table
+	newInsert.Insert.Columns = columns
+
+	return newInsert
+}
+
+type insertStatementImpl struct {
+	jet.StatementImpl
+
+	Insert    jet.ClauseInsert
+	Values    jet.ClauseValues
+	Select    jet.ClauseQuery
+	Returning jet.ClauseReturning
+}
+
+func (i *insertStatementImpl) VALUES(value interface{}, values ...interface{}) InsertStatement {
+	i.Values.Rows = append(i.Values.Rows, jet.UnwindRowFromValues(value, values))
+	return i
+}
+
+func (i *insertStatementImpl) MODEL(data interface{}) InsertStatement {
+	i.Values.Rows = append(i.Values.Rows, jet.UnwindRowFromModel(i.getColumns(), data))
+	return i
+}
+
+func (i *insertStatementImpl) MODELS(data interface{}) InsertStatement {
+	i.Values.Rows = append(i.Values.Rows, jet.UnwindRowsFromModels(i.getColumns(), data)...)
+	return i
+}
+
+func (i *insertStatementImpl) RETURNING(projections ...jet.Projection) InsertStatement {
+	i.Returning.Projections = projections
+	return i
+}
+
+func (i *insertStatementImpl) QUERY(selectStatement SelectStatement) InsertStatement {
+	i.Select.Query = selectStatement
+	return i
+}
+
+func (i *insertStatementImpl) getColumns() []jet.IColumn {
+	if len(i.Insert.Columns) > 0 {
+		return i.Insert.Columns
+	}
+
+	return i.Insert.Table.Columns()
+}