Implement write canceling in COBSFramer
This commit is contained in:
parent
cd01f07007
commit
46407b3685
1 changed files with 26 additions and 4 deletions
|
@ -510,12 +510,34 @@ func (c *COBSFramer) WriteFrames(ctx context.Context, writer io.Writer, frameCha
|
||||||
|
|
||||||
encoded := c.encoder.Encode(frame)
|
encoded := c.encoder.Encode(frame)
|
||||||
|
|
||||||
n, err := io.Copy(writer, bytes.NewReader(encoded))
|
err := c.writeRaw(ctx, writer, encoded)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("cold not write data: %w", err)
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
c.logger.Tracef("wrote %d bytes", n)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.logger.WithField("count", n).Tracef("wrote bytes")
|
||||||
|
|
||||||
|
close(errChan)
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return nil
|
||||||
|
case err := <-errChan:
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue