Skip to content

Commit

Permalink
feat: add cookies
Browse files Browse the repository at this point in the history
  • Loading branch information
franklinkim committed Mar 8, 2024
1 parent 19b1b58 commit a527a59
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions integration/watermill/keel/subscriber.go
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,14 @@ func (s *Subscriber) handle(l *zap.Logger, r *http.Request, event *mpv2.Event) e
msg.Metadata.Set(name, strings.Join(headers, ","))
}

if cookies := r.Cookies(); len(cookies) > 0 {
values := make([]string, len(cookies))
for i, cookie := range r.Cookies() {
values[i] = cookie.String()
}
msg.Metadata.Set("Cookies", strings.Join(values, "; "))
}

for k, v := range msg.Metadata {
l = l.With(zap.String(k, v))
}
Expand All @@ -190,13 +198,13 @@ func (s *Subscriber) handle(l *zap.Logger, r *http.Request, event *mpv2.Event) e
// wait for ACK
select {
case <-msg.Acked():
l.Debug("message acked")
l.Info("message acked")
return nil
case <-msg.Nacked():
l.Debug("message nacked")
l.Info("message nacked")
return ErrMessageNacked
case <-r.Context().Done():
l.Debug("message cancled")
l.Info("message cancled")
return ErrContextCanceled
}
}
Expand Down

0 comments on commit a527a59

Please sign in to comment.