diff --git a/multinode/adaptor.go b/multinode/adaptor.go index a87f759..a9c3e42 100644 --- a/multinode/adaptor.go +++ b/multinode/adaptor.go @@ -80,7 +80,7 @@ func (m *Adapter[RPC, HEAD]) RegisterSub(sub *ManagedSubscription, stopInFLightC return nil } -func (m *Adapter[RPC, HEAD]) removeSub(sub Subscription) { +func (m *Adapter[RPC, HEAD]) RemoveSub(sub Subscription) { m.subsSliceMu.Lock() defer m.subsSliceMu.Unlock() delete(m.subs, sub) @@ -143,7 +143,7 @@ func (m *Adapter[RPC, HEAD]) SubscribeToHeads(ctx context.Context) (<-chan HEAD, sub := &ManagedSubscription{ Subscription: &poller, - onUnsubscribe: m.removeSub, + onUnsubscribe: m.RemoveSub, } err := m.RegisterSub(sub, chStopInFlight) @@ -176,7 +176,7 @@ func (m *Adapter[RPC, HEAD]) SubscribeToFinalizedHeads(ctx context.Context) (<-c sub := &ManagedSubscription{ Subscription: &poller, - onUnsubscribe: m.removeSub, + onUnsubscribe: m.RemoveSub, } err := m.RegisterSub(sub, chStopInFlight) diff --git a/multinode/adaptor_test.go b/multinode/adaptor_test.go index c4c6b3a..1265f02 100644 --- a/multinode/adaptor_test.go +++ b/multinode/adaptor_test.go @@ -136,7 +136,7 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) { mockSub := newMockSub() sub := &ManagedSubscription{ Subscription: mockSub, - onUnsubscribe: c.removeSub, + onUnsubscribe: c.RemoveSub, } err := c.RegisterSub(sub, make(chan struct{})) require.NoError(t, err) @@ -151,7 +151,7 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) { mockSub := newMockSub() sub := &ManagedSubscription{ Subscription: mockSub, - onUnsubscribe: c.removeSub, + onUnsubscribe: c.RemoveSub, } err := c.RegisterSub(sub, chStopInFlight) require.Error(t, err) @@ -164,12 +164,12 @@ func TestMultiNodeClient_RegisterSubs(t *testing.T) { mockSub1 := newMockSub() sub1 := &ManagedSubscription{ Subscription: mockSub1, - onUnsubscribe: c.removeSub, + onUnsubscribe: c.RemoveSub, } mockSub2 := newMockSub() sub2 := &ManagedSubscription{ Subscription: mockSub2, - onUnsubscribe: c.removeSub, + onUnsubscribe: c.RemoveSub, } err := c.RegisterSub(sub1, chStopInFlight) require.NoError(t, err)