Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions insert.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,16 @@ func (q *InsertBuilder) Values(values ...interface{}) *InsertBuilder {
return q
}

// SubQuery add values from a subquery to the insert query.
func (q *InsertBuilder) SubQuery(v SelectQuery) *InsertBuilder {
if len(v.Fields()) != len(q.fields) {
panic(`Number of values has to match the number of fields`)
}

q.values = append(q.values, []Field{MakeField(v)})
return q
}

// Upsert turns the INSERT query into an upsert query, only usable if your driver supports it
func (q *InsertBuilder) Upsert(query Query, conflict ...Field) *InsertBuilder {
if q.ignoreConflict {
Expand Down
22 changes: 22 additions & 0 deletions internal/tests/tests_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -67,10 +67,12 @@ func runTests(t *testing.T) {
if driver == `msqb.Driver` {
testUpsertSeparate(t)
testInsert(t)
testInsertSubQuery(t)
testUpsertSeparate(t)
} else {
testUpsert(t)
testInsert(t)
testInsertSubQuery(t)
testUpsert(t)
}
if driver == `pgqb.Driver` {
Expand Down Expand Up @@ -179,6 +181,26 @@ func testInsert(test *testing.T) {
tx.MustCommit()
}

func testInsertSubQuery(test *testing.T) {
o := model.One()
sq := o.Select(o.ID, qb.Value(1), qb.Value(`Test comment sq`))

t := model.Two()

tx := db.MustBegin()

q := t.Insert(t.OneID, t.Number, t.Comment).
SubQuery(sq).
Values(1, 2, `Test comment 2`)

res := tx.MustExec(q)

assert := assert.New(test)
assert.Eq(int64(2), res.MustRowsAffected())

tx.MustCommit()
}

func testUpdate(test *testing.T) {
t := model.Two()

Expand Down
Loading