aboutsummaryrefslogtreecommitdiff
path: root/gopls/internal/regtest/bench/workspace_symbols_test.go
blob: 975422ac6517f3bf46d74c6913265dd1f50965b5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// Copyright 2022 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package bench

import (
	"flag"
	"fmt"
	"testing"
)

var symbolQuery = flag.String("symbol_query", "test", "symbol query to use in benchmark")

// BenchmarkWorkspaceSymbols benchmarks the time to execute a workspace symbols
// request (controlled by the -symbol_query flag).
func BenchmarkWorkspaceSymbols(b *testing.B) {
	for name := range repos {
		b.Run(name, func(b *testing.B) {
			env := getRepo(b, name).sharedEnv(b)
			symbols := env.Symbol(*symbolQuery) // warm the cache

			if testing.Verbose() {
				fmt.Println("Results:")
				for i, symbol := range symbols {
					fmt.Printf("\t%d. %s (%s)\n", i, symbol.Name, symbol.ContainerName)
				}
			}

			b.ResetTimer()

			for i := 0; i < b.N; i++ {
				env.Symbol(*symbolQuery)
			}
		})
	}
}