discovery: update UpdatesInHorizon

Author: ellemouton

discovery/chan_series.go

@@ -30,8 +30,9 @@ type ChannelGraphTimeSeries interface {
 	// update timestamp between the start time and end time. We'll use this
 	// to catch up a remote node to the set of channel updates that they
 	// may have missed out on within the target chain.
-	UpdatesInHorizon(chain chainhash.Hash,
-		startTime time.Time, endTime time.Time) ([]lnwire.Message, error)
+	UpdatesInHorizon(chain chainhash.Hash, startTime time.Time,
+		endTime time.Time, startBlock,
+		endBlock uint32) ([]lnwire.Message, error)
 
 	// FilterKnownChanIDs takes a target chain, and a set of channel ID's,
 	// and returns a filtered set of chan ID's. This filtered set of chan
@@ -103,15 +104,16 @@ func (c *ChanSeries) HighestChanID(chain chainhash.Hash) (*lnwire.ShortChannelID
 // within the target chain.
 //
 // NOTE: This is part of the ChannelGraphTimeSeries interface.
-func (c *ChanSeries) UpdatesInHorizon(chain chainhash.Hash,
-	startTime time.Time, endTime time.Time) ([]lnwire.Message, error) {
+func (c *ChanSeries) UpdatesInHorizon(chain chainhash.Hash, startTime time.Time,
+	endTime time.Time, startBlock, endBlock uint32) ([]lnwire.Message,
+	error) {
 
 	var updates []lnwire.Message
 
 	// First, we'll query for all the set of channels that have an update
 	// that falls within the specified horizon.
 	chansInHorizon, err := c.graph.ChanUpdatesInHorizon(
-		startTime, endTime, 0, 0,
+		startTime, endTime, startBlock, endBlock,
 	)
 	if err != nil {
 		return nil, err

discovery/syncer.go

@@ -1194,6 +1194,9 @@ func (g *GossipSyncer) ApplyGossipFilter(filter *lnwire.GossipTimestampRange) er
 		time.Duration(g.remoteUpdateHorizon.TimestampRange) * time.Second,
 	)
 
+	startBlock := g.remoteUpdateHorizon.FirstBlockHeight.UnwrapOr(0)
+	endBlock := g.remoteUpdateHorizon.BlockRange.UnwrapOr(0)
+
 	g.Unlock()
 
 	// If requested, don't reply with historical gossip data when the remote
@@ -1205,7 +1208,7 @@ func (g *GossipSyncer) ApplyGossipFilter(filter *lnwire.GossipTimestampRange) er
 	// Now that the remote peer has applied their filter, we'll query the
 	// database for all the messages that are beyond this filter.
 	newUpdatestoSend, err := g.cfg.channelSeries.UpdatesInHorizon(
-		g.cfg.chainHash, startTime, endTime,
+		g.cfg.chainHash, startTime, endTime, startBlock, endBlock,
 	)
 	if err != nil {
 		return err

discovery/syncer_test.go

@@ -28,10 +28,13 @@ var (
 )
 
 type horizonQuery struct {
-	chain chainhash.Hash
-	start time.Time
-	end   time.Time
+	chain      chainhash.Hash
+	start      time.Time
+	end        time.Time
+	startBlock uint32
+	endBlock   uint32
 }
+
 type filterRangeReq struct {
 	startHeight, endHeight uint32
 }
@@ -82,10 +85,11 @@ func (m *mockChannelGraphTimeSeries) HighestChanID(chain chainhash.Hash) (*lnwir
 	return &m.highestID, nil
 }
 func (m *mockChannelGraphTimeSeries) UpdatesInHorizon(chain chainhash.Hash,
-	startTime time.Time, endTime time.Time) ([]lnwire.Message, error) {
+	startTime time.Time, endTime time.Time, startBlock, endBlock uint32) (
+	[]lnwire.Message, error) {
 
 	m.horizonReq <- horizonQuery{
-		chain, startTime, endTime,
+		chain, startTime, endTime, startBlock, endBlock,
 	}
 
 	return <-m.horizonResp, nil