[#6] Add caching from multiple environments

This commit is contained in:
Nikita Zinkevich 2024-08-14 15:53:53 +03:00
parent aca2229ab7
commit f33aa47969

View file

@ -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,15 @@ 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)
}
if validationBlock == nil {
return nil, fmt.Errorf("validation block does not exist: %w", err)
}
dbPath := path.Join(dir, validationBlock.Hash().String()+".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 +67,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 {