blob: 57f2cf2530a84759e9278c30c114fe089d63beb6 [file] [log] [blame]
// Copyright 2019 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 postgres
import (
"context"
"testing"
"golang.org/x/pkgsite/internal/config"
"golang.org/x/pkgsite/internal/database"
)
var testQueries = []string{
"github",
"cloud",
"golang",
"go",
"mutex",
"elasticsearch",
"errors",
"kubernetes",
"github golang",
"hashicorp",
"test",
"teest",
"imports",
"net",
"s3blob",
"k8s",
}
func BenchmarkSearch(b *testing.B) {
ctx := context.Background()
cfg, err := config.Init(ctx)
if err != nil {
b.Fatal(err)
}
ddb, err := database.Open("postgres", cfg.DBConnInfo(), "bench")
if err != nil {
b.Fatal(err)
}
db := New(ddb)
searchers := map[string]func(context.Context, string, SearchOptions) ([]*SearchResult, error){
"db.Search": db.Search,
}
for name, search := range searchers {
for _, query := range testQueries {
b.Run(name+":"+query, func(b *testing.B) {
for i := 0; i < b.N; i++ {
if _, err := search(ctx, query, SearchOptions{MaxResults: 10, MaxResultCount: 100}); err != nil {
b.Fatal(err)
}
}
})
}
}
}