mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-07 10:12:19 +00:00
8b168aa450
* add in rpc debug server * sets the valid debug level * builds * make post * Merge branch 'master' into debuglog * Merge refs/heads/master into debuglog * caps * Merge branch 'debuglog' of github.com:prysmaticlabs/prysm into debuglog * comment
59 lines
2.0 KiB
Go
59 lines
2.0 KiB
Go
// Package debug defines a gRPC server implementation of a debugging service
|
|
// which allows for helpful endpoints to debug a beacon node at runtime, this server is
|
|
// gated behind the feature flag --enable-debug-rpc-endpoints.
|
|
package debug
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
gethlog "github.com/ethereum/go-ethereum/log"
|
|
ptypes "github.com/gogo/protobuf/types"
|
|
golog "github.com/ipfs/go-log"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/blockchain"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
pbrpc "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
|
|
"github.com/sirupsen/logrus"
|
|
gologging "github.com/whyrusleeping/go-logging"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// Server defines a server implementation of the gRPC Debug service,
|
|
// providing RPC endpoints for runtime debugging of a node, this server is
|
|
// gated behind the feature flag --enable-debug-rpc-endpoints.
|
|
type Server struct {
|
|
GenesisTimeFetcher blockchain.TimeFetcher
|
|
StateGen *stategen.State
|
|
}
|
|
|
|
// SetLoggingLevel of a beacon node according to a request type,
|
|
// either INFO, DEBUG, or TRACE.
|
|
func (ds *Server) SetLoggingLevel(ctx context.Context, req *pbrpc.LoggingLevelRequest) (*ptypes.Empty, error) {
|
|
var verbosity string
|
|
switch req.Level {
|
|
case pbrpc.LoggingLevelRequest_INFO:
|
|
verbosity = "info"
|
|
case pbrpc.LoggingLevelRequest_DEBUG:
|
|
verbosity = "debug"
|
|
case pbrpc.LoggingLevelRequest_TRACE:
|
|
verbosity = "trace"
|
|
default:
|
|
return nil, status.Error(codes.InvalidArgument, "Expected valid verbosity level as argument")
|
|
}
|
|
level, err := logrus.ParseLevel(verbosity)
|
|
if err != nil {
|
|
return nil, status.Error(codes.Internal, "Could not parse verbosity level")
|
|
}
|
|
logrus.SetLevel(level)
|
|
if level == logrus.TraceLevel {
|
|
// Libp2p specific logging.
|
|
golog.SetAllLoggers(gologging.DEBUG)
|
|
// Geth specific logging.
|
|
glogger := gethlog.NewGlogHandler(gethlog.StreamHandler(os.Stderr, gethlog.TerminalFormat(true)))
|
|
glogger.Verbosity(gethlog.LvlTrace)
|
|
gethlog.Root().SetHandler(glogger)
|
|
}
|
|
return &ptypes.Empty{}, nil
|
|
}
|