-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
encode.go
67 lines (58 loc) · 2.27 KB
/
encode.go
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
60
61
62
63
64
65
66
67
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
package batch // import "github.com/open-telemetry/opentelemetry-collector-contrib/exporter/awskinesisexporter/internal/batch"
import (
"errors"
"go.opentelemetry.io/collector/pdata/plog"
"go.opentelemetry.io/collector/pdata/pmetric"
"go.opentelemetry.io/collector/pdata/ptrace"
"github.com/open-telemetry/opentelemetry-collector-contrib/exporter/awskinesisexporter/internal/key"
"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/translator/zipkin/zipkinv2"
)
var (
// ErrUnsupportedEncoding is used when the encoder type does not support the type of encoding
ErrUnsupportedEncoding = errors.New("unsupported type to encode")
// ErrUnknownExportEncoder is used when a named encoding doesn't not exist
ErrUnknownExportEncoder = errors.New("unknown encoding export format")
)
// Encoder transforms the internal pipeline format into a configurable
// format that is then used to export to kinesis.
type Encoder interface {
Metrics(md pmetric.Metrics) (*Batch, error)
Traces(td ptrace.Traces) (*Batch, error)
Logs(ld plog.Logs) (*Batch, error)
}
func NewEncoder(named string, batchOptions ...Option) (Encoder, error) {
bm := &batchMarshaller{
batchOptions: batchOptions,
partitioner: key.Randomized,
logsMarshaller: unsupported{},
tracesMarshaller: unsupported{},
metricsMarshaller: unsupported{},
}
switch named {
case "zipkin_proto":
bm.tracesMarshaller = zipkinv2.NewProtobufTracesMarshaler()
case "zipkin_json":
bm.tracesMarshaller = zipkinv2.NewJSONTracesMarshaler()
case "otlp", "otlp_proto":
bm.logsMarshaller = &plog.ProtoMarshaler{}
bm.metricsMarshaller = &pmetric.ProtoMarshaler{}
bm.tracesMarshaller = &ptrace.ProtoMarshaler{}
case "otlp_json":
bm.logsMarshaller = &plog.JSONMarshaler{}
bm.metricsMarshaller = &pmetric.JSONMarshaler{}
bm.tracesMarshaller = &ptrace.JSONMarshaler{}
case "jaeger_proto":
// Jaeger encoding is a special case
// since the internal libraries offer no means of ptrace.TraceMarshaller.
// In order to preserve historical behavior, a custom type
// is used until it can be replaced.
return &jaegerEncoder{
batchOptions: batchOptions,
}, nil
default:
return nil, ErrUnknownExportEncoder
}
return bm, nil
}