Add pubsub support to drand beacon

Signed-off-by: Jakub Sztandera <kubuxu@protocol.ai>
2020-05-29 17:55:17 +02:00
parent f06794ae84
commit 71eb41b9aa
6 changed files with 42 additions and 28 deletions
@@ -123,12 +123,19 @@ func RetrievalResolver(l *discovery.Local) retrievalmarket.PeerResolver {
 	return discovery.Multi(l)
 }

-func RandomBeacon(cs *store.ChainStore, _ dtypes.AfterGenesisSet) (beacon.RandomBeacon, error) {
-	gen, err := cs.GetGenesis()
+type RandomBeaconParams struct {
+	fx.In
+
+	PubSub *pubsub.PubSub `optional:"true"`
+	Cs     *store.ChainStore
+}
+
+func RandomBeacon(p RandomBeaconParams, _ dtypes.AfterGenesisSet) (beacon.RandomBeacon, error) {
+	gen, err := p.Cs.GetGenesis()
 	if err != nil {
 		return nil, err
 	}

 	//return beacon.NewMockBeacon(build.BlockDelay * time.Second)
-	return drand.NewDrandBeacon(gen.Timestamp, build.BlockDelay)
+	return drand.NewDrandBeacon(gen.Timestamp, build.BlockDelay, p.PubSub)
 }