Skip to content

Commit

Permalink
metrics: use slices package for sorting (ethereum#27493 ethereum#27909)
Browse files Browse the repository at this point in the history
Co-authored-by: Felix Lange <fjl@twurst.com>
  • Loading branch information
2 people authored and JukLee0ira committed Dec 20, 2024
1 parent 372294d commit a2de051
Show file tree
Hide file tree
Showing 6 changed files with 22 additions and 35 deletions.
1 change: 1 addition & 0 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ require (
github.com/mattn/go-isatty v0.0.17
github.com/shirou/gopsutil v3.21.4-0.20210419000835-c7a38de76ee5+incompatible
github.com/urfave/cli/v2 v2.27.5
golang.org/x/exp v0.0.0-20231006140011-7918f672742d
gopkg.in/natefinch/lumberjack.v2 v2.2.1
)

Expand Down
2 changes: 2 additions & 0 deletions go.sum
Original file line number Diff line number Diff line change
Expand Up @@ -218,6 +218,8 @@ golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a/go.mod h1:LzIPMQfyMNhhGPh
golang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
golang.org/x/crypto v0.29.0 h1:L5SG1JTTXupVV3n6sUqMTeWbjAyfPwoda2DLX8J8FrQ=
golang.org/x/crypto v0.29.0/go.mod h1:+F4F4N5hv6v38hfeYwTdx20oUvLLc+QfrE9Ax9HtgRg=
golang.org/x/exp v0.0.0-20231006140011-7918f672742d h1:jtJma62tbqLibJ5sFQz8bKtEM8rJBtfilJ2qTU199MI=
golang.org/x/exp v0.0.0-20231006140011-7918f672742d/go.mod h1:ldy0pHrwJyGW56pPQzzkH36rKxoZW1tw7ZJpeKx+hdo=
golang.org/x/mod v0.17.0 h1:zY54UmvipHiNd+pm+m0x9KhZ9hl1/7QNMyxXbc6ICqA=
golang.org/x/mod v0.17.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
Expand Down
12 changes: 3 additions & 9 deletions metrics/resetting_timer.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@ package metrics

import (
"math"
"sort"
"sync"
"time"

"golang.org/x/exp/slices"
)

// Initial slice capacity for the values stored in a ResettingTimer
Expand Down Expand Up @@ -182,7 +183,7 @@ func (t *ResettingTimerSnapshot) Mean() float64 {
}

func (t *ResettingTimerSnapshot) calc(percentiles []float64) {
sort.Sort(Int64Slice(t.values))
slices.Sort(t.values)

count := len(t.values)
if count > 0 {
Expand Down Expand Up @@ -228,10 +229,3 @@ func (t *ResettingTimerSnapshot) calc(percentiles []float64) {

t.calculated = true
}

// Int64Slice attaches the methods of sort.Interface to []int64, sorting in increasing order.
type Int64Slice []int64

func (s Int64Slice) Len() int { return len(s) }
func (s Int64Slice) Less(i, j int) bool { return s[i] < s[j] }
func (s Int64Slice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
15 changes: 5 additions & 10 deletions metrics/sample.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,10 @@ package metrics
import (
"math"
"math/rand"
"sort"
"sync"
"time"

"golang.org/x/exp/slices"
)

const rescaleThreshold = time.Hour
Expand Down Expand Up @@ -282,17 +283,17 @@ func SampleMin(values []int64) int64 {
}

// SamplePercentiles returns an arbitrary percentile of the slice of int64.
func SamplePercentile(values int64Slice, p float64) float64 {
func SamplePercentile(values []int64, p float64) float64 {
return SamplePercentiles(values, []float64{p})[0]
}

// SamplePercentiles returns a slice of arbitrary percentiles of the slice of
// int64.
func SamplePercentiles(values int64Slice, ps []float64) []float64 {
func SamplePercentiles(values []int64, ps []float64) []float64 {
scores := make([]float64, len(ps))
size := len(values)
if size > 0 {
sort.Sort(values)
slices.Sort(values)
for i, p := range ps {
pos := p * float64(size+1)
if pos < 1.0 {
Expand Down Expand Up @@ -633,9 +634,3 @@ func (h *expDecaySampleHeap) down(i, n int) {
i = j
}
}

type int64Slice []int64

func (p int64Slice) Len() int { return len(p) }
func (p int64Slice) Less(i, j int) bool { return p[i] < p[j] }
func (p int64Slice) Swap(i, j int) { p[i], p[j] = p[j], p[i] }
20 changes: 7 additions & 13 deletions metrics/writer.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,10 @@ package metrics
import (
"fmt"
"io"
"sort"
"strings"
"time"

"golang.org/x/exp/slices"
)

// Write sorts writes each metric in the given registry periodically to the
Expand All @@ -18,12 +20,11 @@ func Write(r Registry, d time.Duration, w io.Writer) {
// WriteOnce sorts and writes metrics in the given registry to the given
// io.Writer.
func WriteOnce(r Registry, w io.Writer) {
var namedMetrics namedMetricSlice
var namedMetrics []namedMetric
r.Each(func(name string, i interface{}) {
namedMetrics = append(namedMetrics, namedMetric{name, i})
})

sort.Sort(namedMetrics)
slices.SortFunc(namedMetrics, namedMetric.cmp)
for _, namedMetric := range namedMetrics {
switch metric := namedMetric.m.(type) {
case Counter:
Expand Down Expand Up @@ -91,13 +92,6 @@ type namedMetric struct {
m interface{}
}

// namedMetricSlice is a slice of namedMetrics that implements sort.Interface.
type namedMetricSlice []namedMetric

func (nms namedMetricSlice) Len() int { return len(nms) }

func (nms namedMetricSlice) Swap(i, j int) { nms[i], nms[j] = nms[j], nms[i] }

func (nms namedMetricSlice) Less(i, j int) bool {
return nms[i].name < nms[j].name
func (m namedMetric) cmp(other namedMetric) int {
return strings.Compare(m.name, other.name)
}
7 changes: 4 additions & 3 deletions metrics/writer_test.go
Original file line number Diff line number Diff line change
@@ -1,19 +1,20 @@
package metrics

import (
"sort"
"testing"

"golang.org/x/exp/slices"
)

func TestMetricsSorting(t *testing.T) {
var namedMetrics = namedMetricSlice{
var namedMetrics = []namedMetric{
{name: "zzz"},
{name: "bbb"},
{name: "fff"},
{name: "ggg"},
}

sort.Sort(namedMetrics)
slices.SortFunc(namedMetrics, namedMetric.cmp)
for i, name := range []string{"bbb", "fff", "ggg", "zzz"} {
if namedMetrics[i].name != name {
t.Fail()
Expand Down

0 comments on commit a2de051

Please sign in to comment.