forked from TrueCloudLab/monza
[#6] Add caching from multiple environments
This commit is contained in:
parent
aca2229ab7
commit
ce6489146f
1 changed files with 18 additions and 2 deletions
|
@ -7,7 +7,6 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"git.frostfs.info/TrueCloudLab/monza/internal/bytecode"
|
"git.frostfs.info/TrueCloudLab/monza/internal/bytecode"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
||||||
|
@ -47,7 +46,12 @@ func Open(ctx context.Context, dir, endpoint string, rewrite bool) (*Chain, erro
|
||||||
return nil, fmt.Errorf("rpc get version: %w", err)
|
return nil, fmt.Errorf("rpc get version: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
dbPath := path.Join(dir, strconv.Itoa(int(v.Protocol.Network))+".db")
|
validationBlock, err := getValidationBlock(cli)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error get validation block: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
dbPath := path.Join(dir, validationBlock.Hash().StringLE()+".db")
|
||||||
if rewrite {
|
if rewrite {
|
||||||
_ = os.Remove(dbPath) // ignore error if file does not exist, etc.
|
_ = os.Remove(dbPath) // ignore error if file does not exist, etc.
|
||||||
}
|
}
|
||||||
|
@ -60,6 +64,18 @@ func Open(ctx context.Context, dir, endpoint string, rewrite bool) (*Chain, erro
|
||||||
return &Chain{db, v.Protocol.StateRootInHeader, cli}, nil
|
return &Chain{db, v.Protocol.StateRootInHeader, cli}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getValidationBlock(cli *rpcclient.Client) (*block.Block, error) {
|
||||||
|
// not 0, because it always has the same hash
|
||||||
|
const validationBlockIndex = 1
|
||||||
|
var err error
|
||||||
|
validBlock, err := cli.GetBlockByIndex(validationBlockIndex)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("error sending request for block: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return validBlock, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (d *Chain) Block(i uint32) (res *block.Block, err error) {
|
func (d *Chain) Block(i uint32) (res *block.Block, err error) {
|
||||||
cached, err := d.block(i)
|
cached, err := d.block(i)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
Loading…
Reference in a new issue