elasticapmreceiver/factory.go at main · ldgrp/elasticapmreceiver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package elasticapmreceiver

import (
	"context"

	"go.opentelemetry.io/collector/component"
	"go.opentelemetry.io/collector/config/confighttp"
	"go.opentelemetry.io/collector/consumer"
	"go.opentelemetry.io/collector/receiver"
)

const (
	typeStr                  = "elasticapm"
	defaultHTTPEndpoint      = "0.0.0.0:8200"
	defaultEventsURLPath     = "/intake/v2/events"
	defaultRUMEventsURLPath  = "/intake/v2/rum/events"
	defaultMaxEventSizeBytes = 300 * 1024
	defaultBatchSize         = 10
)

func NewFactory() receiver.Factory {
	return receiver.NewFactory(
		typeStr,
		createDefaultConfig,
		receiver.WithTraces(createTraces, component.StabilityLevelDevelopment),
	)
}

func createDefaultConfig() component.Config {
	return &Config{
		HTTPServerSettings: &confighttp.HTTPServerSettings{
			Endpoint: defaultHTTPEndpoint,
		},
		EventsURLPath:    defaultEventsURLPath,
		RUMEventsUrlPath: defaultRUMEventsURLPath,
		MaxEventSize:     defaultMaxEventSizeBytes,
		BatchSize:        defaultBatchSize,
	}
}

func createTraces(
	_ context.Context,
	params receiver.CreateSettings,
	baseCfg component.Config,
	nextConsumer consumer.Traces,
) (receiver.Traces, error) {
	cfg := baseCfg.(*Config)
	r, err := newElasticAPMReceiver(cfg, params)

	if err != nil {
		return nil, err
	}

	if err = r.registerTraceConsumer(nextConsumer); err != nil {
		return nil, err
	}

	return r, nil
}