diff --git a/observer/channel_peer.go b/observer/channel_peer.go index 0fdb553..28aee14 100644 --- a/observer/channel_peer.go +++ b/observer/channel_peer.go @@ -42,11 +42,13 @@ type ( } PeerChannelsFetcher interface { + Uri() string api.ChannelListGetter api.ChainInfoGetter } PeerChannels interface { + Uri() string Channels() map[string]*ChannelInfo } @@ -135,6 +137,10 @@ func (cp *ChannelPeer) Observe(ctx context.Context) { }() } +func (cp *ChannelPeer) Uri() string { + return cp.channelFetcher.Uri() +} + func (cp *ChannelPeer) Channels() map[string]*ChannelInfo { cp.mu.Lock() defer cp.mu.Unlock() diff --git a/observer/channel_peer_fetcher_mock.go b/observer/channel_peer_fetcher_mock.go index 2dc28a6..ba7c1a1 100644 --- a/observer/channel_peer_fetcher_mock.go +++ b/observer/channel_peer_fetcher_mock.go @@ -18,6 +18,10 @@ func NewChannelPeerFetcherMock(channels map[string]uint64) *ChannelPeerFetcherMo } } +func (c *ChannelPeerFetcherMock) Uri() string { + return "mock" +} + func (c *ChannelPeerFetcherMock) GetChannels(context.Context) (*peer.ChannelQueryResponse, error) { var channels []*peer.ChannelInfo for channelName := range c.channels { diff --git a/observer/channel_peer_mock.go b/observer/channel_peer_mock.go index a641fa5..db86bb8 100644 --- a/observer/channel_peer_mock.go +++ b/observer/channel_peer_mock.go @@ -9,6 +9,10 @@ type ChannelPeerMock struct { channelsInfo map[string]*ChannelInfo } +func (m *ChannelPeerMock) Uri() string { + return "mock" +} + func NewChannelPeerMock(channelsInfo ...*ChannelInfo) *ChannelPeerMock { channels := make(map[string]*ChannelInfo, len(channelsInfo)) for _, channelInfo := range channelsInfo {