Add WithDatastore option.

libp2p · Feb 6, 2021 · 75e9366 · 75e9366
1 parent fe35a3c
commit 75e9366
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pubsub.go b/pubsub.go
@@ -589,3 +589,11 @@ func WithRebroadcastInitialDelay(duration time.Duration) Option {
 		return nil
 	}
 }
+
+// WithDatastore returns an option that overrides the default datastore.
+func WithDatastore(datastore ds.Datastore) Option {
+	return func(store *PubsubValueStore) error {
+		store.ds = datastore
+		return nil
+	}
+}