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

Clean up old consumers

Browse source
This commit is contained in:
Neil Alexander 2022-02-02 11:57:24 +00:00
parent 7719048161
commit b8beae8f98
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
1 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
setup/jetstream/helpers.go
View file

@ -9,17 +9,25 @@ import (
)
func JetStreamConsumer(
ctx context.Context, nats nats.JetStreamContext, subj, durable string,
ctx context.Context, js nats.JetStreamContext, subj, durable string,
f func(ctx context.Context, msg *nats.Msg) bool,
opts ...nats.SubOpt,
) error {
if cinfo, err := nats.ConsumerInfo(subj, durable); err == nil && cinfo.PushBound {
if err := nats.DeleteConsumer(subj, durable); err != nil {
return fmt.Errorf("nats.DeleteConsumer: %w", err)
defer func() {
// If there are existing consumers from before they were pull
// consumers, we need to clean up the old push consumers. However,
// in order to not affect the interest-based policies, we need to
// do this *after* creating the new pull consumers, which have
// "Pull" suffixed to their name.
if _, err := js.ConsumerInfo(subj, durable); err == nil {
if err := js.DeleteConsumer(subj, durable); err != nil {
logrus.WithContext(ctx).Warnf("Failed to clean up old consumer %q", durable)
}
}
}
}()
sub, err := nats.PullSubscribe(subj, durable, opts...)
name := durable + "Pull"
sub, err := js.PullSubscribe(subj, name, opts...)
if err != nil {
return fmt.Errorf("nats.SubscribeSync: %w", err)
}
@ -32,10 +40,14 @@ func JetStreamConsumer(
logrus.WithContext(ctx).WithField("subject", subj).Fatal(err)
}
for {
msg, err := sub.NextMsgWithContext(ctx)
msgs, err := sub.Fetch(1, nats.Context(ctx))
if err != nil {
handle(fmt.Errorf("sub.NextMsgWithContext: %w", err))
handle(fmt.Errorf("sub.Fetch: %w", err))
}
if len(msgs) < 1 {
continue
}
msg := msgs[0]
if err = msg.InProgress(); err != nil {
handle(fmt.Errorf("msg.InProgress: %w", err))
}