mirror of
https://github.com/Jguer/yay.git
synced 2026-07-16 18:58:36 -04:00
19cd422953
* query: precompute metricScore once per result Normalize the search term and name/provider/desc at the metric boundary, drop the per-comparison distance cache, and score each result via prepareMetrics before sorting. Adds a search-flow benchmark to track regressions. * update deps
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
//go:build !integration
|
|
|
|
package query
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Jguer/aur"
|
|
|
|
"github.com/Jguer/yay/v13/pkg/db/mock"
|
|
mockaur "github.com/Jguer/yay/v13/pkg/dep/mock"
|
|
"github.com/Jguer/yay/v13/pkg/settings/parser"
|
|
"github.com/Jguer/yay/v13/pkg/text"
|
|
)
|
|
|
|
func buildSearchFlowBenchmarkData(repoCount, aurCount int) (*mock.DBExecutor, *mockaur.MockAUR, *text.Logger, []string) {
|
|
logger := text.NewLogger(io.Discard, io.Discard, strings.NewReader(""), false, "bench")
|
|
extraDB := mock.NewDB("extra")
|
|
communityDB := mock.NewDB("community")
|
|
|
|
repoPkgs := make([]mock.IPackage, 0, repoCount)
|
|
for i := range repoCount {
|
|
db := extraDB
|
|
if i%2 == 1 {
|
|
db = communityDB
|
|
}
|
|
|
|
repoPkgs = append(repoPkgs, &mock.Package{
|
|
PBase: fmt.Sprintf("yay-tool-%04d", i),
|
|
PName: fmt.Sprintf("yay-tool-%04d", i),
|
|
PVersion: fmt.Sprintf("1.%d.0", i%17),
|
|
PDescription: fmt.Sprintf("Yet another yay tool package %04d for search benchmarking", i),
|
|
PSize: int64(1024 + i),
|
|
PISize: int64(2048 + i),
|
|
PDB: db,
|
|
PArchitecture: "x86_64",
|
|
PProvides: mock.DependList{Depends: []mock.Depend{{Name: fmt.Sprintf("yay-provider-%04d", i)}, {Name: "yay"}}},
|
|
})
|
|
}
|
|
|
|
aurPkgs := make([]aur.Pkg, 0, aurCount)
|
|
for i := range aurCount {
|
|
aurPkgs = append(aurPkgs, aur.Pkg{
|
|
Description: fmt.Sprintf("Yet another yay helper package %04d with search benchmark metadata", i),
|
|
FirstSubmitted: 1_500_000_000 + i,
|
|
ID: 2_000_000 + i,
|
|
LastModified: 1_760_000_000 + i,
|
|
Maintainer: "bench",
|
|
Name: fmt.Sprintf("yay-aur-%04d", i),
|
|
NumVotes: 100 + (i % 250),
|
|
OutOfDate: 0,
|
|
PackageBase: fmt.Sprintf("yay-aur-%04d", i),
|
|
PackageBaseID: 300_000 + i,
|
|
Popularity: 1.0 + float64(i%100)/10,
|
|
URL: "https://example.invalid/yay",
|
|
URLPath: "/snapshot.tar.gz",
|
|
Version: fmt.Sprintf("2.%d.0", i%23),
|
|
Provides: []string{"yay"},
|
|
})
|
|
}
|
|
|
|
mockDB := &mock.DBExecutor{
|
|
ReposFn: func() []string {
|
|
return []string{"extra", "community"}
|
|
},
|
|
SyncPackagesFn: func(pkgs ...string) []mock.IPackage {
|
|
return repoPkgs
|
|
},
|
|
LocalPackageFn: func(string) mock.IPackage {
|
|
return nil
|
|
},
|
|
}
|
|
|
|
mockAUR := &mockaur.MockAUR{
|
|
GetFn: func(ctx context.Context, query *aur.Query) ([]aur.Pkg, error) {
|
|
return aurPkgs, nil
|
|
},
|
|
}
|
|
|
|
return mockDB, mockAUR, logger, []string{"yay"}
|
|
}
|
|
|
|
func BenchmarkExecuteSearchFlowLarge(b *testing.B) {
|
|
mockDB, mockAUR, logger, searchTerms := buildSearchFlowBenchmarkData(1500, 1500)
|
|
|
|
b.ReportAllocs()
|
|
for b.Loop() {
|
|
qb := NewSourceQueryBuilder(
|
|
mockAUR,
|
|
logger,
|
|
"",
|
|
parser.ModeAny,
|
|
"",
|
|
false,
|
|
false,
|
|
true,
|
|
)
|
|
qb.Execute(b.Context(), mockDB, searchTerms)
|
|
}
|
|
}
|