TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

cli: allow to use incremental dumps in db restore

Browse source 
Signed-off-by: Evgeniy Stratonikov <evgeniy@nspcc.ru>
This commit is contained in:
Evgeniy Stratonikov 2021-07-12 17:59:28 +03:00
parent 05cb6586a6
commit df5ee4abdc
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cli/server/server.go
View file

@ -64,6 +64,10 @@ func NewCommands() []cli.Command {
Name: "dump", Name: "dump",
Usage: "directory for storing JSON dumps", Usage: "directory for storing JSON dumps",
}, },
cli.BoolFlag{
Name: "incremental, n",
Usage: "use if dump is incremental",
},
) )
return []cli.Command{ return []cli.Command{
{ {
@ -240,6 +244,16 @@ func restoreDB(ctx *cli.Context) error {
defer prometheus.ShutDown() defer prometheus.ShutDown()
defer pprof.ShutDown() defer pprof.ShutDown()
var start uint32
if ctx.Bool("incremental") {
start = reader.ReadU32LE()
if chain.BlockHeight()+1 < start {
return cli.NewExitError(fmt.Errorf("expected height: %d, dump starts at %d",
chain.BlockHeight()+1, start), 1)
}
skip = chain.BlockHeight() + 1 - start
}
var allBlocks = reader.ReadU32LE() var allBlocks = reader.ReadU32LE()
if reader.Err != nil { if reader.Err != nil {
return cli.NewExitError(err, 1) return cli.NewExitError(err, 1)