-
Notifications
You must be signed in to change notification settings - Fork 6
/
related_data.go
90 lines (80 loc) · 2.67 KB
/
related_data.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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
/*
* Copyright The OpenTelemetry Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package otlp
import (
colarspb "github.com/f5/otel-arrow-adapter/api/experimental/arrow/v1"
"github.com/f5/otel-arrow-adapter/pkg/otel"
"github.com/f5/otel-arrow-adapter/pkg/otel/common/otlp"
"github.com/f5/otel-arrow-adapter/pkg/record_message"
"github.com/f5/otel-arrow-adapter/pkg/werror"
)
// Infrastructure used to process related records.
type (
RelatedData struct {
LogRecordID uint16
ResAttrMapStore *otlp.Attributes16Store
ScopeAttrMapStore *otlp.Attributes16Store
LogRecordAttrMapStore *otlp.Attributes16Store
}
)
func NewRelatedData() *RelatedData {
return &RelatedData{
ResAttrMapStore: otlp.NewAttributes16Store(),
ScopeAttrMapStore: otlp.NewAttributes16Store(),
LogRecordAttrMapStore: otlp.NewAttributes16Store(),
}
}
func (r *RelatedData) LogRecordIDFromDelta(delta uint16) uint16 {
r.LogRecordID += delta
return r.LogRecordID
}
func RelatedDataFrom(records []*record_message.RecordMessage) (relatedData *RelatedData, logsRecord *record_message.RecordMessage, err error) {
defer func() {
for _, record := range records {
record.Record().Release()
}
}()
relatedData = NewRelatedData()
// Create the attribute map stores for all the attribute records.
for _, record := range records {
switch record.PayloadType() {
case colarspb.ArrowPayloadType_RESOURCE_ATTRS:
err = otlp.Attributes16StoreFrom(record.Record(), relatedData.ResAttrMapStore)
if err != nil {
return nil, nil, werror.Wrap(err)
}
case colarspb.ArrowPayloadType_SCOPE_ATTRS:
err = otlp.Attributes16StoreFrom(record.Record(), relatedData.ScopeAttrMapStore)
if err != nil {
return nil, nil, werror.Wrap(err)
}
case colarspb.ArrowPayloadType_LOG_ATTRS:
err = otlp.Attributes16StoreFrom(record.Record(), relatedData.LogRecordAttrMapStore)
if err != nil {
return nil, nil, werror.Wrap(err)
}
case colarspb.ArrowPayloadType_LOGS:
if logsRecord != nil {
return nil, nil, werror.Wrap(otel.ErrMultipleTracesRecords)
}
logsRecord = record
default:
return nil, nil, werror.Wrap(otel.UnknownPayloadType)
}
}
return
}