Skip to content

Commit e6317e4

Browse files
authored
accumulator: add error handling for writing to buffer (#569)
* accumulator: add error handling for writing to buffer Errors were ignored so far when trying to write to the buffer. * Update accumulator/batch.go * Apply suggestions from code review * Apply suggestions from code review
1 parent cedd35b commit e6317e4

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

accumulator/batch.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,10 @@ func (b *Batch) OnAgentInit(reqID, contentEncoding string, raw []byte) error {
151151
b.mu.Lock()
152152
defer b.mu.Unlock()
153153
if b.metadataBytes == 0 && len(metadata) > 0 {
154-
b.metadataBytes, _ = b.buf.Write(metadata)
154+
b.metadataBytes, err = b.buf.Write(metadata)
155+
if err != nil {
156+
return fmt.Errorf("failed to write metadata to buffer: %v", err)
157+
}
155158
}
156159
i, ok := b.invocations[reqID]
157160
if !ok {
@@ -195,7 +198,10 @@ func (b *Batch) AddAgentData(apmData APMData) error {
195198
// first line being metadata.
196199
data, after, _ := bytes.Cut(raw, newLineSep)
197200
if b.metadataBytes == 0 {
198-
b.metadataBytes, _ = b.buf.Write(data)
201+
b.metadataBytes, err = b.buf.Write(data)
202+
if err != nil {
203+
return fmt.Errorf("failed to write data to buffer: %v", err)
204+
}
199205
}
200206
for {
201207
data, after, _ = bytes.Cut(after, newLineSep)

0 commit comments

Comments
 (0)