chore: update keydb dependency

- Update keydb dependency from v0.1.0-alpha to v0.2.0-alpha
- Refactor dedup service initialization in test to use new constructor
- Remove unused keydb cache import and factory function
- Update mockedCloudStorage to implement new Download method signature

Author: mihir20

go.mod

@@ -87,7 +87,7 @@ require (
 	github.com/rudderlabs/analytics-go v3.3.3+incompatible
 	github.com/rudderlabs/bing-ads-go-sdk v0.2.3
 	github.com/rudderlabs/compose-test v0.1.3
-	github.com/rudderlabs/keydb v0.1.0-alpha
+	github.com/rudderlabs/keydb v0.2.0-alpha
 	github.com/rudderlabs/rudder-go-kit v0.59.4
 	github.com/rudderlabs/rudder-observability-kit v0.0.4
 	github.com/rudderlabs/rudder-schemas v0.7.0

go.sum

@@ -1201,6 +1201,8 @@ github.com/rudderlabs/goqu/v10 v10.3.1 h1:rnfX+b4EwBWQ2UQfIGeEW299JBBkK5biEbnf7K
 github.com/rudderlabs/goqu/v10 v10.3.1/go.mod h1:LH2vI5gGHBxEQuESqFyk5ZA2anGINc8o25hbidDWOYw=
 github.com/rudderlabs/keydb v0.1.0-alpha h1:h87QGW9ut6jQExQjtZsp2BxPc0Hix58WELQ5OrzialM=
 github.com/rudderlabs/keydb v0.1.0-alpha/go.mod h1:gnyirkn1FWIQPCvo/E6rFmfDQ2pjVkAYHF3hJvsBpC0=
+github.com/rudderlabs/keydb v0.2.0-alpha h1:LcjiKWewZIO6OLeUvWwte8Qe3VeD1+/5jhzpHM7ZzVk=
+github.com/rudderlabs/keydb v0.2.0-alpha/go.mod h1:7a49ZEx7gtZJYP9/sZoYQk2F3GlFaSoX3oubkyn+Hu8=
 github.com/rudderlabs/parquet-go v0.0.3 h1:/zgRj929pGKHsthc0kw8stVEcFu1JUcpxDRlhxjSLic=
 github.com/rudderlabs/parquet-go v0.0.3/go.mod h1:WmwBOdvwpXl2aZGRk3NxxgzC/DaWGfax3jrCRhKhtSo=
 github.com/rudderlabs/rudder-go-kit v0.59.4 h1:iKtxLhRLvrfwck53l/2ZM1bnykuJ41l1ohreUpR9l6E=

services/dedup/dedup_test.go

@@ -17,7 +17,6 @@ import (
 	"golang.org/x/sync/errgroup"
 	"google.golang.org/grpc"
 
-	keydbcache "github.com/rudderlabs/keydb/cache"
 	keydbclient "github.com/rudderlabs/keydb/client"
 	keydb "github.com/rudderlabs/keydb/node"
 	keydbproto "github.com/rudderlabs/keydb/proto"
@@ -395,12 +394,8 @@ func startKeydb(t testing.TB, conf *config.Config) {
 		SnapshotInterval: time.Minute,
 		Addresses:        []string{address},
 	}
-	cf := func(hashRange uint32) (keydb.Cache, error) {
-		conf.Set("BadgerDB.Dedup.Path", t.TempDir())
-		return keydbcache.BadgerFactory(conf, logger.NOP)(hashRange)
-	}
 	require.NoError(t, err)
-	service, err = keydb.NewService(ctx, nodeConfig, cf, &mockedCloudStorage{}, logger.NOP)
+	service, err = keydb.NewService(ctx, nodeConfig, &mockedCloudStorage{}, conf, stats.NOP, logger.NOP)
 	require.NoError(t, err)
 
 	// Create a gRPC server
@@ -444,7 +439,9 @@ func (m *mockedFilemanagerSession) Next() (fileObjects []*filemanager.FileInfo,
 
 type mockedCloudStorage struct{}
 
-func (m *mockedCloudStorage) Download(_ context.Context, _ io.WriterAt, _ string) error { return nil }
+func (m *mockedCloudStorage) Download(_ context.Context, _ io.WriterAt, _ string, options ...filemanager.DownloadOption) error {
+	return nil
+}
 func (m *mockedCloudStorage) UploadReader(_ context.Context, _ string, _ io.Reader) (filemanager.UploadedFile, error) {
 	return filemanager.UploadedFile{}, nil
 }