mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 18:20:27 +00:00
3d3aed2bb3
Squash of: * We have to keep all the container hostnames until the end so we can count how many we've filtered * Adding tests for ContainerHostnameRenderer and PodServiceRenderer with filters * Because we filter on image name we need the image name before filtering * Alternative approach to passing decorators. * Refactor out some of the decorator capture * Don't memoise decorated calls to Render * Fixing filtered counts on containers topology Tricky, because we need the filters to be silent sometimes (when they're in the middle), but not when they're at the top, so we take the "top" filter's stats. However, this means we have to compose all user-specified filters into a single Filter layer, so we can get all stats. There are no more Silent filters, as all filters are silent (unless they are at the top). Additionally, I clarified some of the filters as their usage/terminology was inconsistent and confused. Now Filter(IsFoo, ...) *keeps* only nodes where IsFoo is true.
114 lines
3.6 KiB
Go
114 lines
3.6 KiB
Go
package render_test
|
|
|
|
import (
|
|
"flag"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/ugorji/go/codec"
|
|
|
|
"github.com/weaveworks/scope/render"
|
|
"github.com/weaveworks/scope/report"
|
|
"github.com/weaveworks/scope/test/fixture"
|
|
)
|
|
|
|
var (
|
|
benchReportFile = flag.String("bench-report-file", "", "json report file to use for benchmarking (relative to this package)")
|
|
benchmarkRenderResult map[string]report.Node
|
|
benchmarkStatsResult render.Stats
|
|
)
|
|
|
|
func BenchmarkEndpointRender(b *testing.B) { benchmarkRender(b, render.EndpointRenderer) }
|
|
func BenchmarkEndpointStats(b *testing.B) { benchmarkStats(b, render.EndpointRenderer) }
|
|
func BenchmarkProcessRender(b *testing.B) { benchmarkRender(b, render.ProcessRenderer) }
|
|
func BenchmarkProcessStats(b *testing.B) { benchmarkStats(b, render.ProcessRenderer) }
|
|
func BenchmarkProcessWithContainerNameRender(b *testing.B) {
|
|
benchmarkRender(b, render.ProcessWithContainerNameRenderer)
|
|
}
|
|
func BenchmarkProcessWithContainerNameStats(b *testing.B) {
|
|
benchmarkStats(b, render.ProcessWithContainerNameRenderer)
|
|
}
|
|
func BenchmarkProcessNameRender(b *testing.B) { benchmarkRender(b, render.ProcessNameRenderer) }
|
|
func BenchmarkProcessNameStats(b *testing.B) { benchmarkStats(b, render.ProcessNameRenderer) }
|
|
func BenchmarkContainerRender(b *testing.B) { benchmarkRender(b, render.ContainerRenderer) }
|
|
func BenchmarkContainerStats(b *testing.B) { benchmarkStats(b, render.ContainerRenderer) }
|
|
func BenchmarkContainerWithImageNameRender(b *testing.B) {
|
|
benchmarkRender(b, render.ContainerWithImageNameRenderer)
|
|
}
|
|
func BenchmarkContainerWithImageNameStats(b *testing.B) {
|
|
benchmarkStats(b, render.ContainerWithImageNameRenderer)
|
|
}
|
|
func BenchmarkContainerImageRender(b *testing.B) {
|
|
benchmarkRender(b, render.ContainerImageRenderer)
|
|
}
|
|
func BenchmarkContainerImageStats(b *testing.B) {
|
|
benchmarkStats(b, render.ContainerImageRenderer)
|
|
}
|
|
func BenchmarkContainerHostnameRender(b *testing.B) {
|
|
benchmarkRender(b, render.ContainerHostnameRenderer)
|
|
}
|
|
func BenchmarkContainerHostnameStats(b *testing.B) {
|
|
benchmarkStats(b, render.ContainerHostnameRenderer)
|
|
}
|
|
func BenchmarkHostRender(b *testing.B) { benchmarkRender(b, render.HostRenderer) }
|
|
func BenchmarkHostStats(b *testing.B) { benchmarkStats(b, render.HostRenderer) }
|
|
func BenchmarkPodRender(b *testing.B) { benchmarkRender(b, render.PodRenderer) }
|
|
func BenchmarkPodStats(b *testing.B) { benchmarkStats(b, render.PodRenderer) }
|
|
func BenchmarkPodServiceRender(b *testing.B) {
|
|
benchmarkRender(b, render.PodServiceRenderer)
|
|
}
|
|
func BenchmarkPodServiceStats(b *testing.B) {
|
|
benchmarkStats(b, render.PodServiceRenderer)
|
|
}
|
|
|
|
func benchmarkRender(b *testing.B, r render.Renderer) {
|
|
|
|
report, err := loadReport()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
b.StopTimer()
|
|
render.ResetCache()
|
|
b.StartTimer()
|
|
benchmarkRenderResult = r.Render(report, render.FilterNoop)
|
|
if len(benchmarkRenderResult) == 0 {
|
|
b.Errorf("Rendered topology contained no nodes")
|
|
}
|
|
}
|
|
}
|
|
|
|
func benchmarkStats(b *testing.B, r render.Renderer) {
|
|
report, err := loadReport()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
// No way to tell if this was successful :(
|
|
b.StopTimer()
|
|
render.ResetCache()
|
|
b.StartTimer()
|
|
benchmarkStatsResult = r.Stats(report, render.FilterNoop)
|
|
}
|
|
}
|
|
|
|
func loadReport() (report.Report, error) {
|
|
if *benchReportFile == "" {
|
|
return fixture.Report, nil
|
|
}
|
|
|
|
b, err := ioutil.ReadFile(*benchReportFile)
|
|
if err != nil {
|
|
return rpt, err
|
|
}
|
|
rpt := report.MakeReport()
|
|
err = codec.NewDecoderBytes(b, &codec.JsonHandle{}).Decode(&rpt)
|
|
return rpt, err
|
|
}
|