mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
028506e7ed
* add recovery for initial sync * add recovery for rpc * adding to other services * remaining services and tests * fix test * remove changes to rpc * handle powchain * handle in powchain * abstract to shared package * gazelle * lint * remove for operations * add in more for operations * travis * fix lint
43 lines
977 B
Go
43 lines
977 B
Go
package messagehandler
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/sirupsen/logrus"
|
|
"go.opencensus.io/trace"
|
|
)
|
|
|
|
var log = logrus.WithField("prefix", "message-handler")
|
|
|
|
// SafelyHandleMessage will recover and log any panic that occurs from the
|
|
// function argument.
|
|
func SafelyHandleMessage(ctx context.Context, fn func(message proto.Message), msg proto.Message) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
printedMsg := "message contains no data"
|
|
if msg != nil {
|
|
printedMsg = proto.MarshalTextString(msg)
|
|
}
|
|
log.WithFields(logrus.Fields{
|
|
"r": r,
|
|
"msg": printedMsg,
|
|
}).Error("Panicked when handling p2p message! Recovering...")
|
|
|
|
if ctx == nil {
|
|
return
|
|
}
|
|
if span := trace.FromContext(ctx); span != nil {
|
|
span.SetStatus(trace.Status{
|
|
Code: trace.StatusCodeInternal,
|
|
Message: fmt.Sprintf("Panic: %v", r),
|
|
})
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Fingers crossed that it doesn't panic...
|
|
fn(msg)
|
|
}
|