-
Notifications
You must be signed in to change notification settings - Fork 0
/
writer.go
79 lines (60 loc) · 1.32 KB
/
writer.go
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
76
77
78
79
package standort
import (
"bufio"
"crypto/sha256"
"hash"
"io"
"github.com/friedenberg/zit/src/alfa/errors"
"github.com/friedenberg/zit/src/alfa/schnittstellen"
"github.com/friedenberg/zit/src/charlie/sha"
)
type Writer interface {
sha.WriteCloser
}
type writer struct {
hash hash.Hash
tee io.Writer
wCompress, wAge io.WriteCloser
wBuf *bufio.Writer
}
func NewWriter(o WriteOptions) (w *writer, err error) {
w = &writer{}
w.wBuf = bufio.NewWriter(o.Writer)
if w.wAge, err = o.Age.Encrypt(w.wBuf); err != nil {
err = errors.Wrap(err)
return
}
w.hash = sha256.New()
w.wCompress = o.CompressionType.NewWriter(w.wAge)
w.tee = io.MultiWriter(w.hash, w.wCompress)
return
}
func (w *writer) ReadFrom(r io.Reader) (n int64, err error) {
if n, err = io.Copy(w.tee, r); err != nil {
err = errors.Wrap(err)
return
}
return
}
func (w *writer) Write(p []byte) (n int, err error) {
return w.tee.Write(p)
}
func (w *writer) Close() (err error) {
if err = w.wCompress.Close(); err != nil {
err = errors.Wrap(err)
return
}
if err = w.wAge.Close(); err != nil {
err = errors.Wrap(err)
return
}
if err = w.wBuf.Flush(); err != nil {
err = errors.Wrap(err)
return
}
return
}
func (w *writer) GetShaLike() (s schnittstellen.ShaLike) {
s = sha.FromHash(w.hash)
return
}