tool: dump tool supports index

This commit is contained in:
Xiang Li 2015-02-17 21:01:06 -08:00
parent 7c65857283
commit d6f8a30f7c

View File

@ -32,17 +32,28 @@ import (
func main() { func main() {
from := flag.String("data-dir", "", "") from := flag.String("data-dir", "", "")
snapfile := flag.String("snap-file", "", "The base name of snapshot file to read") snapfile := flag.String("start-snap", "", "The base name of snapshot file to start dumping")
index := flag.Uint64("start-index", 0, "The index to start dumping")
flag.Parse() flag.Parse()
if *from == "" { if *from == "" {
log.Fatal("Must provide -data-dir flag") log.Fatal("Must provide -data-dir flag.")
}
if *snapfile != "" && *index != 0 {
log.Fatal("start-snap and start-index flags cannot be used together.")
} }
var ( var (
walsnap walpb.Snapshot
snapshot *raftpb.Snapshot snapshot *raftpb.Snapshot
err error err error
) )
isIndex := *index != 0
if isIndex {
fmt.Printf("Start dumping log entries from index %d.\n", *index)
walsnap.Index = *index
} else {
if *snapfile == "" { if *snapfile == "" {
ss := snap.New(snapDir(*from)) ss := snap.New(snapDir(*from))
snapshot, err = ss.Load() snapshot, err = ss.Load()
@ -50,7 +61,6 @@ func main() {
snapshot, err = snap.Read(path.Join(snapDir(*from), *snapfile)) snapshot, err = snap.Read(path.Join(snapDir(*from), *snapfile))
} }
var walsnap walpb.Snapshot
switch err { switch err {
case nil: case nil:
walsnap.Index, walsnap.Term = snapshot.Metadata.Index, snapshot.Metadata.Term walsnap.Index, walsnap.Term = snapshot.Metadata.Index, snapshot.Metadata.Term
@ -62,6 +72,8 @@ func main() {
default: default:
log.Fatalf("Failed loading snapshot: %v", err) log.Fatalf("Failed loading snapshot: %v", err)
} }
fmt.Println("Start dupmping log entries from snapshot.")
}
w, err := wal.Open(walDir(*from), walsnap) w, err := wal.Open(walDir(*from), walsnap)
if err != nil { if err != nil {
@ -69,7 +81,7 @@ func main() {
} }
wmetadata, state, ents, err := w.ReadAll() wmetadata, state, ents, err := w.ReadAll()
w.Close() w.Close()
if err != nil { if err != nil && (!isIndex || err != wal.ErrSnapshotNotFound) {
log.Fatalf("Failed reading WAL: %v", err) log.Fatalf("Failed reading WAL: %v", err)
} }
id, cid := parseWALMetadata(wmetadata) id, cid := parseWALMetadata(wmetadata)