mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Alex ShtinandGitHub
8e8f1b0f7c
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.
## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
90 lines
3.1 KiB
Go
90 lines
3.1 KiB
Go
package metrics
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pborman/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/uber-go/tally/v4"
|
|
)
|
|
|
|
var defaultConfig = ClientConfig{
|
|
Tags: nil,
|
|
ExcludeTags: map[string][]string{
|
|
"taskqueue": {"__sticky__"},
|
|
"activityType": {},
|
|
"workflowType": {},
|
|
},
|
|
Prefix: "",
|
|
PerUnitHistogramBoundaries: map[string][]float64{
|
|
Dimensionless: {0, 10, 100},
|
|
Bytes: {1024, 2048},
|
|
Milliseconds: {10, 500, 1000, 5000, 10000},
|
|
Seconds: {0.01, 0.5, 1, 5, 10},
|
|
},
|
|
}
|
|
|
|
func TestTallyScope(t *testing.T) {
|
|
scope := tally.NewTestScope("test", map[string]string{})
|
|
mp := NewTallyMetricsHandler(defaultConfig, scope)
|
|
recordTallyMetrics(mp)
|
|
|
|
snap := scope.Snapshot()
|
|
counters, gauges, timers, histograms := snap.Counters(), snap.Gauges(), snap.Timers(), snap.Histograms()
|
|
|
|
assert.EqualValues(t, 8, counters["test.hits+"].Value())
|
|
assert.EqualValues(t, map[string]string{}, counters["test.hits+"].Tags())
|
|
|
|
assert.EqualValues(t, 11, counters["test.hits-tagged+taskqueue=__sticky__"].Value())
|
|
assert.EqualValues(t, map[string]string{"taskqueue": "__sticky__"}, counters["test.hits-tagged+taskqueue=__sticky__"].Tags())
|
|
|
|
assert.EqualValues(t, 14, counters["test.hits-tagged-excluded+taskqueue="+tagExcludedValue].Value())
|
|
assert.EqualValues(t, map[string]string{"taskqueue": tagExcludedValue}, counters["test.hits-tagged-excluded+taskqueue="+tagExcludedValue].Tags())
|
|
|
|
assert.EqualValues(t, float64(-100), gauges["test.temp+location=Mare Imbrium"].Value())
|
|
assert.EqualValues(t, map[string]string{
|
|
"location": "Mare Imbrium",
|
|
}, gauges["test.temp+location=Mare Imbrium"].Tags())
|
|
|
|
assert.EqualValues(t, []time.Duration{
|
|
1248 * time.Millisecond,
|
|
5255 * time.Millisecond,
|
|
}, timers["test.latency+"].Values())
|
|
assert.EqualValues(t, map[string]string{}, timers["test.latency+"].Tags())
|
|
|
|
assert.EqualValues(t, map[float64]int64{
|
|
1024: 0,
|
|
2048: 0,
|
|
math.MaxFloat64: 1,
|
|
}, histograms["test.transmission+"].Values())
|
|
assert.EqualValues(t, map[time.Duration]int64(nil), histograms["test.transmission+"].Durations())
|
|
assert.EqualValues(t, map[string]string{}, histograms["test.transmission+"].Tags())
|
|
|
|
newTaggedHandler := mp.WithTags(NamespaceTag(uuid.New()))
|
|
recordTallyMetrics(newTaggedHandler)
|
|
snap = scope.Snapshot()
|
|
counters = snap.Counters()
|
|
|
|
assert.EqualValues(t, 11, counters["test.hits-tagged+taskqueue=__sticky__"].Value())
|
|
assert.EqualValues(t, map[string]string{"taskqueue": "__sticky__"}, counters["test.hits-tagged+taskqueue=__sticky__"].Tags())
|
|
}
|
|
|
|
func recordTallyMetrics(h Handler) {
|
|
hitsCounter := h.Counter("hits")
|
|
gauge := h.Gauge("temp")
|
|
timer := h.Timer("latency")
|
|
histogram := h.Histogram("transmission", Bytes)
|
|
hitsTaggedCounter := h.Counter("hits-tagged")
|
|
hitsTaggedExcludedCounter := h.Counter("hits-tagged-excluded")
|
|
|
|
hitsCounter.Record(8)
|
|
gauge.Record(-100, StringTag("location", "Mare Imbrium"))
|
|
timer.Record(1248 * time.Millisecond)
|
|
timer.Record(5255 * time.Millisecond)
|
|
histogram.Record(1234567)
|
|
hitsTaggedCounter.Record(11, UnsafeTaskQueueTag("__sticky__"))
|
|
hitsTaggedExcludedCounter.Record(14, UnsafeTaskQueueTag("filtered"))
|
|
}
|