@@ -67,24 +67,24 @@ func (ps *peerState) IsZero() bool {
6767 return ps .partsMetadata == nil && ps .sentPartsMetadata == nil
6868}
6969
70- type partialMessageStatePerTopicGroup struct {
70+ type partialMessageStatePerGroupPerTopic struct {
7171 peerState map [peer.ID ]* peerState
7272 groupTTL int
7373 initiatedBy peer.ID // zero value if we initiated the group
7474}
7575
76- func newPartialMessageStatePerTopicGroup (groupTTL int ) * partialMessageStatePerTopicGroup {
77- return & partialMessageStatePerTopicGroup {
76+ func newPartialMessageStatePerTopicGroup (groupTTL int ) * partialMessageStatePerGroupPerTopic {
77+ return & partialMessageStatePerGroupPerTopic {
7878 peerState : make (map [peer.ID ]* peerState ),
7979 groupTTL : max (groupTTL , minGroupTTL ),
8080 }
8181}
8282
83- func (s * partialMessageStatePerTopicGroup ) remotePeerInitiated () bool {
83+ func (s * partialMessageStatePerGroupPerTopic ) remotePeerInitiated () bool {
8484 return s .initiatedBy != ""
8585}
8686
87- func (s * partialMessageStatePerTopicGroup ) clearPeerMetadata (peerID peer.ID ) {
87+ func (s * partialMessageStatePerGroupPerTopic ) clearPeerMetadata (peerID peer.ID ) {
8888 if peerState , ok := s .peerState [peerID ]; ok {
8989 peerState .partsMetadata = nil
9090 if peerState .IsZero () {
@@ -131,9 +131,8 @@ type PartialMessageExtension struct {
131131 // publishing a partial message for the group.
132132 GroupTTLByHeatbeat int
133133
134- // map topic -> map[group]partialMessageStatePerTopicGroup
135- // TODO rename this to ...PerGroupPerTopic
136- statePerTopicPerGroup map [string ]map [string ]* partialMessageStatePerTopicGroup
134+ // map topic -> map[group]partialMessageStatePerGroupPerTopic
135+ statePerTopicPerGroup map [string ]map [string ]* partialMessageStatePerGroupPerTopic
137136
138137 // map[topic]counter
139138 peerInitiatedGroupCounter map [string ]* peerInitiatedGroupCounterState
@@ -155,10 +154,10 @@ type Router interface {
155154 PeerRequestsPartial (peer peer.ID , topic string ) bool
156155}
157156
158- func (e * PartialMessageExtension ) groupState (topic string , groupID []byte , peerInitiated bool , from peer.ID ) (* partialMessageStatePerTopicGroup , error ) {
157+ func (e * PartialMessageExtension ) groupState (topic string , groupID []byte , peerInitiated bool , from peer.ID ) (* partialMessageStatePerGroupPerTopic , error ) {
159158 statePerTopic , ok := e .statePerTopicPerGroup [topic ]
160159 if ! ok {
161- statePerTopic = make (map [string ]* partialMessageStatePerTopicGroup )
160+ statePerTopic = make (map [string ]* partialMessageStatePerGroupPerTopic )
162161 e .statePerTopicPerGroup [topic ] = statePerTopic
163162 }
164163 if _ , ok := e .peerInitiatedGroupCounter [topic ]; ! ok {
@@ -207,7 +206,7 @@ func (e *PartialMessageExtension) Init(router Router) error {
207206 e .PeerInitiatedGroupLimitPerTopicPerPeer = defaultPeerInitiatedGroupLimitPerTopicPerPeer
208207 }
209208
210- e .statePerTopicPerGroup = make (map [string ]map [string ]* partialMessageStatePerTopicGroup )
209+ e .statePerTopicPerGroup = make (map [string ]map [string ]* partialMessageStatePerGroupPerTopic )
211210 e .peerInitiatedGroupCounter = make (map [string ]* peerInitiatedGroupCounterState )
212211
213212 return nil
0 commit comments