Implement write canceling in COBSFramer

main
Jan Dittberner 1 year ago
parent cd01f07007
commit 46407b3685

@ -510,12 +510,34 @@ func (c *COBSFramer) WriteFrames(ctx context.Context, writer io.Writer, frameCha
encoded := c.encoder.Encode(frame)
n, err := io.Copy(writer, bytes.NewReader(encoded))
err := c.writeRaw(ctx, writer, encoded)
if err != nil {
return fmt.Errorf("cold not write data: %w", err)
return err
}
}
}
}
func (c *COBSFramer) writeRaw(ctx context.Context, writer io.Writer, raw []byte) error {
errChan := make(chan error)
go func() {
n, err := io.Copy(writer, bytes.NewReader(raw))
if err != nil {
errChan <- fmt.Errorf("could not write data: %w", err)
c.logger.Tracef("wrote %d bytes", n)
return
}
c.logger.WithField("count", n).Tracef("wrote bytes")
close(errChan)
}()
select {
case <-ctx.Done():
return nil
case err := <-errChan:
return err
}
}

Loading…
Cancel
Save