-
Notifications
You must be signed in to change notification settings - Fork 0
/
reader.go
65 lines (53 loc) · 981 Bytes
/
reader.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
package main
import (
"os"
"fmt"
"io"
"strings"
"bufio"
)
var _ = fmt.Println
const Separator = " "
type Reader struct {
*bufio.Reader
}
func NewReader(r io.Reader) *Reader {
return &Reader{bufio.NewReader(r)}
}
func (r *Reader) ReadEntry() (entry *Entry, err error) {
bytes, _, err := r.ReadLine()
if err != nil {
return
}
line := string(bytes)
sep := strings.Index(line, Separator)
if sep == -1 {
return
}
entry = &Entry{Checksum: line[0:sep], Filename: line[sep+len(Separator):]}
return
}
func (r *Reader) Each(f func(*Entry)) (err error) {
for {
if e, err := r.ReadEntry(); err == io.EOF {
break
} else if err != nil {
return err
} else if e != nil {
f(e)
}
}
return
}
func Load(filename string) (m map[string]string, err error) {
file, err := os.Open(filename)
if err != nil {
return
}
m = make(map[string]string)
r := NewReader(file)
err = r.Each(func(e *Entry) {
m[e.Filename] = e.Checksum
})
return m, err
}