Skip to content

Commit

Permalink
chore: add options for retry/storage
Browse files Browse the repository at this point in the history
  • Loading branch information
VihasMakwana committed Dec 11, 2024
1 parent ccf3cae commit fa7f573
Showing 1 changed file with 16 additions and 2 deletions.
18 changes: 16 additions & 2 deletions testbed/datasenders/stanza.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,9 @@ import (
)

type FileLogWriter struct {
file *os.File
file *os.File
retry string
storage string
}

// Ensure FileLogWriter implements LogDataSender.
Expand All @@ -41,6 +43,16 @@ func NewFileLogWriter() *FileLogWriter {
return f
}

func (f *FileLogWriter) WithRetry(retry string) *FileLogWriter {
f.retry = retry
return f
}

func (f *FileLogWriter) WithStorage(storage string) *FileLogWriter {
f.storage = storage
return f
}

func (f *FileLogWriter) Capabilities() consumer.Capabilities {
return consumer.Capabilities{MutatesData: false}
}
Expand Down Expand Up @@ -120,7 +132,9 @@ func (f *FileLogWriter) GenConfigYAMLStr() string {
layout: '%%Y-%%m-%%d'
severity:
parse_from: attributes.sev
`, f.file.Name())
%s
%s
`, f.file.Name(), f.retry, f.storage)
}

func (f *FileLogWriter) ProtocolName() string {
Expand Down

0 comments on commit fa7f573

Please sign in to comment.