logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-02 12:53:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: extract RawID from ID

Browse Source
This commit is contained in:
Wondertan 2022-01-22 13:34:19 +02:00 committed by vyzo
parent 6faf473cad
commit 9fbf906fb5
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
midgen.go
View File

@ -2,6 +2,8 @@ package pubsub
import (
"sync"
pb "github.com/libp2p/go-libp2p-pubsub/pb"
)
// msgIDGenerator handles computing IDs for msgs
@ -33,6 +35,12 @@ func (m *msgIDGenerator) ID(msg *Message) string {
return msg.ID
}
msg.ID = m.RawID(msg.Message)
return msg.ID
}
// RawID computes ID for the proto 'msg'.
func (m *msgIDGenerator) RawID(msg *pb.Message) string {
m.topicGensLk.RLock()
gen, ok := m.topicGens[msg.GetTopic()]
m.topicGensLk.RUnlock()
@ -40,6 +48,5 @@ func (m *msgIDGenerator) ID(msg *Message) string {
gen = m.Default
}
msg.ID = gen(msg.Message)
return msg.ID
return gen(msg)
}