TrueCloudLab/monza
7
1
Fork 5
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity

[#6] Add caching from multiple environments

Browse source
This commit is contained in:
Nikita Zinkevich 2024-08-14 15:53:53 +03:00
parent aca2229ab7
commit f33aa47969
1 changed files with 21 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
internal/chain/chain.go
View file

@ -7,7 +7,6 @@ import (
"fmt"
"os"
"path"
"strconv"
"git.frostfs.info/TrueCloudLab/monza/internal/bytecode"
"github.com/nspcc-dev/neo-go/pkg/core/block"
@ -47,7 +46,15 @@ func Open(ctx context.Context, dir, endpoint string, rewrite bool) (*Chain, erro
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)
}
if validationBlock == nil {
return nil, fmt.Errorf("validation block does not exist: %w", err)
}
dbPath := path.Join(dir, validationBlock.Hash().String()+".db")
if rewrite {
_ = os.Remove(dbPath) // ignore error if file does not exist, etc.
}
@ -60,6 +67,18 @@ func Open(ctx context.Context, dir, endpoint string, rewrite bool) (*Chain, erro
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) {
cached, err := d.block(i)
if err != nil {