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,7 @@ 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.")
 
 	// Infrastructure setup flags
 	storageDir    = flag.String("storage_dir", "", "Path to root of log storage.")
@@ -224,6 +226,10 @@ func newStorage(ctx context.Context, signer note.Signer) (st *storage.CTStorage,
 	}
 	asOpts := tposix_as.AntispamOpts{
 		PushbackThreshold: *pushbackMaxAntispamLag,
+		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"), 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