mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2024-12-24 21:24:05 -08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Close consumed WAL segments, verify checksums on read

Browse source
This commit is contained in:
Fabian Reinartz 2017-02-14 00:42:06 -08:00
parent eeb03e97e2
commit 84e8027a8e
1 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
wal.go
View file

@ -97,7 +97,7 @@ type walHandler struct {
// ReadAll consumes all entries in the WAL and triggers the registered handlers.
func (w *WAL) ReadAll(h *walHandler) error {
for _, f := range w.files {
for i, f := range w.files {
dec := newWALDecoder(f, h)
for {
@ -109,6 +109,13 @@ func (w *WAL) ReadAll(h *walHandler) error {
return err
}
}
// Close completed file after we are done reading it.
if i < len(w.files)-1 {
if err := f.Close(); err != nil {
return err
}
}
}
return nil
}
@ -524,14 +531,19 @@ func (d *walDecoder) entry() error {
}
buf := d.buf[:length]
if _, err := d.r.Read(buf); err != nil {
cw := crc32.NewIEEE()
tr := io.TeeReader(d.r, cw)
if _, err := tr.Read(buf); err != nil {
return err
}
// Read away checksum.
// TODO(fabxc): verify it
if _, err := d.r.Read(b[:4]); err != nil {
_, err := d.r.Read(b[:4])
if err != nil {
return err
}
if exp, has := binary.BigEndian.Uint32(b[:4]), cw.Sum32(); has != exp {
return errors.Errorf("unexpected CRC32 checksum %x, want %x", has, exp)
}
switch etype {
case WALEntrySeries: