73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
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()
|
|
}
|