Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

wal: backup broken wal before repairing

Browse Source
This commit is contained in:
Xiang Li 2015-03-29 15:42:59 -07:00
parent 1231f82f22
commit 684ebd95ae
2 changed files with 34 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
wal/repair.go
View File

@ -16,6 +16,7 @@ package wal
import (
"io"
"log"
"os"
"path"
@ -30,6 +31,7 @@ func Repair(dirpath string) bool {
if err != nil {
return false
}
defer f.Close()
n := 0
rec := &walpb.Record{}
@ -45,12 +47,30 @@ func Repair(dirpath string) bool {
case io.EOF:
return true
case io.ErrUnexpectedEOF:
err = f.Truncate(int64(n))
if err != nil {
log.Printf("wal: repairing %v", f.Name())
bf, bferr := os.Create(f.Name() + ".broken")
if bferr != nil {
log.Printf("wal: could not repair %v, failed to create backup file", f.Name())
return false
}
err = f.Sync()
if err != nil {
defer bf.Close()
if _, err = f.Seek(0, os.SEEK_SET); err != nil {
log.Printf("wal: could not repair %v, failed to read file", f.Name())
return false
}
if _, err = io.Copy(bf, f); err != nil {
log.Printf("wal: could not repair %v, failed to copy file", f.Name())
return false
}
if err = f.Truncate(int64(n)); err != nil {
log.Printf("wal: could not repair %v, failed to truncate file", f.Name())
return false
}
if err = f.Sync(); err != nil {
log.Printf("wal: could not repair %v, failed to sync file", f.Name())
return false
}
return true

11
wal/util.go
View File

@ -15,12 +15,18 @@
package wal
import (
"errors"
"fmt"
"log"
"strings"
"github.com/coreos/etcd/pkg/fileutil"
)
var (
badWalName = errors.New("bad wal name")
)
func Exist(dirpath string) bool {
names, err := fileutil.ReadDir(dirpath)
if err != nil {
@ -76,8 +82,11 @@ func checkWalNames(names []string) []string {
}
func parseWalName(str string) (seq, index uint64, err error) {
if !strings.HasSuffix(str, ".wal") {
return 0, 0, badWalName
}
_, err = fmt.Sscanf(str, "%016x-%016x.wal", &seq, &index)
return
return seq, index, err
}
func walName(seq, index uint64) string {