mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2024-11-27 08:41:57 -06:00
Code Issues Packages Projects Releases Wiki Activity

Prometheus metrics

Browse source
This commit is contained in:
Neil Alexander 2021-11-03 12:15:53 +00:00
parent e745a7663f
commit f484285f17
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
roomserver/internal/input/input.go
View file

@ -57,12 +57,14 @@ func (r *Inputer) Start() error {
r.InputRoomEventTopic, r.InputRoomEventTopic,
func(msg *nats.Msg) { func(msg *nats.Msg) {
_ = msg.InProgress() _ = msg.InProgress()
roomID := msg.Header.Get("room_id")
defer roomserverInputBackpressure.With(prometheus.Labels{"room_id": roomID}).Dec()
var inputRoomEvent api.InputRoomEvent var inputRoomEvent api.InputRoomEvent
if err := json.Unmarshal(msg.Data, &inputRoomEvent); err != nil { if err := json.Unmarshal(msg.Data, &inputRoomEvent); err != nil {
_ = msg.Nak() _ = msg.Nak()
return return
} }
inbox, _ := r.workers.LoadOrStore(msg.Header.Get("room_id"), &phony.Inbox{}) inbox, _ := r.workers.LoadOrStore(roomID, &phony.Inbox{})
inbox.(*phony.Inbox).Act(nil, func() { inbox.(*phony.Inbox).Act(nil, func() {
if _, err := r.processRoomEvent(context.TODO(), &inputRoomEvent); err != nil { if _, err := r.processRoomEvent(context.TODO(), &inputRoomEvent); err != nil {
sentry.CaptureException(err) sentry.CaptureException(err)
@ -90,7 +92,8 @@ func (r *Inputer) InputRoomEvents(
Subject: r.InputRoomEventTopic, Subject: r.InputRoomEventTopic,
Header: nats.Header{}, Header: nats.Header{},
} }
msg.Header.Set("room_id", e.Event.RoomID()) roomID := e.Event.RoomID()
msg.Header.Set("room_id", roomID)
msg.Data, err = json.Marshal(e) msg.Data, err = json.Marshal(e)
if err != nil { if err != nil {
response.ErrMsg = err.Error() response.ErrMsg = err.Error()
@ -100,6 +103,7 @@ func (r *Inputer) InputRoomEvents(
response.ErrMsg = err.Error() response.ErrMsg = err.Error()
return return
} }
roomserverInputBackpressure.With(prometheus.Labels{"room_id": roomID}).Inc()
} }
} }