|
|
|
@ -9,26 +9,26 @@ import (
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
// Option for creating new Pipes.
|
|
|
|
|
type Option func(*pipe) |
|
|
|
|
type Option func(*pipeOption) |
|
|
|
|
|
|
|
|
|
// WithoutSizeLimit returns an Option for Pipe to have no size limit.
|
|
|
|
|
func WithoutSizeLimit() Option { |
|
|
|
|
return func(p *pipe) { |
|
|
|
|
p.limit = -1 |
|
|
|
|
return func(opt *pipeOption) { |
|
|
|
|
opt.limit = -1 |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// WithSizeLimit returns an Option for Pipe to have the given size limit.
|
|
|
|
|
func WithSizeLimit(limit int32) Option { |
|
|
|
|
return func(p *pipe) { |
|
|
|
|
p.limit = limit |
|
|
|
|
return func(opt *pipeOption) { |
|
|
|
|
opt.limit = limit |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// DiscardOverflow returns an Option for Pipe to discard writes if full.
|
|
|
|
|
func DiscardOverflow() Option { |
|
|
|
|
return func(p *pipe) { |
|
|
|
|
p.discardOverflow = true |
|
|
|
|
return func(opt *pipeOption) { |
|
|
|
|
opt.discardOverflow = true |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -49,14 +49,16 @@ func OptionsFromContext(ctx context.Context) []Option {
|
|
|
|
|
// New creates a new Reader and Writer that connects to each other.
|
|
|
|
|
func New(opts ...Option) (*Reader, *Writer) { |
|
|
|
|
p := &pipe{ |
|
|
|
|
limit: -1, |
|
|
|
|
readSignal: signal.NewNotifier(), |
|
|
|
|
writeSignal: signal.NewNotifier(), |
|
|
|
|
done: done.New(), |
|
|
|
|
option: pipeOption{ |
|
|
|
|
limit: -1, |
|
|
|
|
}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for _, opt := range opts { |
|
|
|
|
opt(p) |
|
|
|
|
opt(&(p.option)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return &Reader{ |
|
|
|
|