Method TestBatchInsert

internal/testsuite/sql_suite.go:1274–1315

()

}

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

	for batchSize := 0; batchSize < 17; batchSize++ {
		err := sess.Collection("artist").Truncate()
		s.NoError(err)

		q := sess.SQL().InsertInto("artist").Columns("name")

		switch s.Adapter() {
		case "postgresql", "cockroachdb":
			q = q.Amend(func(query string) string {
				return query + ` ON CONFLICT DO NOTHING`
			})
		}

		batch := q.Batch(batchSize)

		totalItems := int(rand.Int31n(21))

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

		err = batch.Wait()
		s.NoError(err)
		s.NoError(batch.Err())

		c, err := sess.Collection("artist").Find().Count()
		s.NoError(err)
		s.Equal(uint64(totalItems), c)

		for i := 0; i < totalItems; i++ {
			c, err := sess.Collection("artist").Find(db.Cond{"name": fmt.Sprintf("artist-%d", i)}).Count()
			s.NoError(err)
			s.Equal(uint64(1), c)
		}
	}
}

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