You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.3 KiB
65 lines
1.3 KiB
/*
|
|
cutWALUntil is a small utility for cutting a WAL until the given height
|
|
(inclusively). Note it does not include last cs.EndHeightMessage.
|
|
|
|
Usage:
|
|
cutWALUntil <path-to-wal> height-to-stop <output-wal>
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
|
|
cs "github.com/tendermint/tendermint/consensus"
|
|
)
|
|
|
|
func main() {
|
|
if len(os.Args) < 4 {
|
|
fmt.Println("3 arguments required: <path-to-wal> <height-to-stop> <output-wal>")
|
|
os.Exit(1)
|
|
}
|
|
|
|
var heightToStop uint64
|
|
var err error
|
|
if heightToStop, err = strconv.ParseUint(os.Args[2], 10, 64); err != nil {
|
|
panic(fmt.Errorf("failed to parse height: %v", err))
|
|
}
|
|
|
|
in, err := os.Open(os.Args[1])
|
|
if err != nil {
|
|
panic(fmt.Errorf("failed to open input WAL file: %v", err))
|
|
}
|
|
defer in.Close()
|
|
|
|
out, err := os.Create(os.Args[3])
|
|
if err != nil {
|
|
panic(fmt.Errorf("failed to open output WAL file: %v", err))
|
|
}
|
|
defer out.Close()
|
|
|
|
enc := cs.NewWALEncoder(out)
|
|
dec := cs.NewWALDecoder(in)
|
|
|
|
for {
|
|
msg, err := dec.Decode()
|
|
if err == io.EOF {
|
|
break
|
|
} else if err != nil {
|
|
panic(fmt.Errorf("failed to decode msg: %v", err))
|
|
}
|
|
|
|
if m, ok := msg.Msg.(cs.EndHeightMessage); ok {
|
|
if m.Height == heightToStop {
|
|
break
|
|
}
|
|
}
|
|
|
|
err = enc.Encode(msg)
|
|
if err != nil {
|
|
panic(fmt.Errorf("failed to encode msg: %v", err))
|
|
}
|
|
}
|
|
}
|