scheduler: add simple benchmark for tasksUpdated (#16422)

In preperation for some refactoring to tasksUpdated, add a benchmark to the
old code so it's easy to compare with the changes, making sure nothing goes
off the rails for performance.
This commit is contained in:
Seth Hoenig 2023-03-13 10:44:14 -05:00 committed by GitHub
parent b3cec771d6
commit 630bd8eb68
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 101 additions and 0 deletions

View file

@ -568,3 +568,93 @@ func MaxParallelJob() *structs.Job {
job.Canonicalize()
return job
}
// BigBenchmarkJob creates a job with many fields set, ideal for benchmarking
// stuff involving jobs.
//
// Should not be used outside of benchmarking - folks should feel free to add
// more fields without risk of breaking test cases down the line.
func BigBenchmarkJob() *structs.Job {
job := MultiTaskGroupJob()
// job affinities
job.Affinities = structs.Affinities{{
LTarget: "left",
RTarget: "right",
Operand: "!=",
Weight: 100,
}, {
LTarget: "a",
RTarget: "b",
Operand: "<",
Weight: 50,
}}
// job spreads
job.Spreads = []*structs.Spread{{
Attribute: "foo.x",
Weight: 100,
SpreadTarget: []*structs.SpreadTarget{{
Value: "x",
Percent: 90,
}, {
Value: "x2",
Percent: 99,
}},
}, {
Attribute: "foo.y",
Weight: 90,
SpreadTarget: []*structs.SpreadTarget{{
Value: "y",
Percent: 10,
}},
}}
// group affinities
job.TaskGroups[0].Affinities = structs.Affinities{{
LTarget: "L",
RTarget: "R",
Operand: "!=",
Weight: 100,
}, {
LTarget: "b",
RTarget: "a",
Operand: ">",
Weight: 50,
}}
// group spreads
job.TaskGroups[0].Spreads = []*structs.Spread{{
Attribute: "bar.x",
Weight: 100,
SpreadTarget: []*structs.SpreadTarget{{
Value: "x",
Percent: 90,
}, {
Value: "x2",
Percent: 99,
}},
}, {
Attribute: "bar.y",
Weight: 90,
SpreadTarget: []*structs.SpreadTarget{{
Value: "y",
Percent: 10,
}},
}}
// task affinities
job.TaskGroups[0].Tasks[0].Affinities = structs.Affinities{{
LTarget: "Left",
RTarget: "Right",
Operand: "!=",
Weight: 100,
}, {
LTarget: "A",
RTarget: "B",
Operand: "<",
Weight: 50,
}}
return job
}

View file

@ -17,6 +17,17 @@ import (
"github.com/hashicorp/nomad/nomad/structs"
)
func BenchmarkTasksUpdated(b *testing.B) {
jobA := mock.BigBenchmarkJob()
jobB := jobA.Copy()
group := jobA.TaskGroups[0].Name
for n := 0; n < b.N; n++ {
if tasksUpdated(jobA, jobB, group) {
b.Fatal("tasks should be the same")
}
}
}
func newNode(name string) *structs.Node {
n := mock.Node()
n.Name = name