generic.go

package blockstore

import (
	"encoding/hex"
	"fmt"

	bin "github.com/gagliardetto/binary"
	"github.com/linxGnu/grocksdb"
)

func ParseBincode[T any](data []byte) (*T, error) {
	dec := bin.NewBinDecoder(data)
	val := new(T)
	err := dec.Decode(val)
	return val, err
}

func GetBincode[T any](db *grocksdb.DB, cf *grocksdb.ColumnFamilyHandle, key []byte) (*T, error) {
	opts := grocksdb.NewDefaultReadOptions()
	res, err := db.GetCF(opts, cf, key)
	if err != nil {
		return nil, err
	}
	if !res.Exists() {
		return nil, ErrNotFound
	}
	defer res.Free()
	return ParseBincode[T](res.Data())
}

func MultiGetBincode[T any](db *grocksdb.DB, cf *grocksdb.ColumnFamilyHandle, key ...[]byte) ([]*T, error) {
	opts := grocksdb.NewDefaultReadOptions()
	rows, err := db.MultiGetCF(opts, cf, key...)
	if err != nil {
		return nil, err
	}
	defer rows.Destroy()

	vals := make([]*T, len(rows))
	for i, row := range rows {
		val, err := ParseBincode[T](row.Data())
		if err != nil {
			fmt.Printf("cannot decode %s: %s", hex.EncodeToString(key[i]), err)
			return nil, err
		}
		vals[i] = val
	}

	return vals, nil
}