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).
163 lines
4.5 KiB
Go
163 lines
4.5 KiB
Go
package rules
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoai/o11y/pkg/flagger"
|
|
"github.com/hanzoai/o11y/pkg/instrumentation/instrumentationtest"
|
|
"github.com/hanzoai/o11y/pkg/querier"
|
|
"github.com/hanzoai/o11y/pkg/querybuilder"
|
|
"github.com/hanzoai/o11y/pkg/telemetrylogs"
|
|
"github.com/hanzoai/o11y/pkg/telemetrymetrics"
|
|
"github.com/hanzoai/o11y/pkg/telemetrystore"
|
|
"github.com/hanzoai/o11y/pkg/telemetrytraces"
|
|
"github.com/hanzoai/o11y/pkg/types/telemetrytypes"
|
|
"github.com/hanzoai/o11y/pkg/types/telemetrytypes/telemetrytypestest"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hanzoai/o11y/pkg/flagger/flaggertest"
|
|
)
|
|
|
|
func prepareQuerierForMetrics(t *testing.T, telemetryStore telemetrystore.TelemetryStore) (querier.Querier, *telemetrytypestest.MockMetadataStore) {
|
|
providerSettings := instrumentationtest.New().ToProviderSettings()
|
|
metadataStore := telemetrytypestest.NewMockMetadataStore()
|
|
|
|
flagger, err := flagger.New(
|
|
context.Background(),
|
|
instrumentationtest.New().ToProviderSettings(),
|
|
flagger.Config{},
|
|
flagger.MustNewRegistry(),
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
metricFieldMapper := telemetrymetrics.NewFieldMapper()
|
|
metricConditionBuilder := telemetrymetrics.NewConditionBuilder(metricFieldMapper)
|
|
metricStmtBuilder := telemetrymetrics.NewMetricQueryStatementBuilder(
|
|
providerSettings,
|
|
metadataStore,
|
|
metricFieldMapper,
|
|
metricConditionBuilder,
|
|
flagger,
|
|
)
|
|
|
|
return querier.New(
|
|
providerSettings,
|
|
telemetryStore,
|
|
metadataStore,
|
|
nil, // prometheus
|
|
nil, // traceStmtBuilder
|
|
nil, // logStmtBuilder
|
|
nil, // auditStmtBuilder
|
|
metricStmtBuilder,
|
|
nil, // meterStmtBuilder
|
|
nil, // traceOperatorStmtBuilder
|
|
nil, // bucketCache
|
|
flagger,
|
|
0,
|
|
), metadataStore
|
|
}
|
|
|
|
func prepareQuerierForLogs(t *testing.T, telemetryStore telemetrystore.TelemetryStore, keysMap map[string][]*telemetrytypes.TelemetryFieldKey) querier.Querier {
|
|
t.Helper()
|
|
providerSettings := instrumentationtest.New().ToProviderSettings()
|
|
metadataStore := telemetrytypestest.NewMockMetadataStore()
|
|
|
|
for _, keys := range keysMap {
|
|
for _, key := range keys {
|
|
key.Signal = telemetrytypes.SignalLogs
|
|
}
|
|
}
|
|
metadataStore.KeysMap = keysMap
|
|
|
|
fl := flaggertest.New(t)
|
|
logFieldMapper := telemetrylogs.NewFieldMapper(fl)
|
|
logConditionBuilder := telemetrylogs.NewConditionBuilder(logFieldMapper, fl)
|
|
logAggExprRewriter := querybuilder.NewAggExprRewriter(
|
|
providerSettings,
|
|
telemetrylogs.DefaultFullTextColumn,
|
|
logFieldMapper,
|
|
logConditionBuilder,
|
|
telemetrylogs.GetBodyJSONKey,
|
|
fl,
|
|
)
|
|
logStmtBuilder := telemetrylogs.NewLogQueryStatementBuilder(
|
|
providerSettings,
|
|
metadataStore,
|
|
logFieldMapper,
|
|
logConditionBuilder,
|
|
logAggExprRewriter,
|
|
telemetrylogs.DefaultFullTextColumn,
|
|
telemetrylogs.GetBodyJSONKey,
|
|
fl,
|
|
nil,
|
|
false,
|
|
100000,
|
|
)
|
|
|
|
return querier.New(
|
|
providerSettings,
|
|
telemetryStore,
|
|
metadataStore,
|
|
nil, // prometheus
|
|
nil, // traceStmtBuilder
|
|
logStmtBuilder, // logStmtBuilder
|
|
nil, // auditStmtBuilder
|
|
nil, // metricStmtBuilder
|
|
nil, // meterStmtBuilder
|
|
nil, // traceOperatorStmtBuilder
|
|
nil, // bucketCache
|
|
fl,
|
|
5*time.Minute, // logTraceIDWindowPadding
|
|
)
|
|
}
|
|
|
|
func prepareQuerierForTraces(t *testing.T, telemetryStore telemetrystore.TelemetryStore, keysMap map[string][]*telemetrytypes.TelemetryFieldKey) querier.Querier {
|
|
t.Helper()
|
|
|
|
providerSettings := instrumentationtest.New().ToProviderSettings()
|
|
metadataStore := telemetrytypestest.NewMockMetadataStore()
|
|
|
|
for _, keys := range keysMap {
|
|
for _, key := range keys {
|
|
key.Signal = telemetrytypes.SignalTraces
|
|
}
|
|
}
|
|
metadataStore.KeysMap = keysMap
|
|
|
|
// Create trace statement builder
|
|
traceFieldMapper := telemetrytraces.NewFieldMapper()
|
|
traceConditionBuilder := telemetrytraces.NewConditionBuilder(traceFieldMapper)
|
|
|
|
fl := flaggertest.New(t)
|
|
traceAggExprRewriter := querybuilder.NewAggExprRewriter(providerSettings, nil, traceFieldMapper, traceConditionBuilder, nil, fl)
|
|
traceStmtBuilder := telemetrytraces.NewTraceQueryStatementBuilder(
|
|
providerSettings,
|
|
metadataStore,
|
|
traceFieldMapper,
|
|
traceConditionBuilder,
|
|
traceAggExprRewriter,
|
|
telemetryStore,
|
|
fl,
|
|
false,
|
|
100000,
|
|
)
|
|
|
|
return querier.New(
|
|
providerSettings,
|
|
telemetryStore,
|
|
metadataStore,
|
|
nil, // prometheus
|
|
traceStmtBuilder, // traceStmtBuilder
|
|
nil, // logStmtBuilder
|
|
nil, // auditStmtBuilder
|
|
nil, // metricStmtBuilder
|
|
nil, // meterStmtBuilder
|
|
nil, // traceOperatorStmtBuilder
|
|
nil, // bucketCache
|
|
fl,
|
|
0,
|
|
)
|
|
}
|