Skip to content

Commit ca33674

Browse files
committed
partialmessages: rename field to reflect use
changes partialMessageStatePerTopicGroup to partialMessageStatePerGroupPerTopic to better reflect its usage.
1 parent b6284b0 commit ca33674

File tree

1 file changed

+10
-11
lines changed

1 file changed

+10
-11
lines changed

partialmessages/partialmsgs.go

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)