From 01d0c21fd52c60af863e8505d606b99c41c10eaa Mon Sep 17 00:00:00 2001
From: i-norden <iansnorden@gmail.com>
Date: Tue, 19 Oct 2021 16:19:44 -0500
Subject: [PATCH] make cache configurable

---
 pkg/reader.go | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/pkg/reader.go b/pkg/reader.go
index 6987eb4..eb58aff 100644
--- a/pkg/reader.go
+++ b/pkg/reader.go
@@ -25,6 +25,7 @@ import (
 	"github.com/ethereum/go-ethereum/core/types"
 	"github.com/ethereum/go-ethereum/ethdb"
 	"github.com/ethereum/go-ethereum/params"
+	"github.com/ethereum/go-ethereum/trie"
 )
 
 // LvlDBReader exposes the necessary read functions on lvldb
@@ -34,16 +35,24 @@ type LvlDBReader struct {
 	chainConfig *params.ChainConfig
 }
 
+// ReaderConfig struct for initializing a LvlDBReader
+type ReaderConfig struct {
+	TrieConfig *trie.Config
+	ChainConfig *params.ChainConfig
+	Path, AncientPath string
+	DBCacheSize int
+}
+
 // NewLvlDBReader creates a new LvlDBReader
-func NewLvlDBReader(path, ancient string, chainConfig *params.ChainConfig) (*LvlDBReader, error) {
-	edb, err := rawdb.NewLevelDBDatabaseWithFreezer(path, 1024, 256, ancient, "eth-statediff-service", true)
+func NewLvlDBReader(conf ReaderConfig) (*LvlDBReader, error) {
+	edb, err := rawdb.NewLevelDBDatabaseWithFreezer(conf.Path, conf.DBCacheSize, 256, conf.AncientPath, "eth-statediff-service", true)
 	if err != nil {
 		return nil, err
 	}
 	return &LvlDBReader{
 		ethDB:       edb,
-		stateDB:     state.NewDatabase(edb),
-		chainConfig: chainConfig,
+		stateDB:     state.NewDatabaseWithConfig(edb, conf.TrieConfig),
+		chainConfig: conf.ChainConfig,
 	}, nil
 }