logos-messaging/go-libp2p-pubsub
1
0
Fork 0
mirror of https://github.com/logos-messaging/go-libp2p-pubsub.git synced 2026-01-07 15:23:08 +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 ( import (
"sync" "sync"
pb "github.com/libp2p/go-libp2p-pubsub/pb"
) )
// msgIDGenerator handles computing IDs for msgs // msgIDGenerator handles computing IDs for msgs
@ -33,6 +35,12 @@ func (m *msgIDGenerator) ID(msg *Message) string {
return msg.ID 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() m.topicGensLk.RLock()
gen, ok := m.topicGens[msg.GetTopic()] gen, ok := m.topicGens[msg.GetTopic()]
m.topicGensLk.RUnlock() m.topicGensLk.RUnlock()
@ -40,6 +48,5 @@ func (m *msgIDGenerator) ID(msg *Message) string {
gen = m.Default gen = m.Default
} }
msg.ID = gen(msg.Message) return gen(msg)
return msg.ID
} }