summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-iotools/write.go
blob: e1b44db24a8a1235e8e9b9818ce9591673c65e85 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package iotools

import "io"

// WriterFunc is a function signature which allows
// a function to implement the io.Writer type.
type WriterFunc func([]byte) (int, error)

func (w WriterFunc) Write(b []byte) (int, error) {
	return w(b)
}

// WriterToFunc is a function signature which allows
// a function to implement the io.WriterTo type.
type WriterToFunc func(io.Writer) (int64, error)

func (wt WriterToFunc) WriteTo(r io.Writer) (int64, error) {
	return wt(r)
}

// WriteCloser wraps an io.Writer and io.Closer in order to implement io.WriteCloser.
func WriteCloser(w io.Writer, c io.Closer) io.WriteCloser {
	return &struct {
		io.Writer
		io.Closer
	}{w, c}
}

// NopWriteCloser wraps an io.Writer to implement io.WriteCloser with empty io.Closer implementation.
func NopWriteCloser(w io.Writer) io.WriteCloser {
	return WriteCloser(w, CloserFunc(func() error {
		return nil
	}))
}