From 2ad98239456f190593c5b79684e991ed4dbed0d4 Mon Sep 17 00:00:00 2001 From: Wilmer Paulino Date: Mon, 23 Apr 2018 12:43:50 -0400 Subject: [PATCH] server: include optional updates when generating a new node announcement --- server.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/server.go b/server.go index d8376ce9..8b04d332 100644 --- a/server.go +++ b/server.go @@ -987,8 +987,8 @@ func (s *server) initTorController() error { // genNodeAnnouncement generates and returns the current fully signed node // announcement. If refresh is true, then the time stamp of the announcement // will be updated in order to ensure it propagates through the network. -func (s *server) genNodeAnnouncement( - refresh bool) (lnwire.NodeAnnouncement, error) { +func (s *server) genNodeAnnouncement(refresh bool, + updates ...func(*lnwire.NodeAnnouncement)) (lnwire.NodeAnnouncement, error) { s.mu.Lock() defer s.mu.Unlock() @@ -997,7 +997,9 @@ func (s *server) genNodeAnnouncement( return *s.currentNodeAnn, nil } - var err error + for _, update := range updates { + update(s.currentNodeAnn) + } newStamp := uint32(time.Now().Unix()) if newStamp <= s.currentNodeAnn.Timestamp {