package tieventbus

import "github.com/ThreeDotsLabs/watermill/message"

func (e *EventbusImpl) publisherWrapMetrics(publisher message.Publisher) (message.Publisher, error) {
	if e.metricsBuilder != nil {
		return e.metricsBuilder.DecoratePublisher(publisher)
	}
	return publisher, nil
}

func (e *EventbusImpl) subscriberWrapMetrics(subscriber message.Subscriber) (message.Subscriber, error) {
	if e.metricsBuilder != nil {
		return e.metricsBuilder.DecorateSubscriber(subscriber)
	}
	return subscriber, nil
}