Method TestBatchInsertReturningKeys

internal/testsuite/sql_suite.go:1354–1404

()

}

func (s *SQLTestSuite) TestBatchInsertReturningKeys() {
	switch s.Adapter() {
	case "postgresql", "cockroachdb":
		// pass
	default:
		s.T().Skip("Currently not supported.")
		return
	}

	sess := s.Session()

	err := sess.Collection("artist").Truncate()
	s.NoError(err)

	batchSize, totalItems := 7, 12

	batch := sess.SQL().InsertInto("artist").Columns("name").Returning("id").Batch(batchSize)

	go func() {
		defer batch.Done()
		for i := 0; i < totalItems; i++ {
			batch.Values(fmt.Sprintf("artist-%d", i))
		}
	}()

	var keyMap []struct {
		ID int `db:"id"`
	}
	for batch.NextResult(&keyMap) {
		// Each insertion must produce new keys.
		s.True(len(keyMap) > 0)
		s.True(len(keyMap) <= batchSize)

		// Find the elements we've just inserted
		keys := make([]int, 0, len(keyMap))
		for i := range keyMap {
			keys = append(keys, keyMap[i].ID)
		}

		// Make sure count matches.
		c, err := sess.Collection("artist").Find(db.Cond{"id": keys}).Count()
		s.NoError(err)
		s.Equal(uint64(len(keyMap)), c)
	}
	s.NoError(batch.Err())

	// Count all new elements
	c, err := sess.Collection("artist").Find().Count()
	s.NoError(err)
	s.Equal(uint64(totalItems), c)
}

func (s *SQLTestSuite) TestPaginator() {
	sess := s.Session()