Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: add field key option to set event partition key #11076

Merged
merged 3 commits into from
May 24, 2022
Merged

feat: add field key option to set event partition key #11076

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6d7944c
feat: add field key option to set event partition key
powersj Mar 7, 2022
008c532
allow either a tag or field
powersj May 10, 2022
62cc244
update readme per review
powersj May 24, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions plugins/outputs/event_hubs/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,15 @@ JSON is probably the easiest to integrate with downstream components.
## The full connection string to the Event Hub (required)
## The shared access key must have "Send" permissions on the target Event Hub.
connection_string = "Endpoint=sb://namespace.servicebus.windows.net/;SharedAccessKeyName=RootManageSharedAccessKey;SharedAccessKey=superSecret1234=;EntityPath=hubName"

## Client timeout (defaults to 30s)
# timeout = "30s"

## Partition key
## Metric tag or field name to use for the event partition key. The value of
## this tag or field is set as the key for events if it exists.
# partition_key = ""
srebhan marked this conversation as resolved.
Show resolved Hide resolved

## Data format to output.
## Each data format has its own unique set of configuration options, read
## more about them here:
Expand Down
16 changes: 14 additions & 2 deletions plugins/outputs/event_hubs/event_hubs.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,8 @@ func (eh *eventHub) SendBatch(ctx context.Context, iterator eventhub.BatchIterat
type EventHubs struct {
Log telegraf.Logger `toml:"-"`
ConnectionString string `toml:"connection_string"`
Timeout config.Duration
Timeout config.Duration `toml:"timeout"`
PartitionKey string `toml:"partition_key"`

Hub EventHubInterface
serializer serializers.Serializer
Expand Down Expand Up @@ -102,7 +103,18 @@ func (e *EventHubs) Write(metrics []telegraf.Metric) error {
continue
}

events = append(events, eventhub.NewEvent(payload))
event := eventhub.NewEvent(payload)
if e.PartitionKey != "" {
if key, ok := metric.GetTag(e.PartitionKey); ok {
event.PartitionKey = &key
} else if key, ok := metric.GetField(e.PartitionKey); ok {
if strKey, ok := key.(string); ok {
event.PartitionKey = &strKey
}
}
}

events = append(events, event)
}

ctx, cancel := context.WithTimeout(context.Background(), time.Duration(e.Timeout))
Expand Down