Make AckLogError accept a logger like previous signature

middleware/ack.go

@@ -2,11 +2,9 @@ package middleware
 
 import (
 	"context"
-	"log"
-
-	amqp "github.com/rabbitmq/amqp091-go"
 
 	amqprpc "github.com/0x4b53/amqp-rpc/v3"
+	amqp "github.com/rabbitmq/amqp091-go"
 )
 
 // OnErrFunc is the function that will be called when the middleware get an
@@ -17,9 +15,11 @@ type OnErrFunc func(err error, correlationID string)
 // AckLogError is a built-in function that will log the error if any is returned
 // from `Ack`.
 //
-//	middleware := AckDelivery(AckLogError)
-func AckLogError(err error, correlationID string) {
-	log.Printf("could not ack delivery (%s): %v\n", correlationID, err)
+//	middleware := AckDelivery(AckLogError(log.Printf))
+func AckLogError(logFn amqprpc.LogFunc) OnErrFunc {
+	return func(err error, correlationID string) {
+		logFn("could not ack delivery (%s): %v\n", correlationID, err)
+	}
 }
 
 // AckDelivery is a middleware that will acknowledge the delivery after the

middleware/ack_test.go

@@ -2,6 +2,7 @@ package middleware
 
 import (
 	"context"
+	"log"
 	"testing"
 
 	amqp "github.com/rabbitmq/amqp091-go"
@@ -30,7 +31,7 @@ func TestAckDelivery(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			acknowledger := &amqprpc.MockAcknowledger{}
 
-			handler := AckDelivery(func(_ error, _ string) {})(tt.handler)
+			handler := AckDelivery(AckLogError(log.Printf))(tt.handler)
 
 			rw := amqprpc.NewResponseWriter(&amqp.Publishing{})
 			d := amqp.Delivery{Acknowledger: acknowledger}