blob: 3adc5f2037049f9d7e13ae2f86b512b845030713 (
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/* Copyright (C) 2021 Nicolas Peugnet <n.peugnet@free.fr>
This file is part of dna-backup.
dna-backup is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
dna-backup is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with dna-backup. If not, see <https://www.gnu.org/licenses/>. */
package utils
import (
"compress/zlib"
"io"
)
// NopCloser returns a WriteCloser with a no-op Close method wrapping
// the provided Writer w.
func NopCloser(w io.Writer) io.WriteCloser {
return nopCloser{w}
}
type nopCloser struct {
io.Writer
}
func (nopCloser) Close() error { return nil }
type ReadWrapper func(r io.Reader) (io.ReadCloser, error)
type WriteWrapper func(w io.Writer) io.WriteCloser
// ZlibReader wraps a reader with a new zlib.Reader.
func ZlibReader(r io.Reader) (io.ReadCloser, error) {
return zlib.NewReader(r)
}
// ZlibWrier wraps a writer with a new zlib.Writer.
func ZlibWriter(w io.Writer) io.WriteCloser {
return zlib.NewWriter(w)
}
func NopReadWrapper(r io.Reader) (io.ReadCloser, error) {
return io.NopCloser(r), nil
}
func NopWriteWrapper(w io.Writer) io.WriteCloser {
return NopCloser(w)
}
type WriteCounter struct {
w io.Writer
count int
}
func NewWriteCounter(writer io.Writer) *WriteCounter {
return &WriteCounter{w: writer}
}
func (c *WriteCounter) Write(p []byte) (n int, err error) {
n, err = c.w.Write(p)
c.count += n
return
}
func (c *WriteCounter) Count() int {
return c.count
}
|
