mirror of https://github.com/knative/docs.git
28 lines
649 B
Go
28 lines
649 B
Go
package binding
|
|
|
|
type finishMessage struct {
|
|
Message
|
|
finish func(error)
|
|
}
|
|
|
|
func (m *finishMessage) GetWrappedMessage() Message {
|
|
return m.Message
|
|
}
|
|
|
|
func (m *finishMessage) Finish(err error) error {
|
|
err2 := m.Message.Finish(err) // Finish original message first
|
|
if m.finish != nil {
|
|
m.finish(err) // Notify callback
|
|
}
|
|
return err2
|
|
}
|
|
|
|
var _ MessageWrapper = (*finishMessage)(nil)
|
|
|
|
// WithFinish returns a wrapper for m that calls finish() and
|
|
// m.Finish() in its Finish().
|
|
// Allows code to be notified when a message is Finished.
|
|
func WithFinish(m Message, finish func(error)) Message {
|
|
return &finishMessage{Message: m, finish: finish}
|
|
}
|