Merge pull request #15 from spectrumjade/hec_interface

Add WithContext methods to the HEC interface
eventbrite · Apr 14, 2019 · 10df423 · 10df423
2 parents bc8861e + dd0f53c
commit 10df423
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/hec.go b/hec.go
@@ -19,9 +19,15 @@ type HEC interface {
 	// WriteBatch writes multiple events via HCE batch mode
 	WriteBatch(events []*Event) error
 
+	// WriteBatchWithContext writes multiple events via HEC batch mode with a context for cancellation
+	WriteBatchWithContext(ctx context.Context, events []*Event) error
+
 	// WriteRaw writes raw data stream via HEC raw mode
 	WriteRaw(reader io.ReadSeeker, metadata *EventMetadata) error
 
+	// WriteRawWithContext writes raw data stream via HEC raw mode with a context for cancellation
+	WriteRawWithContext(ctx context.Context, reader io.ReadSeeker, metadata *EventMetadata) error
+
 	// WaitForAcknowledgement blocks until the Splunk indexer acknowledges data sent to it
 	WaitForAcknowledgement() error