vitalif
/
etcd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

snap: refactoring

release-2.0
Xiang Li 2014-08-20 13:42:23 -07:00 committed by Yicheng Qin
parent f664c02a40
commit f891199ab0
1 changed files with 37 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
snap/snapshotter.go
View File

@ -56,37 +56,47 @@ func (s *Snapshotter) Load() (*raft.Snapshot, error) {
if err != nil {
return nil, err
}
var snap raft.Snapshot
var serializedSnap Snapshot
var b []byte
var snap *raft.Snapshot
for _, name := range names {
fpath := path.Join(s.dir, name)
b, err = ioutil.ReadFile(fpath)
if err != nil {
log.Printf("Snapshotter cannot read file %v: %v", name, err)
renameBroken(fpath)
continue
if snap, err = loadSnap(s.dir, name); err == nil {
break
}
if err = serializedSnap.Unmarshal(b); err != nil {
log.Printf("Corrupted snapshot file %v: %v", name, err)
renameBroken(fpath)
continue
}
crc := crc32.Update(0, crcTable, serializedSnap.Data)
if crc != serializedSnap.Crc {
log.Printf("Corrupted snapshot file %v: crc mismatch", name)
renameBroken(fpath)
err = ErrCRCMismatch
continue
}
if err = json.Unmarshal(serializedSnap.Data, &snap); err != nil {
log.Printf("Corrupted snapshot file %v: %v", name, err)
renameBroken(fpath)
continue
}
break
}
return snap, err
}
func loadSnap(dir, name string) (*raft.Snapshot, error) {
var err error
var b []byte
fpath := path.Join(dir, name)
defer func() {
if err != nil {
renameBroken(fpath)
}
}()
b, err = ioutil.ReadFile(fpath)
if err != nil {
log.Printf("Snapshotter cannot read file %v: %v", name, err)
return nil, err
}
var serializedSnap Snapshot
if err = serializedSnap.Unmarshal(b); err != nil {
log.Printf("Corrupted snapshot file %v: %v", name, err)
return nil, err
}
crc := crc32.Update(0, crcTable, serializedSnap.Data)
if crc != serializedSnap.Crc {
log.Printf("Corrupted snapshot file %v: crc mismatch", name)
err = ErrCRCMismatch
return nil, err
}
var snap raft.Snapshot
if err = json.Unmarshal(serializedSnap.Data, &snap); err != nil {
log.Printf("Corrupted snapshot file %v: %v", name, err)
return nil, err
}
return &snap, nil