erigon-pulse/cl/persistence/raw_block_saver.go

package persistence

import (
	"context"
	"io"
	"os"
	"sync"

	libcommon "github.com/ledgerwatch/erigon-lib/common"
	"github.com/ledgerwatch/erigon/cl/clparams"
	"github.com/spf13/afero"
	"go.uber.org/zap/buffer"
)

var bPool = sync.Pool{
	New: func() interface{} {
		return &buffer.Buffer{}
	},
}

type aferoRawBeaconBlockChain struct {
	fs  afero.Fs
	cfg *clparams.BeaconChainConfig
}

func NewAferoRawBlockSaver(fs afero.Fs, cfg *clparams.BeaconChainConfig) RawBeaconBlockChain {
	return aferoRawBeaconBlockChain{
		fs:  fs,
		cfg: cfg,
	}
}

func AferoRawBeaconBlockChainFromOsPath(cfg *clparams.BeaconChainConfig, path string) (RawBeaconBlockChain, afero.Fs) {
	dataDirFs := afero.NewBasePathFs(afero.NewOsFs(), path)
	return NewAferoRawBlockSaver(dataDirFs, cfg), dataDirFs
}

func (a aferoRawBeaconBlockChain) BlockWriter(ctx context.Context, slot uint64, blockRoot libcommon.Hash) (io.WriteCloser, error) {
	folderPath, path := rootToPaths(slot, blockRoot, a.cfg)
	_ = a.fs.MkdirAll(folderPath, 0o755)
	return a.fs.OpenFile(path, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0o755)
}

func (a aferoRawBeaconBlockChain) BlockReader(ctx context.Context, slot uint64, blockRoot libcommon.Hash) (io.ReadCloser, error) {
	_, path := rootToPaths(slot, blockRoot, a.cfg)
	return a.fs.OpenFile(path, os.O_RDONLY, 0o755)
}

func (a aferoRawBeaconBlockChain) DeleteBlock(ctx context.Context, slot uint64, blockRoot libcommon.Hash) error {
	_, path := rootToPaths(slot, blockRoot, a.cfg)
	return a.fs.Remove(path)
}