77 lines
1.1 KiB
Go
77 lines
1.1 KiB
Go
|
package gemini
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type contextReader struct {
|
||
|
|
ctx context.Context
|
||
|
|
done <-chan struct{}
|
||
|
|
cancel func()
|
||
|
|
rc io.ReadCloser
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *contextReader) Read(p []byte) (int, error) {
|
||
|
|
select {
|
||
|
|
case <-r.done:
|
||
|
|
r.rc.Close()
|
||
|
|
return 0, r.ctx.Err()
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
n, err := r.rc.Read(p)
|
||
|
|
if err != nil {
|
||
|
|
r.cancel()
|
||
|
|
}
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *contextReader) Close() error {
|
||
|
|
r.cancel()
|
||
|
|
return r.rc.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
type contextWriter struct {
|
||
|
|
ctx context.Context
|
||
|
|
done <-chan struct{}
|
||
|
|
cancel func()
|
||
|
|
wc io.WriteCloser
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *contextWriter) Write(b []byte) (int, error) {
|
||
|
|
select {
|
||
|
|
case <-w.done:
|
||
|
|
w.wc.Close()
|
||
|
|
return 0, w.ctx.Err()
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
n, err := w.wc.Write(b)
|
||
|
|
if err != nil {
|
||
|
|
w.cancel()
|
||
|
|
}
|
||
|
|
return n, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *contextWriter) Close() error {
|
||
|
|
w.cancel()
|
||
|
|
return w.wc.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
type nopCloser struct {
|
||
|
|
io.Writer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (nopCloser) Close() error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type nopReadCloser struct{}
|
||
|
|
|
||
|
|
func (nopReadCloser) Read(p []byte) (int, error) {
|
||
|
|
return 0, io.EOF
|
||
|
|
}
|
||
|
|
|
||
|
|
func (nopReadCloser) Close() error {
|
||
|
|
return nil
|
||
|
|
}
|