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).
157 lines
4.8 KiB
Go
157 lines
4.8 KiB
Go
package segmentanalytics
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
|
|
"github.com/hanzoai/o11y/pkg/analytics"
|
|
"github.com/hanzoai/o11y/pkg/errors"
|
|
"github.com/hanzoai/o11y/pkg/factory"
|
|
"github.com/hanzoai/o11y/pkg/types/analyticstypes"
|
|
segment "github.com/segmentio/analytics-go/v3"
|
|
)
|
|
|
|
type provider struct {
|
|
settings factory.ScopedProviderSettings
|
|
client segment.Client
|
|
stopC chan struct{}
|
|
}
|
|
|
|
func NewFactory() factory.ProviderFactory[analytics.Analytics, analytics.Config] {
|
|
return factory.NewProviderFactory(factory.MustNewName("segment"), New)
|
|
}
|
|
|
|
func New(ctx context.Context, providerSettings factory.ProviderSettings, config analytics.Config) (analytics.Analytics, error) {
|
|
settings := factory.NewScopedProviderSettings(providerSettings, "github.com/hanzoai/o11y/pkg/analytics/segmentanalytics")
|
|
|
|
client, err := segment.NewWithConfig(config.Segment.Key, segment.Config{
|
|
Logger: newSegmentLogger(settings),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &provider{
|
|
settings: settings,
|
|
client: client,
|
|
stopC: make(chan struct{}),
|
|
}, nil
|
|
}
|
|
|
|
func (provider *provider) Start(_ context.Context) error {
|
|
<-provider.stopC
|
|
return nil
|
|
}
|
|
|
|
func (provider *provider) Send(ctx context.Context, messages ...analyticstypes.Message) {
|
|
for _, message := range messages {
|
|
err := provider.client.Enqueue(message)
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (provider *provider) TrackGroup(ctx context.Context, group, event string, properties map[string]any) {
|
|
if properties == nil {
|
|
provider.settings.Logger().WarnContext(ctx, "empty attributes received, skipping event", slog.String("group", group), slog.String("event", event))
|
|
return
|
|
}
|
|
|
|
err := provider.client.Enqueue(analyticstypes.Track{
|
|
UserId: "stats_" + group,
|
|
Event: event,
|
|
Properties: analyticstypes.NewPropertiesFromMap(properties),
|
|
Context: &analyticstypes.Context{
|
|
Extra: map[string]interface{}{
|
|
analyticstypes.KeyGroupID: group,
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
}
|
|
|
|
func (provider *provider) TrackUser(ctx context.Context, group, user, event string, properties map[string]any) {
|
|
if properties == nil {
|
|
provider.settings.Logger().WarnContext(ctx, "empty attributes received, skipping event", slog.String("user", user), slog.String("group", group), slog.String("event", event))
|
|
return
|
|
}
|
|
|
|
err := provider.client.Enqueue(analyticstypes.Track{
|
|
UserId: user,
|
|
Event: event,
|
|
Properties: analyticstypes.NewPropertiesFromMap(properties),
|
|
Context: &analyticstypes.Context{
|
|
Extra: map[string]interface{}{
|
|
analyticstypes.KeyGroupID: group,
|
|
},
|
|
},
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
}
|
|
|
|
func (provider *provider) IdentifyGroup(ctx context.Context, group string, traits map[string]any) {
|
|
if traits == nil {
|
|
provider.settings.Logger().WarnContext(ctx, "empty attributes received, skipping identify", slog.String("group", group))
|
|
return
|
|
}
|
|
|
|
// identify the user
|
|
err := provider.client.Enqueue(analyticstypes.Identify{
|
|
UserId: "stats_" + group,
|
|
Traits: analyticstypes.NewTraitsFromMap(traits),
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
|
|
// identify the group using the stats user
|
|
err = provider.client.Enqueue(analyticstypes.Group{
|
|
UserId: "stats_" + group,
|
|
GroupId: group,
|
|
Traits: analyticstypes.NewTraitsFromMap(traits),
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
}
|
|
|
|
func (provider *provider) IdentifyUser(ctx context.Context, group, user string, traits map[string]any) {
|
|
if traits == nil {
|
|
provider.settings.Logger().WarnContext(ctx, "empty attributes received, skipping identify", slog.String("user", user), slog.String("group", group))
|
|
return
|
|
}
|
|
|
|
// identify the user
|
|
err := provider.client.Enqueue(analyticstypes.Identify{
|
|
UserId: user,
|
|
Traits: analyticstypes.NewTraitsFromMap(traits),
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
|
|
// associate the user with the group
|
|
err = provider.client.Enqueue(analyticstypes.Group{
|
|
UserId: user,
|
|
GroupId: group,
|
|
Traits: analyticstypes.NewTraits().Set("id", group), // A trait is required
|
|
})
|
|
if err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to send message to segment", errors.Attr(err))
|
|
}
|
|
}
|
|
|
|
func (provider *provider) Stop(ctx context.Context) error {
|
|
if err := provider.client.Close(); err != nil {
|
|
provider.settings.Logger().WarnContext(ctx, "unable to close segment client", errors.Attr(err))
|
|
}
|
|
|
|
close(provider.stopC)
|
|
return nil
|
|
}
|