add registrations cleanup

mehdihadeli · Oct 28, 2022 · 053a7a6 · 053a7a6
1 parent a32a663
commit 053a7a6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/mediatr.go b/mediatr.go
@@ -94,6 +94,14 @@ func RegisterNotificationHandlers[TEvent any](handlers ...NotificationHandler[TE
 	return nil
 }
 
+func ClearRequestRegistrations() {
+	requestHandlersRegistrations = map[reflect.Type]interface{}{}
+}
+
+func ClearNotificationRegistrations() {
+	notificationHandlersRegistrations = map[reflect.Type][]interface{}{}
+}
+
 // Send the request to its corresponding request handler.
 func Send[TRequest any, TResponse any](ctx context.Context, request TRequest) (TResponse, error) {
 	requestType := reflect.TypeOf(request)