diff --git a/hec.go b/hec.go
index 39aa203..76de35a 100644
--- 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