mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b9f9cdf918
Re-sync the entire SigNoz-derived Go tree to a clean fork of latest SigNoz main (upstream 3e6339019), eliminating the version-skew that broke `go build ./...` (mixed-era packages, ~14 imported-but-absent packages, partial signoz->o11y / clickhouse->datastore rebrands, self-import cycle, duplicate package decls). - pkg/ + cmd/ wholesale replaced with latest SigNoz; imports rewritten github.com/SigNoz/signoz/ -> github.com/hanzoai/o11y/. - Adopt SigNoz's CURRENT structure (resourcefilter->telemetryresourcefilter, integrationtypes->cloudintegrationtypes, roletypes/grammar/times folded in upstream); old paths dropped. - Fork deps repathed to hanzoai + bumped: signoz-otel-collector v0.144.6, clickhouse-go-mock v0.14.1, govaluate v0.1.0; expr-lang/expr => hanzoai/expr v1.17.8 (replace). go.mod reconciled from latest SigNoz; go.sum re-tidied. - ee/ + cmd/enterprise stripped — o11y stays MIT-only; community edition imports no ee packages. - Preserved o11y additions: module path github.com/hanzoai/o11y, mount.go (cloud.Register/zip mount adapter, HIP-0106), NOTICE + LICENSE (SigNoz §4c attribution retained; upstream pkg files are headerless per SigNoz convention). - Reverted o11y cosmetic renames (o11y*/datastore*) and experimental originals (authz/iamauthz, billing, zapreceiver, zapmetricreceiver) to SigNoz canonical to kill the skew; snapshots retained in git history for re-layering. go build ./... = exit 0. go vet clean except benign "unreachable code" in generated ANTLR parser files (inherited from upstream).
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package statsreporter
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/hanzoai/o11y/pkg/errors"
|
|
"github.com/hanzoai/o11y/pkg/factory"
|
|
"github.com/hanzoai/o11y/pkg/types/ctxtypes"
|
|
"github.com/hanzoai/o11y/pkg/types/instrumentationtypes"
|
|
"github.com/hanzoai/o11y/pkg/valuer"
|
|
)
|
|
|
|
// Aggregator aggregates stats from every registered StatsCollector for a single organization.
|
|
type Aggregator interface {
|
|
Aggregate(ctx context.Context, orgID valuer.UUID) (map[string]any, error)
|
|
}
|
|
|
|
type aggregator struct {
|
|
// settings
|
|
settings factory.ScopedProviderSettings
|
|
|
|
// a list of collectors, used to collect stats from across the codebase
|
|
collectors []StatsCollector
|
|
}
|
|
|
|
func NewAggregator(providerSettings factory.ProviderSettings, collectors []StatsCollector) Aggregator {
|
|
settings := factory.NewScopedProviderSettings(providerSettings, "github.com/hanzoai/o11y/pkg/statsreporter")
|
|
|
|
return &aggregator{
|
|
settings: settings,
|
|
collectors: collectors,
|
|
}
|
|
}
|
|
|
|
func (aggregator *aggregator) Aggregate(ctx context.Context, orgID valuer.UUID) (map[string]any, error) {
|
|
ctx = ctxtypes.NewContextWithCommentVals(ctx, map[string]string{
|
|
instrumentationtypes.CodeNamespace: "statsreporter",
|
|
instrumentationtypes.CodeFunctionName: "Aggregate",
|
|
})
|
|
var wg sync.WaitGroup
|
|
wg.Add(len(aggregator.collectors))
|
|
|
|
stats := make(map[string]any, 0)
|
|
mtx := sync.Mutex{}
|
|
|
|
for _, collector := range aggregator.collectors {
|
|
go func(collector StatsCollector) {
|
|
defer wg.Done()
|
|
|
|
collectorStats, err := collector.Collect(ctx, orgID)
|
|
if err != nil {
|
|
aggregator.settings.Logger().ErrorContext(ctx, "failed to collect stats", errors.Attr(err))
|
|
return
|
|
}
|
|
|
|
mtx.Lock()
|
|
for k, v := range collectorStats {
|
|
stats[k] = v
|
|
}
|
|
mtx.Unlock()
|
|
}(collector)
|
|
}
|
|
wg.Wait()
|
|
|
|
return stats, nil
|
|
}
|