mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 18:20:27 +00:00
2c6b6e6707
Squash of: - use detailed.Summaries to render topology nodes - ban merging nodes of different topologies (they should be mapped) - need to prune parents when mapping node types - render container images by id if they have no name - remove separate render ids and prune parents in NewDerived* - don't render metrics/metadata for groups of nodes - fixing up tests - removing pending unit tests (for mapping.go, for now) - updating experimental dir for RenderableNode removal
105 lines
3.6 KiB
Go
105 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)
|
|
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)
|
|
}
|
|
}
|
|
|
|
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
|
|
}
|