feat: handle panic (#254)

utils/debug/debug.go

@@ -0,0 +1,23 @@
+package debug
+
+import (
+	"fmt"
+)
+
+var (
+	PanicError = fmt.Errorf("panic")
+)
+
+type PanicErrorMessage struct {
+	Msg        interface{}
+	Inner      string
+	Stacktrace []byte
+}
+
+func (e *PanicErrorMessage) Error() string {
+	return fmt.Sprintf("%s", e.Inner)
+}
+
+func (e *PanicErrorMessage) Unwrap() []error {
+	return []error{PanicError}
+}

utils/debug/decoder.go

@@ -0,0 +1,22 @@
+package debug
+
+import (
+	"runtime/debug"
+
+	"github.com/netsampler/goflow2/v2/utils"
+)
+
+func PanicDecoderWrapper(wrapped utils.DecoderFunc) utils.DecoderFunc {
+	return func(msg interface{}) (err error) {
+
+		defer func() {
+			if pErr := recover(); pErr != nil {
+
+				pErrC, _ := pErr.(string)
+				err = &PanicErrorMessage{Msg: msg, Inner: pErrC, Stacktrace: debug.Stack()}
+			}
+		}()
+		err = wrapped(msg)
+		return err
+	}
+}

utils/debug/producer.go

@@ -0,0 +1,39 @@
+package debug
+
+import (
+	"runtime/debug"
+
+	"github.com/netsampler/goflow2/v2/producer"
+)
+
+type PanicProducerWrapper struct {
+	wrapped producer.ProducerInterface
+}
+
+func (p *PanicProducerWrapper) Produce(msg interface{}, args *producer.ProduceArgs) (flowMessageSet []producer.ProducerMessage, err error) {
+
+	defer func() {
+		if pErr := recover(); pErr != nil {
+
+			pErrC, _ := pErr.(string)
+			err = &PanicErrorMessage{Msg: msg, Inner: pErrC, Stacktrace: debug.Stack()}
+		}
+	}()
+
+	flowMessageSet, err = p.wrapped.Produce(msg, args)
+	return flowMessageSet, err
+}
+
+func (p *PanicProducerWrapper) Close() {
+	p.wrapped.Close()
+}
+
+func (p *PanicProducerWrapper) Commit(flowMessageSet []producer.ProducerMessage) {
+	p.wrapped.Commit(flowMessageSet)
+}
+
+func WrapPanicProducer(wrapped producer.ProducerInterface) producer.ProducerInterface {
+	return &PanicProducerWrapper{
+		wrapped: wrapped,
+	}
+}