Skip to content

Commit ac1bace

Browse files
gavincabbagesbylica-splunk
authored andcommitted
[receiver/awsfirehosereceiver] Add support for CloudWatch logs (open-telemetry#35077)
**Description:** Add support for CloudWatch logs in JSON format. The receiver can now be configured to receiver either `cwmetrics` or `cwlogs` record type for CloudWatch metrics logs, respectively. The former is only valid for metrics receivers and the latter for logs receivers. Some changes were necessary to the default record type handling to conform to the contracts required by the generated tests and ensure the default config returned by the factory was usable. Previously the default record type was `cwmetrics` but this is not sensible for logs receivers. To accommodate this, the default record type is now telemetry-type-specific. That is, for metrics it is `cwmetrics` and for logs it is `cwlogs`. Each receiver makes this determination on its own. Therefore an empty string is now the default returned by the factory and an empty string is also acceptable in the users config. It is no longer required to specify a record type, although it is still supported, so this is not a breaking change. **Link to tracking Issue:** n/a **Testing:** Unit tests covering new code as well as manual smoke testing against AWS Firehose and CloudWatch themselves. **Documentation:** Component README update.
1 parent dd596b5 commit ac1bace

31 files changed

+784
-43
lines changed
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
# Use this changelog template to create an entry for release notes.
2+
3+
# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
4+
change_type: enhancement
5+
6+
# The name of the component, or a single word describing the area of concern, (e.g. filelogreceiver)
7+
component: awsfirehosereceiver
8+
9+
# A brief description of the change. Surround your text with quotes ("") if it needs to start with a backtick (`).
10+
note: Add support for CloudWatch logs
11+
12+
# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
13+
issues: [35077]
14+
15+
# (Optional) One or more lines of additional information to render under the primary note.
16+
# These lines will be padded with 2 spaces and then inserted directly into the document.
17+
# Use pipe (|) for multiline entries.
18+
subtext:
19+
20+
# If your change doesn't affect end users or the exported elements of any package,
21+
# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
22+
# Optional: The change log or logs in which this entry should be included.
23+
# e.g. '[user]' or '[user, api]'
24+
# Include 'user' if the change is relevant to end users.
25+
# Include 'api' if there is a change to a library API.
26+
# Default: '[user]'
27+
change_logs: []

receiver/awsfirehosereceiver/README.md

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
<!-- status autogenerated section -->
44
| Status | |
55
| ------------- |-----------|
6-
| Stability | [alpha]: metrics |
6+
| Stability | [alpha]: metrics, logs |
77
| Distributions | [contrib] |
88
| Issues | [![Open issues](https://img.shields.io/github/issues-search/open-telemetry/opentelemetry-collector-contrib?query=is%3Aissue%20is%3Aopen%20label%3Areceiver%2Fawsfirehose%20&label=open&color=orange&logo=opentelemetry)](https://github.com/open-telemetry/opentelemetry-collector-contrib/issues?q=is%3Aopen+is%3Aissue+label%3Areceiver%2Fawsfirehose) [![Closed issues](https://img.shields.io/github/issues-search/open-telemetry/opentelemetry-collector-contrib?query=is%3Aissue%20is%3Aclosed%20label%3Areceiver%2Fawsfirehose%20&label=closed&color=blue&logo=opentelemetry)](https://github.com/open-telemetry/opentelemetry-collector-contrib/issues?q=is%3Aclosed+is%3Aissue+label%3Areceiver%2Fawsfirehose) |
99
| [Code Owners](https://github.com/open-telemetry/opentelemetry-collector-contrib/blob/main/CONTRIBUTING.md#becoming-a-code-owner) | [@Aneurysm9](https://www.github.com/Aneurysm9) |
@@ -62,3 +62,24 @@ See [documentation](https://docs.aws.amazon.com/firehose/latest/dev/create-desti
6262
The record type for the CloudWatch metric stream. Expects the format for the records to be JSON.
6363
See [documentation](https://docs.aws.amazon.com/AmazonCloudWatch/latest/monitoring/CloudWatch-Metric-Streams.html) for details.
6464

65+
### cwlogs
66+
The record type for the CloudWatch log stream. Expects the format for the records to be JSON.
67+
For example:
68+
69+
```json
70+
{
71+
"messageType": "DATA_MESSAGE",
72+
"owner": "111122223333",
73+
"logGroup": "my-log-group",
74+
"logStream": "my-log-stream",
75+
"subscriptionFilters": ["my-subscription-filter"],
76+
"logEvents": [
77+
{
78+
"id": "123",
79+
"timestamp": 1725544035523,
80+
"message": "My log message."
81+
}
82+
]
83+
}
84+
```
85+

receiver/awsfirehosereceiver/config.go

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,10 @@ func (c *Config) Validate() error {
3030
if c.Endpoint == "" {
3131
return errors.New("must specify endpoint")
3232
}
33-
if c.RecordType == "" {
34-
return errors.New("must specify record type")
33+
// If a record type is specified, it must be valid.
34+
// An empty string is acceptable, however, because it will use a telemetry-type-specific default.
35+
if c.RecordType != "" {
36+
return validateRecordType(c.RecordType)
3537
}
36-
return validateRecordType(c.RecordType)
38+
return nil
3739
}

receiver/awsfirehosereceiver/config_test.go

Lines changed: 37 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
package awsfirehosereceiver
55

66
import (
7+
"fmt"
78
"path/filepath"
89
"testing"
910

@@ -18,29 +19,40 @@ import (
1819
)
1920

2021
func TestLoadConfig(t *testing.T) {
21-
cm, err := confmaptest.LoadConf(filepath.Join("testdata", "config.yaml"))
22-
require.NoError(t, err)
23-
24-
factory := NewFactory()
25-
cfg := factory.CreateDefaultConfig()
26-
27-
sub, err := cm.Sub(component.NewIDWithName(metadata.Type, "").String())
28-
require.NoError(t, err)
29-
require.NoError(t, sub.Unmarshal(cfg))
30-
31-
assert.NoError(t, component.ValidateConfig(cfg))
32-
33-
require.Equal(t, &Config{
34-
RecordType: "cwmetrics",
35-
AccessKey: "some_access_key",
36-
ServerConfig: confighttp.ServerConfig{
37-
Endpoint: "0.0.0.0:4433",
38-
TLSSetting: &configtls.ServerConfig{
39-
Config: configtls.Config{
40-
CertFile: "server.crt",
41-
KeyFile: "server.key",
42-
},
43-
},
44-
},
45-
}, cfg)
22+
for _, configType := range []string{
23+
"cwmetrics", "cwlogs", "invalid",
24+
} {
25+
t.Run(configType, func(t *testing.T) {
26+
fileName := fmt.Sprintf("%s_config.yaml", configType)
27+
cm, err := confmaptest.LoadConf(filepath.Join("testdata", fileName))
28+
require.NoError(t, err)
29+
30+
factory := NewFactory()
31+
cfg := factory.CreateDefaultConfig()
32+
33+
sub, err := cm.Sub(component.NewIDWithName(metadata.Type, "").String())
34+
require.NoError(t, err)
35+
require.NoError(t, sub.Unmarshal(cfg))
36+
37+
err = component.ValidateConfig(cfg)
38+
if configType == "invalid" {
39+
assert.Error(t, err)
40+
} else {
41+
assert.NoError(t, err)
42+
require.Equal(t, &Config{
43+
RecordType: configType,
44+
AccessKey: "some_access_key",
45+
ServerConfig: confighttp.ServerConfig{
46+
Endpoint: "0.0.0.0:4433",
47+
TLSSetting: &configtls.ServerConfig{
48+
Config: configtls.Config{
49+
CertFile: "server.crt",
50+
KeyFile: "server.key",
51+
},
52+
},
53+
},
54+
}, cfg)
55+
}
56+
})
57+
}
4658
}

receiver/awsfirehosereceiver/factory.go

Lines changed: 24 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,19 +16,20 @@ import (
1616
"github.com/open-telemetry/opentelemetry-collector-contrib/internal/common/testutil"
1717
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/metadata"
1818
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler"
19+
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler/cwlog"
1920
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler/cwmetricstream"
2021
)
2122

2223
const (
23-
defaultRecordType = cwmetricstream.TypeStr
24-
defaultEndpoint = "0.0.0.0:4433"
25-
defaultPort = 4433
24+
defaultEndpoint = "0.0.0.0:4433"
25+
defaultPort = 4433
2626
)
2727

2828
var (
2929
errUnrecognizedRecordType = errors.New("unrecognized record type")
3030
availableRecordTypes = map[string]bool{
3131
cwmetricstream.TypeStr: true,
32+
cwlog.TypeStr: true,
3233
}
3334
)
3435

@@ -38,7 +39,8 @@ func NewFactory() receiver.Factory {
3839
return receiver.NewFactory(
3940
metadata.Type,
4041
createDefaultConfig,
41-
receiver.WithMetrics(createMetricsReceiver, metadata.MetricsStability))
42+
receiver.WithMetrics(createMetricsReceiver, metadata.MetricsStability),
43+
receiver.WithLogs(createLogsReceiver, metadata.LogsStability))
4244
}
4345

4446
// validateRecordType checks the available record types for the
@@ -59,11 +61,18 @@ func defaultMetricsUnmarshalers(logger *zap.Logger) map[string]unmarshaler.Metri
5961
}
6062
}
6163

64+
// defaultLogsUnmarshalers creates a map of the available logs unmarshalers.
65+
func defaultLogsUnmarshalers(logger *zap.Logger) map[string]unmarshaler.LogsUnmarshaler {
66+
u := cwlog.NewUnmarshaler(logger)
67+
return map[string]unmarshaler.LogsUnmarshaler{
68+
u.Type(): u,
69+
}
70+
}
71+
6272
// createDefaultConfig creates a default config with the endpoint set
6373
// to port 8443 and the record type set to the CloudWatch metric stream.
6474
func createDefaultConfig() component.Config {
6575
return &Config{
66-
RecordType: defaultRecordType,
6776
ServerConfig: confighttp.ServerConfig{
6877
Endpoint: testutil.EndpointForPort(defaultPort),
6978
},
@@ -79,3 +88,13 @@ func createMetricsReceiver(
7988
) (receiver.Metrics, error) {
8089
return newMetricsReceiver(cfg.(*Config), set, defaultMetricsUnmarshalers(set.Logger), nextConsumer)
8190
}
91+
92+
// createMetricsReceiver implements the CreateMetricsReceiver function type.
93+
func createLogsReceiver(
94+
_ context.Context,
95+
set receiver.Settings,
96+
cfg component.Config,
97+
nextConsumer consumer.Logs,
98+
) (receiver.Logs, error) {
99+
return newLogsReceiver(cfg.(*Config), set, defaultLogsUnmarshalers(set.Logger), nextConsumer)
100+
}

receiver/awsfirehosereceiver/factory_test.go

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,19 @@ func TestCreateMetrics(t *testing.T) {
2929
require.NotNil(t, r)
3030
}
3131

32+
func TestCreateLogsReceiver(t *testing.T) {
33+
r, err := createLogsReceiver(
34+
context.Background(),
35+
receivertest.NewNopSettings(),
36+
createDefaultConfig(),
37+
consumertest.NewNop(),
38+
)
39+
require.NoError(t, err)
40+
require.NotNil(t, r)
41+
}
42+
3243
func TestValidateRecordType(t *testing.T) {
33-
require.NoError(t, validateRecordType(defaultRecordType))
44+
require.NoError(t, validateRecordType(defaultMetricsRecordType))
45+
require.NoError(t, validateRecordType(defaultLogsRecordType))
3446
require.Error(t, validateRecordType("nop"))
3547
}

receiver/awsfirehosereceiver/generated_component_test.go

Lines changed: 7 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

receiver/awsfirehosereceiver/internal/metadata/generated_status.go

Lines changed: 1 addition & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
// Copyright The OpenTelemetry Authors
2+
// SPDX-License-Identifier: Apache-2.0
3+
4+
package compression // import "github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler/cwlog/compression"
5+
6+
import (
7+
"bytes"
8+
"compress/gzip"
9+
)
10+
11+
// Zip returns a gzip-compressed representation of the input bytes.
12+
func Zip(data []byte) ([]byte, error) {
13+
var b bytes.Buffer
14+
w := gzip.NewWriter(&b)
15+
16+
_, err := w.Write(data)
17+
if err != nil {
18+
return nil, err
19+
}
20+
21+
if err = w.Flush(); err != nil {
22+
return nil, err
23+
}
24+
25+
if err = w.Close(); err != nil {
26+
return nil, err
27+
}
28+
29+
return b.Bytes(), nil
30+
}
31+
32+
// Unzip expects gzip-compressed input bytes and returns their uncompressed form.
33+
func Unzip(data []byte) ([]byte, error) {
34+
b := bytes.NewBuffer(data)
35+
36+
r, err := gzip.NewReader(b)
37+
if err != nil {
38+
return nil, err
39+
}
40+
41+
var rv bytes.Buffer
42+
_, err = rv.ReadFrom(r)
43+
if err != nil {
44+
return nil, err
45+
}
46+
47+
return rv.Bytes(), nil
48+
}
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
// Copyright The OpenTelemetry Authors
2+
// SPDX-License-Identifier: Apache-2.0
3+
4+
package cwlog // import "github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler/cwlog"
5+
6+
type cWLog struct {
7+
MessageType string `json:"messageType"`
8+
Owner string `json:"owner"`
9+
LogGroup string `json:"logGroup"`
10+
LogStream string `json:"logStream"`
11+
SubscriptionFilters []string `json:"subscriptionFilters"`
12+
LogEvents []struct {
13+
ID string `json:"id"`
14+
Timestamp int64 `json:"timestamp"`
15+
Message string `json:"message"`
16+
} `json:"logEvents"`
17+
}
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
// Copyright The OpenTelemetry Authors
2+
// SPDX-License-Identifier: Apache-2.0
3+
4+
package cwlog // import "github.com/open-telemetry/opentelemetry-collector-contrib/receiver/awsfirehosereceiver/internal/unmarshaler/cwlog"
5+
6+
import (
7+
"go.opentelemetry.io/collector/pdata/pcommon"
8+
"go.opentelemetry.io/collector/pdata/plog"
9+
conventions "go.opentelemetry.io/collector/semconv/v1.6.1"
10+
)
11+
12+
const (
13+
attributeAWSCloudWatchLogGroupName = "aws.cloudwatch.log_group_name"
14+
attributeAWSCloudWatchLogStreamName = "aws.cloudwatch.log_stream_name"
15+
)
16+
17+
// resourceAttributes are the CloudWatch log attributes that define a unique resource.
18+
type resourceAttributes struct {
19+
owner, logGroup, logStream string
20+
}
21+
22+
// resourceLogsBuilder provides convenient access to the a Resource's LogRecordSlice.
23+
type resourceLogsBuilder struct {
24+
rls plog.LogRecordSlice
25+
}
26+
27+
// setAttributes applies the resourceAttributes to the provided Resource.
28+
func (ra *resourceAttributes) setAttributes(resource pcommon.Resource) {
29+
attrs := resource.Attributes()
30+
attrs.PutStr(conventions.AttributeCloudAccountID, ra.owner)
31+
attrs.PutStr(attributeAWSCloudWatchLogGroupName, ra.logGroup)
32+
attrs.PutStr(attributeAWSCloudWatchLogStreamName, ra.logStream)
33+
}
34+
35+
// newResourceLogsBuilder to capture logs for the Resource defined by the provided attributes.
36+
func newResourceLogsBuilder(logs plog.Logs, attrs resourceAttributes) *resourceLogsBuilder {
37+
rls := logs.ResourceLogs().AppendEmpty()
38+
attrs.setAttributes(rls.Resource())
39+
return &resourceLogsBuilder{rls.ScopeLogs().AppendEmpty().LogRecords()}
40+
}
41+
42+
// AddLog events to the LogRecordSlice. Resource attributes are captured when creating
43+
// the resourceLogsBuilder, so we only need to consider the LogEvents themselves.
44+
func (rlb *resourceLogsBuilder) AddLog(log cWLog) {
45+
for _, event := range log.LogEvents {
46+
logLine := rlb.rls.AppendEmpty()
47+
logLine.SetTimestamp(pcommon.Timestamp(event.Timestamp))
48+
logLine.Body().SetStr(event.Message)
49+
}
50+
}
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
{"CHANGE":-0.09,"PRICE":4.96,"TICKER_SYMBOL":"KIN","SECTOR":"ENERGY"}
2+
{"CHANGE":-1.47,"PRICE":134.74,"TICKER_SYMBOL":"DFG","SECTOR":"TECHNOLOGY"}
3+
{"CHANGE":1.96,"PRICE":57.53,"TICKER_SYMBOL":"SAC","SECTOR":"ENERGY"}
4+
{"CHANGE":0.04,"PRICE":32.84,"TICKER_SYMBOL":"PJN","SECTOR":"RETAIL"}
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test","logStream":"test","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695232","timestamp":1725544035523,"message":"Hello world, here is our first log message!"}]}
2+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test","logStream":"test","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695233","timestamp":1725554035523,"message":"Hello world, here is our second log message!"}]}
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test","logStream":"test","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695232","timestamp":1725544035523,"message":"Hello world, here is our first log message!"}]}
2+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test","logStream":"test","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695233","timestamp":1725554035523,"message":"Hello world, here is our second log message!"}]}
3+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test2","logStream":"test1","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695234","timestamp":1725564035523,"message":"Hello world, here is our third log message!"}]}
4+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test2","logStream":"test2","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695235","timestamp":1725574035523,"message":"Hello world, here is our fourth log message!"}]}
5+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test2","logStream":"test1","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695236","timestamp":1725584035523,"message":"Hello world, here is our fifth log message!"}]}
6+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test2","logStream":"test2","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695237","timestamp":1725594035523,"message":"Hello world, here is our sixth log message!"}]}
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
{"messageType":"DATA_MESSAGE","owner":"123","logGroup":"test","logStream":"test","subscriptionFilters":["test"],"logEvents":[{"id":"38480917865042697267627490045603633139480491071049695232","timestamp":1725544035523,"message":"Hello world, here is our first log message!"}]}

0 commit comments

Comments
 (0)