Method TestPreparedStatementsCache

internal/testsuite/sql_suite.go:91–181

()

}

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

	sess.SetPreparedStatementCache(true)
	defer sess.SetPreparedStatementCache(false)

	var tMu sync.Mutex
	tFatal := func(err error) {
		tMu.Lock()
		defer tMu.Unlock()

		s.T().Errorf("tmu: %v", err)
	}

	// This limit was chosen because, by default, MySQL accepts 16k statements
	// and dies. See https://github.com/upper/db/issues/287
	limit := 20000

	if detectrace.WithRace() {
		// When running this test under the Go race detector we quickly reach the limit
		// of 8128 alive goroutines it can handle, so we set it to a safer number.
		//
		// Note that in order to fully stress this feature you'll have to run this
		// test without the race detector.
		limit = 100
	}

	var wg sync.WaitGroup

	for i := 0; i < limit; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()

			// This query is different on each iteration and generates a new
			// prepared statement everytime it's called.
			res := sess.Collection("artist").Find().Select(db.Raw(fmt.Sprintf("count(%d) AS c", i)))

			var count map[string]uint64
			err := res.One(&count)
			if err != nil {
				tFatal(err)
			}
		}(i)
	}
	wg.Wait()

	// Concurrent Insert can open many connections on MySQL / PostgreSQL, this
	// sets a limit on them.
	sess.SetMaxOpenConns(90)

	switch s.Adapter() {
	case "ql":
		limit = 1000
	}

	for i := 0; i < limit; i++ {
		wg.Add(1)