Function BenchmarkJoin

engine/seminaivebottomup_test.go:1462–1492

(b *testing.B)

}

func BenchmarkJoin(b *testing.B) {
	b.ReportAllocs()
	for j := 0; j < b.N; j++ {
		// Given two relations has_num(Thing, Num) has_prop(Thing, Property),
		// we select from one and join with the other.
		program := []ast.Clause{
			clause(`foo(X, Prop) :- has_num(X, Num), Num <= 1, has_prop(X, Prop).`),
		}
		store := factstore.NewIndexedInMemoryStore()
		for i := 0; i < 1_000_000; i++ {
			id, _ := ast.Name(fmt.Sprintf("/thing/%d", i))
			store.Add(ast.NewAtom("has_num", id, ast.Number(int64(i%3))))
			store.Add(ast.NewAtom("has_prop", id, ast.String(fmt.Sprintf("/property/%d", i))))
		}
		programInfo, err := analysis.AnalyzeOneUnit(parse.SourceUnit{Clauses: program}, asMap(store.ListPredicates()))
		if err != nil {
			b.Fatal(fmt.Errorf("analysis: %w", err))
		}
		strata, predToStratum, err := analysis.Stratify(analysis.Program{
			EdbPredicates: programInfo.EdbPredicates,
			IdbPredicates: programInfo.IdbPredicates,
			Rules:         programInfo.Rules,
		})
		if err != nil {
			b.Fatal(fmt.Errorf("stratification: %w", err))
		}
		if _, err := EvalStratifiedProgramWithStats(programInfo, strata, predToStratum, store); err != nil {
			b.Fatal(fmt.Errorf("evaluation: %w", err))
		}
	}
}