mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-21 05:43:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

Generate random message ID

Browse source
This commit is contained in:
Neil Alexander 2020-07-14 11:35:19 +01:00
parent fef9ed2a17
commit 01f2f80154
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
federationsender/consumers/eduserver.go
View file

@ -16,6 +16,8 @@ import (
"context"
"encoding/json"
"fmt"
"math/rand"
"time"
"github.com/Shopify/sarama"
"github.com/matrix-org/dendrite/eduserver/api"
@ -106,6 +108,15 @@ func (t *OutputEDUConsumer) onSendToDeviceEvent(msg *sarama.ConsumerMessage) err
return nil
}
// Generate a random message ID for idempotency
chars := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
random := rand.New(rand.NewSource(time.Now().UnixNano()))
messageID := make([]byte, 32)
for i := range messageID {
messageID[i] = chars[random.Intn(len(chars))]
}
// Pack the EDU and marshal it
edu := &gomatrixserverlib.EDU{
Type: gomatrixserverlib.MDirectToDevice,
Origin: string(t.ServerName),
@ -113,6 +124,7 @@ func (t *OutputEDUConsumer) onSendToDeviceEvent(msg *sarama.ConsumerMessage) err
tdm := gomatrixserverlib.ToDeviceMessage{
Sender: ote.Sender,
Type: ote.Type,
MessageID: string(messageID),
Messages: map[string]map[string]json.RawMessage{
ote.UserID: {
ote.DeviceID: ote.Content,