From 7265a5e42b67e12621e8913f552c1b6d3b4f21ae Mon Sep 17 00:00:00 2001
From: Joost Jager <joost.jager@gmail.com>
Date: Wed, 13 Apr 2022 10:19:25 +0200
Subject: [PATCH] beacon: add constructor

---
 server.go         | 5 +----
 witness_beacon.go | 7 +++++++
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/server.go b/server.go
index 1254c62d9..1ccd9e5cd 100644
--- a/server.go
+++ b/server.go
@@ -595,10 +595,7 @@ func newServer(cfg *Config, listenAddrs []net.Addr,
 		quit:       make(chan struct{}),
 	}
 
-	s.witnessBeacon = &preimageBeacon{
-		wCache:      dbs.ChanStateDB.NewWitnessCache(),
-		subscribers: make(map[uint64]*preimageSubscriber),
-	}
+	s.witnessBeacon = newPreimageBeacon(dbs.ChanStateDB.NewWitnessCache())
 
 	currentHash, currentHeight, err := s.cc.ChainIO.GetBestBlock()
 	if err != nil {
diff --git a/witness_beacon.go b/witness_beacon.go
index 1593bee76..b3ec95eb5 100644
--- a/witness_beacon.go
+++ b/witness_beacon.go
@@ -28,6 +28,13 @@ type preimageBeacon struct {
 	subscribers   map[uint64]*preimageSubscriber
 }
 
+func newPreimageBeacon(wCache *channeldb.WitnessCache) *preimageBeacon {
+	return &preimageBeacon{
+		wCache:      wCache,
+		subscribers: make(map[uint64]*preimageSubscriber),
+	}
+}
+
 // SubscribeUpdates returns a channel that will be sent upon *each* time a new
 // preimage is discovered.
 func (p *preimageBeacon) SubscribeUpdates() *contractcourt.WitnessSubscription {