Use badger opts

Author: AlCutter

cmd/experimental/migrate/posix/main.go

@@ -29,6 +29,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/dgraph-io/badger/v4"
 	"github.com/transparency-dev/tessera"
 	"github.com/transparency-dev/tessera/api/layout"
 	"github.com/transparency-dev/tessera/client"
@@ -105,6 +106,10 @@ func main() {
 	if *persistentAntispam {
 		as_opts := tposix_as.AntispamOpts{
 			MaxBatchSize: *antispamBatchSize,
+			BadgerOptions: func(o badger.Options) badger.Options {
+				return o.WithIndexCacheSize(4 << 30)
+
+			},
 		}
 		antispam, err = tposix_as.NewAntispam(ctx, filepath.Join(*storageDir, ".state", "antispam"), as_opts)
 		if err != nil {

cmd/tesseract/posix/main.go

@@ -34,6 +34,7 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/dgraph-io/badger/v4"
 	"github.com/dustin/go-humanize"
 	"github.com/transparency-dev/tessera"
 	tposix "github.com/transparency-dev/tessera/storage/posix"
@@ -97,6 +98,9 @@ var (
 	clientHTTPMaxIdle           = flag.Int("client_http_max_idle", 20, "Maximum number of idle HTTP connections for outgoing requests.")
 	clientHTTPMaxIdlePerHost    = flag.Int("client_http_max_idle_per_host", 10, "Maximum number of idle HTTP connections per host for outgoing requests.")
 	garbageCollectionInterval   = flag.Duration("garbage_collection_interval", 10*time.Second, "Interval between scans to remove obsolete partial tiles and entry bundles. Set to 0 to disable.")
+	antispamBatchSize           = flag.Uint("antispam_batch_size", 20000, "Maximum number of antispam rows to insert per batch update.")
+	antispamBlockCacheSize      = flag.String("antispam_block_cache_size", "2GB", "Amount of RAM to allocate for antispam block cache, set to zero to disable.")
+	antispamIndexCacheSize      = flag.String("antispam_index_cache_size", "2GB", "Amount of RAM to allocate for antispam index cache, set to zero for unlimited.")
 
 	// Infrastructure setup flags
 	storageDir    = flag.String("storage_dir", "", "Path to root of log storage.")
@@ -222,8 +226,22 @@ func newStorage(ctx context.Context, signer note.Signer) (st *storage.CTStorage,
 	if err != nil {
 		return nil, fmt.Errorf("failed to initialize POSIX Tessera storage driver: %v", err)
 	}
+	antispamIndexCacheBytes, error := humanize.ParseBytes(*antispamIndexCacheSize)
+	if error != nil {
+		return nil, fmt.Errorf("invalid antispam index cache size: %v", error)
+	}
+	antispamBlockCacheBytes, error := humanize.ParseBytes(*antispamBlockCacheSize)
+	if error != nil {
+		return nil, fmt.Errorf("invalid antispam block cache size: %v", error)
+	}
 	asOpts := tposix_as.AntispamOpts{
 		PushbackThreshold: *pushbackMaxAntispamLag,
+		MaxBatchSize:      *antispamBatchSize,
+		BadgerOptions: func(o badger.Options) badger.Options {
+			return o.
+				WithIndexCacheSize(int64(antispamIndexCacheBytes)).
+				WithBlockCacheSize(int64(antispamBlockCacheBytes))
+		},
 	}
 	antispam, err := tposix_as.NewAntispam(ctx, filepath.Join(*storageDir, ".state", "antispam"), asOpts)
 	if err != nil {

go.mod

@@ -19,6 +19,7 @@ require (
 	github.com/cenkalti/backoff/v5 v5.0.3
 	github.com/charmbracelet/bubbletea v1.3.10
 	github.com/charmbracelet/x/ansi v0.11.6
+	github.com/dgraph-io/badger/v4 v4.9.1
 	github.com/dustin/go-humanize v1.0.1
 	github.com/fsouza/fake-gcs-server v1.54.0
 	github.com/gdamore/tcell/v2 v2.13.8
@@ -90,7 +91,6 @@ require (
 	github.com/clipperhouse/stringish v0.1.1 // indirect
 	github.com/clipperhouse/uax29/v2 v2.5.0 // indirect
 	github.com/cncf/xds/go v0.0.0-20251210132809-ee656c7534f5 // indirect
-	github.com/dgraph-io/badger/v4 v4.9.1 // indirect
 	github.com/dgraph-io/ristretto/v2 v2.2.0 // indirect
 	github.com/envoyproxy/go-control-plane/envoy v1.36.0 // indirect
 	github.com/envoyproxy/protoc-gen-validate v1.3.0 // indirect