add recovery middleware

pkg/middleware/logging.go

@@ -7,6 +7,7 @@ import (
 	"github.com/lerenn/asyncapi-codegen/pkg/log"
 )
 
+// Logging is a middleware that logs messages in reception and in publication
 func Logging(logger log.Interface) Middleware {
 	return func(ctx context.Context, next Next) context.Context {
 		apiContext.IfEquals(ctx, apiContext.KeyIsMessageDirection, "reception", func() {

pkg/middleware/recovery.go

@@ -0,0 +1,26 @@
+package middleware
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/lerenn/asyncapi-codegen/pkg/log"
+)
+
+// Recovery is a middleware that recovers from panic in middlewares coming after
+// it and user code from subscription.
+func Recovery(logger log.Interface) Middleware {
+	return func(ctx context.Context, next Next) context.Context {
+		// Recover in case of panic
+		defer func() {
+			if r := recover(); r != nil {
+				logger.Error(ctx, fmt.Sprintf("Recovered from panic: %v", r))
+			}
+		}()
+
+		// Call next middleware
+		next(ctx)
+
+		return ctx
+	}
+}

pkg/utils/duplicates.go

@@ -1,5 +1,6 @@
 package utils
 
+// RemoveDuplicate removes duplicate values from a slice
 func RemoveDuplicate[T string | int](sliceList []T) []T {
 	allKeys := make(map[T]bool)
 	list := []T{}

pkg/utils/reference.go

@@ -1,5 +1,6 @@
 package utils
 
+// ToReference returns a pointer to the given value
 func ToReference[T any](t T) *T {
 	return &t
 }

pkg/utils/string.go

@@ -2,6 +2,7 @@ package utils
 
 import "unicode"
 
+// UpperFirstLetter returns the given string with the first letter in uppercase
 func UpperFirstLetter(str string) string {
 	r := []rune(str)
 	r[0] = unicode.ToUpper(r[0])

pkg/utils/utils.go

@@ -1,5 +1,6 @@
 package utils
 
+// IsInSlice checks if a string is in a slice
 func IsInSlice(slice []string, match string) bool {
 	for _, v := range slice {
 		if v == match {