mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 03:30:35 +00:00
5a66807989
* First take at updating everything to v5 * Patch gRPC gateway to use prysm v5 Fix patch * Update go ssz --------- Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
57 lines
2.2 KiB
Go
57 lines
2.2 KiB
Go
package sync
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/libp2p/go-libp2p/core/network"
|
|
"github.com/prysmaticlabs/prysm/v5/config/params"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var defaultReadDuration = ttfbTimeout
|
|
var defaultWriteDuration = params.BeaconConfig().RespTimeoutDuration() // RESP_TIMEOUT
|
|
|
|
// SetRPCStreamDeadlines sets read and write deadlines for libp2p-based connection streams.
|
|
func SetRPCStreamDeadlines(stream network.Stream) {
|
|
SetStreamReadDeadline(stream, defaultReadDuration)
|
|
SetStreamWriteDeadline(stream, defaultWriteDuration)
|
|
}
|
|
|
|
// SetStreamReadDeadline for reading from libp2p connection streams, deciding when to close
|
|
// a connection based on a particular duration.
|
|
//
|
|
// NOTE: libp2p uses the system clock time for determining the deadline so we use
|
|
// time.Now() instead of the synchronized roughtime.Now(). If the system
|
|
// time is corrupted (i.e. time does not advance), the node will experience
|
|
// issues being able to properly close streams, leading to unexpected failures and possible
|
|
// memory leaks.
|
|
func SetStreamReadDeadline(stream network.Stream, duration time.Duration) {
|
|
if err := stream.SetReadDeadline(time.Now().Add(duration)); err != nil &&
|
|
!strings.Contains(err.Error(), "stream closed") {
|
|
log.WithError(err).WithFields(logrus.Fields{
|
|
"peer": stream.Conn().RemotePeer(),
|
|
"protocol": stream.Protocol(),
|
|
"direction": stream.Stat().Direction,
|
|
}).Debug("Could not set stream deadline")
|
|
}
|
|
}
|
|
|
|
// SetStreamWriteDeadline for writing to libp2p connection streams, deciding when to close
|
|
// a connection based on a particular duration.
|
|
//
|
|
// NOTE: libp2p uses the system clock time for determining the deadline so we use
|
|
// time.Now() instead of the synchronized roughtime.Now(). If the system
|
|
// time is corrupted (i.e. time does not advance), the node will experience
|
|
// issues being able to properly close streams, leading to unexpected failures and possible
|
|
// memory leaks.
|
|
func SetStreamWriteDeadline(stream network.Stream, duration time.Duration) {
|
|
if err := stream.SetWriteDeadline(time.Now().Add(duration)); err != nil {
|
|
log.WithError(err).WithFields(logrus.Fields{
|
|
"peer": stream.Conn().RemotePeer(),
|
|
"protocol": stream.Protocol(),
|
|
"direction": stream.Stat().Direction,
|
|
}).Debug("Could not set stream deadline")
|
|
}
|
|
}
|