Compare commits
7 Commits
Author | SHA1 | Date | |
---|---|---|---|
fc9be7fb46 | |||
1dc55cc0c9 | |||
21494e0e7a | |||
0975e8fb4d | |||
8dfa97b54f | |||
439d5cf125 | |||
e02f291db2 |
29
go.mod
29
go.mod
@@ -1,20 +1,25 @@
|
|||||||
module go.unistack.org/micro-meter-prometheus/v4
|
module go.unistack.org/micro-meter-prometheus/v3
|
||||||
|
|
||||||
go 1.20
|
go 1.21
|
||||||
|
|
||||||
|
toolchain go1.22.0
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/prometheus/client_golang v1.16.0
|
github.com/prometheus/client_golang v1.20.4
|
||||||
github.com/prometheus/client_model v0.4.0
|
github.com/prometheus/client_model v0.6.1
|
||||||
github.com/prometheus/common v0.44.0
|
github.com/prometheus/common v0.59.1
|
||||||
go.unistack.org/micro/v4 v4.0.7
|
go.unistack.org/micro/v3 v3.10.91
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/beorn7/perks v1.0.1 // indirect
|
github.com/beorn7/perks v1.0.1 // indirect
|
||||||
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||||
github.com/golang/protobuf v1.5.3 // indirect
|
github.com/golang/protobuf v1.5.4 // indirect
|
||||||
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
|
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
||||||
github.com/prometheus/procfs v0.11.1 // indirect
|
github.com/prometheus/procfs v0.15.1 // indirect
|
||||||
golang.org/x/sys v0.11.0 // indirect
|
go.unistack.org/micro-proto/v3 v3.4.1 // indirect
|
||||||
google.golang.org/protobuf v1.31.0 // indirect
|
golang.org/x/sys v0.25.0 // indirect
|
||||||
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230530153820-e85fd2cbaebc // indirect
|
||||||
|
google.golang.org/grpc v1.57.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.34.2 // indirect
|
||||||
)
|
)
|
||||||
|
90
go.sum
90
go.sum
@@ -1,50 +1,46 @@
|
|||||||
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||||
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||||
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/go-logfmt/logfmt v0.5.1/go.mod h1:WYhtIu8zTZfxdn5+rREduYbwxfcBr/Vr6KEVveWlfTs=
|
github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
|
||||||
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
github.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=
|
||||||
github.com/golang/protobuf v1.3.5/go.mod h1:6O5/vntMXwX2lRkT1hjjk0nAC1IDOTvTlVgjlRvqsdk=
|
github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
|
||||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
|
||||||
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
|
github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA=
|
||||||
github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
|
github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw=
|
||||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq1c1nUAm88MOHcQC9l5mIlSMApZMrHA=
|
||||||
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
|
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
|
||||||
github.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
|
github.com/prometheus/client_golang v1.20.4 h1:Tgh3Yr67PaOv/uTqloMsCEdeuFTatm5zIq5+qNN23vI=
|
||||||
github.com/matttproud/golang_protobuf_extensions v1.0.4 h1:mmDVorXM7PCGKw94cs5zkfA9PSy5pEvNWRP0ET0TIVo=
|
github.com/prometheus/client_golang v1.20.4/go.mod h1:PIEt8X02hGcP8JWbeHyeZ53Y/jReSnHgO035n//V5WE=
|
||||||
github.com/matttproud/golang_protobuf_extensions v1.0.4/go.mod h1:BSXmuO+STAnVfrANrmjBb36TMTDstsz7MSK+HVaYKv4=
|
github.com/prometheus/client_model v0.6.1 h1:ZKSh/rekM+n3CeS952MLRAdFwIKqeY8b62p8ais2e9E=
|
||||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
github.com/prometheus/client_model v0.6.1/go.mod h1:OrxVMOVHjw3lKMa8+x6HeMGkHMQyHDk9E3jmP2AmGiY=
|
||||||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
github.com/prometheus/common v0.59.1 h1:LXb1quJHWm1P6wq/U824uxYi4Sg0oGvNeUm1z5dJoX0=
|
||||||
github.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
|
github.com/prometheus/common v0.59.1/go.mod h1:GpWM7dewqmVYcd7SmRaiWVe9SSqjf0UrwnYnpEZNuT0=
|
||||||
github.com/prometheus/client_golang v1.14.0 h1:nJdhIvne2eSX/XRAFV9PcvFFRbrjbcTUj0VP62TMhnw=
|
github.com/prometheus/procfs v0.15.1 h1:YagwOFzUgYfKKHX6Dr+sHT7km/hxC76UB0learggepc=
|
||||||
github.com/prometheus/client_golang v1.14.0/go.mod h1:8vpkKitgIVNcqrRBWh1C4TIUQgYNtG/XQE4E/Zae36Y=
|
github.com/prometheus/procfs v0.15.1/go.mod h1:fB45yRUv8NstnjriLhBQLuOUt+WW4BsoGhij/e3PBqk=
|
||||||
github.com/prometheus/client_golang v1.16.0 h1:yk/hx9hDbrGHovbci4BY+pRMfSuuat626eFsHb7tmT8=
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
github.com/prometheus/client_golang v1.16.0/go.mod h1:Zsulrv/L9oM40tJ7T815tM89lFEugiJ9HzIqaAx4LKc=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/prometheus/client_model v0.3.0 h1:UBgGFHqYdG/TPFD1B1ogZywDqEkwp3fBMvqdiQ7Xew4=
|
go.unistack.org/micro-proto/v3 v3.4.1 h1:UTjLSRz2YZuaHk9iSlVqqsA50JQNAEK2ZFboGqtEa9Q=
|
||||||
github.com/prometheus/client_model v0.3.0/go.mod h1:LDGWKZIo7rky3hgvBe+caln+Dr3dPggB5dvjtD7w9+w=
|
go.unistack.org/micro-proto/v3 v3.4.1/go.mod h1:okx/cnOhzuCX0ggl/vToatbCupi0O44diiiLLsZ93Zo=
|
||||||
github.com/prometheus/client_model v0.4.0 h1:5lQXD3cAg1OXBf4Wq03gTrXHeaV0TQvGfUooCfx1yqY=
|
go.unistack.org/micro/v3 v3.10.90 h1:EE7t4J14yjZC/lW6e0BfiBnU/xt/KqGkqSxumJhI/fI=
|
||||||
github.com/prometheus/client_model v0.4.0/go.mod h1:oMQmHW1/JoDwqLtg57MGgP/Fb1CJEYF2imWWhWtMkYU=
|
go.unistack.org/micro/v3 v3.10.90/go.mod h1:erMgt3Bl7vQQ0e9UpQyR5NlLiZ9pKeEJ9+1tfYFaqUg=
|
||||||
github.com/prometheus/common v0.41.0 h1:npo01n6vUlRViIj5fgwiK8vlNIh8bnoxqh3gypKsyAw=
|
go.unistack.org/micro/v3 v3.10.91 h1:vuJY4tXwpqimwIkEJ3TozMYNVQQs+C5QMlQWPgSY/YM=
|
||||||
github.com/prometheus/common v0.44.0 h1:+5BrQJwiBB9xsMygAB3TNvpQKOwlkc25LbISbrdOOfY=
|
go.unistack.org/micro/v3 v3.10.91/go.mod h1:erMgt3Bl7vQQ0e9UpQyR5NlLiZ9pKeEJ9+1tfYFaqUg=
|
||||||
github.com/prometheus/common v0.44.0/go.mod h1:ofAIvZbQ1e/nugmZGz4/qCb9Ap1VoSTIO7x0VV9VvuY=
|
golang.org/x/net v0.28.0 h1:a9JDOJc5GMUJ0+UDqmLT86WiEy7iWyIhz8gz8E4e5hE=
|
||||||
github.com/prometheus/procfs v0.9.0 h1:wzCHvIvM5SxWqYvwgVL7yJY8Lz3PKn49KQtpgMYJfhI=
|
golang.org/x/net v0.28.0/go.mod h1:yqtgsTWOOnlGLG9GFRrK3++bGOUEkNBoHZc8MEDWPNg=
|
||||||
github.com/prometheus/procfs v0.11.1 h1:xRC8Iq1yyca5ypa9n1EZnWZkt7dwcoRPQwX/5gwaUuI=
|
golang.org/x/sys v0.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34=
|
||||||
github.com/prometheus/procfs v0.11.1/go.mod h1:eesXgaPo1q7lBpVMoMy0ZOFTth9hBn4W/y0/p/ScXhY=
|
golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
go.unistack.org/micro/v4 v4.0.1 h1:xo1IxbVfgh8i0eY0VeYa3cbb13u5n/Mxnp3FOgWD4Jo=
|
golang.org/x/text v0.17.0 h1:XtiM5bkSOt+ewxlOE/aE/AKEHibwj/6gvWMl9Rsh0Qc=
|
||||||
go.unistack.org/micro/v4 v4.0.1/go.mod h1:p/J5UcSJjfHsWGT31uKoghQ5rUQZzQJBAFy+Z4+ZVMs=
|
golang.org/x/text v0.17.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
|
||||||
go.unistack.org/micro/v4 v4.0.7 h1:2lwtZlHcSwgkahhFbkI4x1lOS79lw8uLHtcEhlFF+AM=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230530153820-e85fd2cbaebc h1:XSJ8Vk1SWuNr8S18z1NZSziL0CPIXLCCMDOEFtHBOFc=
|
||||||
go.unistack.org/micro/v4 v4.0.7/go.mod h1:bVEYTlPi0EsdgZZt311bIroDg9ict7ky3C87dSCCAGk=
|
google.golang.org/genproto/googleapis/rpc v0.0.0-20230530153820-e85fd2cbaebc/go.mod h1:66JfowdXAEgad5O9NnYcsNPLCPZJD++2L9X0PCMODrA=
|
||||||
golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
google.golang.org/grpc v1.57.0 h1:kfzNeI/klCGD2YPMUlaGNT3pxvYfga7smW3Vth8Zsiw=
|
||||||
golang.org/x/sys v0.9.0 h1:KS/R3tvhPqvJvwcKfnBHJwwthS11LRhmM5D59eEXa0s=
|
google.golang.org/grpc v1.57.0/go.mod h1:Sd+9RMTACXwmub0zcNY2c4arhtrbBYD1AUHI/dt16Mo=
|
||||||
golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM=
|
google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=
|
||||||
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
|
|
||||||
google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
|
|
||||||
google.golang.org/protobuf v1.31.0 h1:g0LDEJHgrBl9N9r17Ru3sqWhkIx2NB67okBHPwC7hs8=
|
|
||||||
google.golang.org/protobuf v1.31.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
|
||||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
|
||||||
|
358
prometheus.go
358
prometheus.go
@@ -4,6 +4,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"hash/fnv"
|
"hash/fnv"
|
||||||
"io"
|
"io"
|
||||||
|
"regexp"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -11,111 +12,54 @@ import (
|
|||||||
"github.com/prometheus/client_golang/prometheus/collectors"
|
"github.com/prometheus/client_golang/prometheus/collectors"
|
||||||
dto "github.com/prometheus/client_model/go"
|
dto "github.com/prometheus/client_model/go"
|
||||||
"github.com/prometheus/common/expfmt"
|
"github.com/prometheus/common/expfmt"
|
||||||
"go.unistack.org/micro/v4/meter"
|
"go.unistack.org/micro/v3/meter"
|
||||||
"go.unistack.org/micro/v4/options"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ meter.Meter = &prometheusMeter{}
|
var _ meter.Meter = (*prometheusMeter)(nil)
|
||||||
|
|
||||||
type prometheusMeter struct {
|
type prometheusMeter struct {
|
||||||
opts meter.Options
|
opts meter.Options
|
||||||
set prometheus.Registerer
|
set prometheus.Registerer
|
||||||
counter map[string]*counters
|
counter *sync.Map
|
||||||
floatCounter map[string]*floatCounters
|
floatCounter *sync.Map
|
||||||
gauge map[string]*gauges
|
gauge *sync.Map
|
||||||
histogram map[string]*histograms
|
histogram *sync.Map
|
||||||
summary map[string]*summaries
|
summary *sync.Map
|
||||||
sync.Mutex
|
sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
type counters struct {
|
type counters struct {
|
||||||
cs map[uint64]*prometheusCounter
|
cs *sync.Map
|
||||||
}
|
}
|
||||||
|
|
||||||
type gauges struct {
|
type gauges struct {
|
||||||
cs map[uint64]*prometheusGauge
|
cs *sync.Map
|
||||||
}
|
}
|
||||||
|
|
||||||
type histograms struct {
|
type histograms struct {
|
||||||
cs map[uint64]*prometheusHistogram
|
cs *sync.Map
|
||||||
}
|
}
|
||||||
|
|
||||||
type summaries struct {
|
type summaries struct {
|
||||||
cs map[uint64]*prometheusSummary
|
cs *sync.Map
|
||||||
}
|
}
|
||||||
|
|
||||||
type floatCounters struct {
|
type floatCounters struct {
|
||||||
cs map[uint64]*prometheusFloatCounter
|
cs *sync.Map
|
||||||
}
|
}
|
||||||
|
|
||||||
func newFloat64(v float64) *float64 {
|
func NewMeter(opts ...meter.Option) *prometheusMeter {
|
||||||
nv := v
|
|
||||||
return &nv
|
|
||||||
}
|
|
||||||
|
|
||||||
func newString(v string) *string {
|
|
||||||
nv := v
|
|
||||||
return &nv
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewMeter(opts ...options.Option) *prometheusMeter {
|
|
||||||
return &prometheusMeter{
|
return &prometheusMeter{
|
||||||
set: prometheus.NewRegistry(), // prometheus.DefaultRegisterer,
|
set: prometheus.NewRegistry(), // prometheus.DefaultRegisterer,
|
||||||
opts: meter.NewOptions(opts...),
|
opts: meter.NewOptions(opts...),
|
||||||
counter: make(map[string]*counters),
|
counter: &sync.Map{},
|
||||||
floatCounter: make(map[string]*floatCounters),
|
floatCounter: &sync.Map{},
|
||||||
gauge: make(map[string]*gauges),
|
gauge: &sync.Map{},
|
||||||
histogram: make(map[string]*histograms),
|
histogram: &sync.Map{},
|
||||||
summary: make(map[string]*summaries),
|
summary: &sync.Map{},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) buildMetric(name string, labels ...string) string {
|
|
||||||
if len(m.opts.MetricPrefix) > 0 {
|
|
||||||
name = m.opts.MetricPrefix + name
|
|
||||||
}
|
|
||||||
|
|
||||||
nl := len(m.opts.Labels) + len(labels)
|
|
||||||
if nl == 0 {
|
|
||||||
return name
|
|
||||||
}
|
|
||||||
|
|
||||||
nlabels := make([]string, 0, nl)
|
|
||||||
nlabels = append(nlabels, m.opts.Labels...)
|
|
||||||
nlabels = append(nlabels, labels...)
|
|
||||||
|
|
||||||
if len(m.opts.LabelPrefix) == 0 {
|
|
||||||
return meter.BuildName(name, nlabels...)
|
|
||||||
}
|
|
||||||
|
|
||||||
for idx := 0; idx < nl; idx += 2 {
|
|
||||||
nlabels[idx] = m.opts.LabelPrefix + nlabels[idx]
|
|
||||||
}
|
|
||||||
return meter.BuildName(name, nlabels...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *prometheusMeter) buildName(name string) string {
|
|
||||||
if len(m.opts.MetricPrefix) > 0 {
|
|
||||||
name = m.opts.MetricPrefix + name
|
|
||||||
}
|
|
||||||
return name
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *prometheusMeter) buildLabels(labels ...string) []string {
|
|
||||||
nl := len(labels)
|
|
||||||
if nl == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
nlabels := make([]string, 0, nl)
|
|
||||||
|
|
||||||
for idx := 0; idx < nl; idx += 2 {
|
|
||||||
nlabels = append(nlabels, m.opts.LabelPrefix+labels[idx])
|
|
||||||
nlabels = append(nlabels, labels[idx+1])
|
|
||||||
}
|
|
||||||
return nlabels
|
|
||||||
}
|
|
||||||
|
|
||||||
func (m *prometheusMeter) Name() string {
|
func (m *prometheusMeter) Name() string {
|
||||||
return m.opts.Name
|
return m.opts.Name
|
||||||
}
|
}
|
||||||
@@ -123,167 +67,182 @@ func (m *prometheusMeter) Name() string {
|
|||||||
func (m *prometheusMeter) Counter(name string, labels ...string) meter.Counter {
|
func (m *prometheusMeter) Counter(name string, labels ...string) meter.Counter {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.counter.Load(name)
|
||||||
cd, ok := m.counter[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &counters{cs: make(map[uint64]*prometheusCounter)}
|
cd := &counters{cs: &sync.Map{}}
|
||||||
c := &prometheusCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.counter[nm] = cd
|
m.counter.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*counters)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.counter[nm] = cd
|
m.counter.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusCounter)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) FloatCounter(name string, labels ...string) meter.FloatCounter {
|
func (m *prometheusMeter) FloatCounter(name string, labels ...string) meter.FloatCounter {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.floatCounter.Load(name)
|
||||||
cd, ok := m.floatCounter[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &floatCounters{cs: make(map[uint64]*prometheusFloatCounter)}
|
cd := &floatCounters{cs: &sync.Map{}}
|
||||||
c := &prometheusFloatCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusFloatCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.floatCounter[nm] = cd
|
m.floatCounter.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*floatCounters)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusFloatCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusFloatCounter{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.floatCounter[nm] = cd
|
m.floatCounter.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusFloatCounter)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Gauge(name string, fn func() float64, labels ...string) meter.Gauge {
|
func (m *prometheusMeter) Gauge(name string, fn func() float64, labels ...string) meter.Gauge {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.gauge.Load(name)
|
||||||
cd, ok := m.gauge[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &gauges{cs: make(map[uint64]*prometheusGauge)}
|
cd := &gauges{cs: &sync.Map{}}
|
||||||
c := &prometheusGauge{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusGauge{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.gauge[nm] = cd
|
m.gauge.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*gauges)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusGauge{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: nm}), labels: labels}
|
c := &prometheusGauge{c: prometheus.NewGauge(prometheus.GaugeOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.gauge[nm] = cd
|
m.gauge.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusGauge)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Histogram(name string, labels ...string) meter.Histogram {
|
func (m *prometheusMeter) Histogram(name string, labels ...string) meter.Histogram {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.histogram.Load(name)
|
||||||
cd, ok := m.histogram[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &histograms{cs: make(map[uint64]*prometheusHistogram)}
|
cd := &histograms{cs: &sync.Map{}}
|
||||||
c := &prometheusHistogram{c: prometheus.NewHistogram(prometheus.HistogramOpts{Name: nm}), labels: labels}
|
c := &prometheusHistogram{c: prometheus.NewHistogram(prometheus.HistogramOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.histogram[nm] = cd
|
m.histogram.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*histograms)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusHistogram{c: prometheus.NewHistogram(prometheus.HistogramOpts{Name: nm}), labels: labels}
|
c := &prometheusHistogram{c: prometheus.NewHistogram(prometheus.HistogramOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.histogram[nm] = cd
|
m.histogram.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusHistogram)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Summary(name string, labels ...string) meter.Summary {
|
func (m *prometheusMeter) Summary(name string, labels ...string) meter.Summary {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.summary.Load(name)
|
||||||
cd, ok := m.summary[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &summaries{cs: make(map[uint64]*prometheusSummary)}
|
cd := &summaries{cs: &sync.Map{}}
|
||||||
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{Name: nm}), labels: labels}
|
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.summary[nm] = cd
|
m.summary.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*summaries)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{Name: nm}), labels: labels}
|
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{Name: name}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.summary[nm] = cd
|
m.summary.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusSummary)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) meter.Summary {
|
func (m *prometheusMeter) SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) meter.Summary {
|
||||||
m.Lock()
|
m.Lock()
|
||||||
defer m.Unlock()
|
defer m.Unlock()
|
||||||
nm := m.buildName(name)
|
labels = append(m.opts.Labels, labels...)
|
||||||
labels = m.buildLabels(append(m.opts.Labels, labels...)...)
|
vcd, ok := m.summary.Load(name)
|
||||||
cd, ok := m.summary[nm]
|
|
||||||
h := newHash(labels)
|
h := newHash(labels)
|
||||||
if !ok {
|
if !ok {
|
||||||
cd = &summaries{cs: make(map[uint64]*prometheusSummary)}
|
cd := &summaries{cs: &sync.Map{}}
|
||||||
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{
|
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{
|
||||||
Name: nm,
|
Name: name,
|
||||||
MaxAge: window,
|
MaxAge: window,
|
||||||
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
||||||
}), labels: labels}
|
}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.summary[nm] = cd
|
m.summary.Store(name, cd)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
c, ok := cd.cs[h]
|
cd := vcd.(*summaries)
|
||||||
|
vc, ok := cd.cs.Load(h)
|
||||||
if !ok {
|
if !ok {
|
||||||
c = &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{
|
c := &prometheusSummary{c: prometheus.NewSummary(prometheus.SummaryOpts{
|
||||||
Name: nm,
|
Name: name,
|
||||||
MaxAge: window,
|
MaxAge: window,
|
||||||
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001},
|
||||||
}), labels: labels}
|
}), labels: labels}
|
||||||
cd.cs[h] = c
|
cd.cs.Store(h, c)
|
||||||
m.summary[nm] = cd
|
m.summary.Store(name, cd)
|
||||||
|
return c
|
||||||
}
|
}
|
||||||
|
c := vc.(*prometheusSummary)
|
||||||
return c
|
return c
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Init(opts ...options.Option) error {
|
func (m *prometheusMeter) Init(opts ...meter.Option) error {
|
||||||
for _, o := range opts {
|
for _, o := range opts {
|
||||||
o(&m.opts)
|
o(&m.opts)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if m.opts.WriteProcessMetrics || m.opts.WriteFDMetrics {
|
||||||
|
pc := collectors.NewProcessCollector(collectors.ProcessCollectorOpts{})
|
||||||
|
_ = m.set.Register(pc)
|
||||||
|
gc := collectors.NewGoCollector(collectors.WithGoCollectorRuntimeMetrics(collectors.GoRuntimeMetricsRule{Matcher: regexp.MustCompile("/.*")}))
|
||||||
|
_ = m.set.Register(gc)
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Write(w io.Writer, opts ...options.Option) error {
|
func (m *prometheusMeter) Write(w io.Writer, opts ...meter.Option) error {
|
||||||
options := m.opts
|
options := m.opts
|
||||||
for _, o := range opts {
|
for _, o := range opts {
|
||||||
o(&options)
|
o(&options)
|
||||||
}
|
}
|
||||||
|
|
||||||
if options.WriteProcessMetrics || options.WriteFDMetrics {
|
|
||||||
c := collectors.NewProcessCollector(collectors.ProcessCollectorOpts{})
|
|
||||||
_ = m.set.Register(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
g, ok := m.set.(prometheus.Gatherer)
|
g, ok := m.set.(prometheus.Gatherer)
|
||||||
if !ok {
|
if !ok {
|
||||||
return fmt.Errorf("set type %T not prometheus.Gatherer", m.set)
|
return fmt.Errorf("set type %T not prometheus.Gatherer", m.set)
|
||||||
@@ -294,82 +253,97 @@ func (m *prometheusMeter) Write(w io.Writer, opts ...options.Option) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
enc := expfmt.NewEncoder(w, expfmt.FmtText)
|
enc := expfmt.NewEncoder(w, expfmt.NewFormat(expfmt.TypeTextPlain))
|
||||||
|
|
||||||
for name, metrics := range m.counter {
|
m.counter.Range(func(k, v any) bool {
|
||||||
|
name := k.(string)
|
||||||
mf := &dto.MetricFamily{
|
mf := &dto.MetricFamily{
|
||||||
Name: newString(name),
|
Name: &name,
|
||||||
Type: dto.MetricType_GAUGE.Enum(),
|
Type: dto.MetricType_GAUGE.Enum(),
|
||||||
Metric: make([]*dto.Metric, 0, len(metrics.cs)),
|
|
||||||
}
|
}
|
||||||
for _, c := range metrics.cs {
|
v.(*counters).cs.Range(func(_, nv any) bool {
|
||||||
|
c := nv.(*prometheusCounter)
|
||||||
m := &dto.Metric{}
|
m := &dto.Metric{}
|
||||||
_ = c.c.Write(m)
|
_ = c.c.Write(m)
|
||||||
fillMetric(m, c.labels)
|
fillMetric(m, c.labels)
|
||||||
mf.Metric = append(mf.Metric, m)
|
mf.Metric = append(mf.Metric, m)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
mfs = append(mfs, mf)
|
mfs = append(mfs, mf)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
for name, metrics := range m.gauge {
|
m.gauge.Range(func(k, v any) bool {
|
||||||
|
name := k.(string)
|
||||||
mf := &dto.MetricFamily{
|
mf := &dto.MetricFamily{
|
||||||
Name: newString(name),
|
Name: &name,
|
||||||
Type: dto.MetricType_GAUGE.Enum(),
|
Type: dto.MetricType_GAUGE.Enum(),
|
||||||
Metric: make([]*dto.Metric, 0, len(metrics.cs)),
|
|
||||||
}
|
}
|
||||||
for _, c := range metrics.cs {
|
v.(*gauges).cs.Range(func(_, nv any) bool {
|
||||||
|
c := nv.(*prometheusGauge)
|
||||||
m := &dto.Metric{}
|
m := &dto.Metric{}
|
||||||
_ = c.c.Write(m)
|
_ = c.c.Write(m)
|
||||||
fillMetric(m, c.labels)
|
fillMetric(m, c.labels)
|
||||||
mf.Metric = append(mf.Metric, m)
|
mf.Metric = append(mf.Metric, m)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
mfs = append(mfs, mf)
|
mfs = append(mfs, mf)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
for name, metrics := range m.floatCounter {
|
m.floatCounter.Range(func(k, v any) bool {
|
||||||
|
name := k.(string)
|
||||||
mf := &dto.MetricFamily{
|
mf := &dto.MetricFamily{
|
||||||
Name: newString(name),
|
Name: &name,
|
||||||
Type: dto.MetricType_GAUGE.Enum(),
|
Type: dto.MetricType_GAUGE.Enum(),
|
||||||
Metric: make([]*dto.Metric, 0, len(metrics.cs)),
|
|
||||||
}
|
}
|
||||||
for _, c := range metrics.cs {
|
v.(*floatCounters).cs.Range(func(_, nv any) bool {
|
||||||
|
c := nv.(*prometheusFloatCounter)
|
||||||
m := &dto.Metric{}
|
m := &dto.Metric{}
|
||||||
_ = c.c.Write(m)
|
_ = c.c.Write(m)
|
||||||
fillMetric(m, c.labels)
|
fillMetric(m, c.labels)
|
||||||
mf.Metric = append(mf.Metric, m)
|
mf.Metric = append(mf.Metric, m)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
mfs = append(mfs, mf)
|
mfs = append(mfs, mf)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
for name, metrics := range m.histogram {
|
m.histogram.Range(func(k, v any) bool {
|
||||||
|
name := k.(string)
|
||||||
mf := &dto.MetricFamily{
|
mf := &dto.MetricFamily{
|
||||||
Name: newString(name),
|
Name: &name,
|
||||||
Type: dto.MetricType_HISTOGRAM.Enum(),
|
Type: dto.MetricType_HISTOGRAM.Enum(),
|
||||||
Metric: make([]*dto.Metric, 0, len(metrics.cs)),
|
|
||||||
}
|
}
|
||||||
for _, c := range metrics.cs {
|
v.(*histograms).cs.Range(func(_, nv any) bool {
|
||||||
|
c := nv.(*prometheusHistogram)
|
||||||
m := &dto.Metric{}
|
m := &dto.Metric{}
|
||||||
_ = c.c.Write(m)
|
_ = c.c.Write(m)
|
||||||
fillMetric(m, c.labels)
|
fillMetric(m, c.labels)
|
||||||
mf.Metric = append(mf.Metric, m)
|
mf.Metric = append(mf.Metric, m)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
mfs = append(mfs, mf)
|
mfs = append(mfs, mf)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
for name, metrics := range m.summary {
|
m.summary.Range(func(k, v any) bool {
|
||||||
|
name := k.(string)
|
||||||
mf := &dto.MetricFamily{
|
mf := &dto.MetricFamily{
|
||||||
Name: newString(name),
|
Name: &name,
|
||||||
Type: dto.MetricType_SUMMARY.Enum(),
|
Type: dto.MetricType_SUMMARY.Enum(),
|
||||||
Metric: make([]*dto.Metric, 0, len(metrics.cs)),
|
|
||||||
}
|
}
|
||||||
for _, c := range metrics.cs {
|
v.(*summaries).cs.Range(func(_, nv any) bool {
|
||||||
|
c := nv.(*prometheusSummary)
|
||||||
m := &dto.Metric{}
|
m := &dto.Metric{}
|
||||||
_ = c.c.Write(m)
|
_ = c.c.Write(m)
|
||||||
fillMetric(m, c.labels)
|
fillMetric(m, c.labels)
|
||||||
mf.Metric = append(mf.Metric, m)
|
mf.Metric = append(mf.Metric, m)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
mfs = append(mfs, mf)
|
mfs = append(mfs, mf)
|
||||||
}
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
for _, mf := range mfs {
|
for _, mf := range mfs {
|
||||||
_ = enc.Encode(mf)
|
_ = enc.Encode(mf)
|
||||||
@@ -382,7 +356,7 @@ func (m *prometheusMeter) Write(w io.Writer, opts ...options.Option) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Clone(opts ...options.Option) meter.Meter {
|
func (m *prometheusMeter) Clone(opts ...meter.Option) meter.Meter {
|
||||||
options := m.opts
|
options := m.opts
|
||||||
for _, o := range opts {
|
for _, o := range opts {
|
||||||
o(&options)
|
o(&options)
|
||||||
@@ -407,7 +381,7 @@ func (m *prometheusMeter) String() string {
|
|||||||
return "prometheus"
|
return "prometheus"
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *prometheusMeter) Set(opts ...options.Option) meter.Meter {
|
func (m *prometheusMeter) Set(opts ...meter.Option) meter.Meter {
|
||||||
nm := &prometheusMeter{opts: m.opts}
|
nm := &prometheusMeter{opts: m.opts}
|
||||||
for _, o := range opts {
|
for _, o := range opts {
|
||||||
o(&nm.opts)
|
o(&nm.opts)
|
||||||
@@ -522,12 +496,18 @@ func newHash(labels []string) uint64 {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func fillMetric(m *dto.Metric, labels []string) *dto.Metric {
|
func fillMetric(m *dto.Metric, labels []string) *dto.Metric {
|
||||||
|
var ok bool
|
||||||
|
seen := make(map[string]bool, len(labels)/2)
|
||||||
m.Label = make([]*dto.LabelPair, 0, len(labels)/2)
|
m.Label = make([]*dto.LabelPair, 0, len(labels)/2)
|
||||||
for idx := 0; idx < len(labels); idx += 2 {
|
for idx := 0; idx < len(labels); idx += 2 {
|
||||||
|
if _, ok = seen[labels[idx]]; ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
m.Label = append(m.Label, &dto.LabelPair{
|
m.Label = append(m.Label, &dto.LabelPair{
|
||||||
Name: newString(labels[idx]),
|
Name: &(labels[idx]),
|
||||||
Value: newString(labels[idx+1]),
|
Value: &(labels[idx+1]),
|
||||||
})
|
})
|
||||||
|
seen[labels[idx]] = true
|
||||||
}
|
}
|
||||||
return m
|
return m
|
||||||
}
|
}
|
||||||
|
@@ -5,37 +5,29 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"go.unistack.org/micro/v4/client"
|
"go.unistack.org/micro/v3/client"
|
||||||
"go.unistack.org/micro/v4/codec"
|
"go.unistack.org/micro/v3/codec"
|
||||||
"go.unistack.org/micro/v4/meter"
|
"go.unistack.org/micro/v3/meter"
|
||||||
"go.unistack.org/micro/v4/meter/wrapper"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestBuildName(t *testing.T) {
|
func TestStd(t *testing.T) {
|
||||||
m := NewMeter()
|
m := NewMeter(meter.WriteProcessMetrics(true), meter.WriteFDMetrics(true))
|
||||||
check := `micro_foo{micro_aaa="b",micro_bar="baz",micro_ccc="d"}`
|
if err := m.Init(); err != nil {
|
||||||
name := m.buildMetric("foo", "bar", "baz", "aaa", "b", "ccc", "d")
|
t.Fatal(err)
|
||||||
if name != check {
|
}
|
||||||
t.Fatalf("metric name error: %s != %s", name, check)
|
buf := bytes.NewBuffer(nil)
|
||||||
|
_ = m.Write(buf)
|
||||||
|
if !bytes.Contains(buf.Bytes(), []byte(`go_goroutine`)) {
|
||||||
|
t.Fatalf("invalid metrics output: %s", buf.Bytes())
|
||||||
}
|
}
|
||||||
|
|
||||||
cnt := m.Counter("counter", "key", "val")
|
|
||||||
cnt.Inc()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestWrapper(t *testing.T) {
|
func TestWrapper(t *testing.T) {
|
||||||
m := NewMeter() // meter.Labels("test_key", "test_val"))
|
m := NewMeter() // meter.Labels("test_key", "test_val"))
|
||||||
|
|
||||||
w := wrapper.NewClientWrapper(
|
|
||||||
wrapper.ServiceName("svc1"),
|
|
||||||
wrapper.ServiceVersion("0.0.1"),
|
|
||||||
wrapper.ServiceID("12345"),
|
|
||||||
wrapper.Meter(m),
|
|
||||||
)
|
|
||||||
|
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
|
|
||||||
c := client.NewClient(client.Wrap(w))
|
c := client.NewClient(client.Meter(m))
|
||||||
if err := c.Init(); err != nil {
|
if err := c.Init(); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
@@ -45,7 +37,7 @@ func TestWrapper(t *testing.T) {
|
|||||||
_, _ = rsp, err
|
_, _ = rsp, err
|
||||||
buf := bytes.NewBuffer(nil)
|
buf := bytes.NewBuffer(nil)
|
||||||
_ = m.Write(buf, meter.WriteProcessMetrics(false), meter.WriteFDMetrics(false))
|
_ = m.Write(buf, meter.WriteProcessMetrics(false), meter.WriteFDMetrics(false))
|
||||||
if !bytes.Contains(buf.Bytes(), []byte(`micro_client_request_inflight{micro_endpoint="svc2.Service.Method"} 0`)) {
|
if !bytes.Contains(buf.Bytes(), []byte(`micro_client_request_inflight{endpoint="Service.Method"} 0`)) {
|
||||||
t.Fatalf("invalid metrics output: %s", buf.Bytes())
|
t.Fatalf("invalid metrics output: %s", buf.Bytes())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -53,17 +45,17 @@ func TestWrapper(t *testing.T) {
|
|||||||
func TestMultiple(t *testing.T) {
|
func TestMultiple(t *testing.T) {
|
||||||
m := NewMeter() // meter.Labels("test_key", "test_val"))
|
m := NewMeter() // meter.Labels("test_key", "test_val"))
|
||||||
|
|
||||||
m.Counter("server", "endpoint", "ep1", "path", "/path1").Inc()
|
m.Counter("micro_server", "endpoint", "ep1", "path", "/path1").Inc()
|
||||||
m.Counter("server", "endpoint", "ep1", "path", "/path1").Inc()
|
m.Counter("micro_server", "endpoint", "ep1", "path", "/path1").Inc()
|
||||||
|
|
||||||
m.Counter("server", "endpoint", "ep2", "path", "/path2").Inc()
|
m.Counter("micro_server", "endpoint", "ep2", "path", "/path2").Inc()
|
||||||
m.Counter("server", "endpoint", "ep2", "path", "/path2").Inc()
|
m.Counter("micro_server", "endpoint", "ep2", "path", "/path2").Inc()
|
||||||
|
|
||||||
m.Counter("server", "endpoint", "ep3", "path", "/path3", "status", "success").Inc()
|
m.Counter("micro_server", "endpoint", "ep3", "path", "/path3", "status", "success").Inc()
|
||||||
|
|
||||||
buf := bytes.NewBuffer(nil)
|
buf := bytes.NewBuffer(nil)
|
||||||
_ = m.Write(buf, meter.WriteProcessMetrics(false), meter.WriteFDMetrics(false))
|
_ = m.Write(buf, meter.WriteProcessMetrics(false), meter.WriteFDMetrics(false))
|
||||||
if !bytes.Contains(buf.Bytes(), []byte(`micro_server{micro_endpoint="ep1",micro_path="/path1"} 2`)) {
|
if !bytes.Contains(buf.Bytes(), []byte(`micro_server{endpoint="ep1",path="/path1"} 2`)) {
|
||||||
// t.Fatal("XXXX")
|
// t.Fatal("XXXX")
|
||||||
t.Fatalf("invalid metrics output: %s", buf.Bytes())
|
t.Fatalf("invalid metrics output: %s", buf.Bytes())
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user