prysm-pulse/beacon-chain/rpc/prysm/v1alpha1/debug/server.go
Radosław Kapka 5569a68452
Code cleanup (#9992)
* Value assigned to a variable is never read before being overwritten

* The result of append is not used anywhere

* Suspicious assignment of range-loop vars detected

* Unused method receiver detected

* Revert "Auxiliary commit to revert individual files from 54edcb445484a2e5d79612e19af8e949b8861253"

This reverts commit bbd1e1beabf7b0c5cfc4f514dcc820062ad6c063.

* Method modifies receiver

* Fix test

* Duplicate imports detected

* Incorrectly formatted error string

* Types of function parameters can be combined

* One more "Unused method receiver detected"

* Unused parameter detected in function
2021-12-07 17:52:39 +00:00

64 lines
2.2 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"
"github.com/golang/protobuf/ptypes/empty"
golog "github.com/ipfs/go-log/v2"
"github.com/prysmaticlabs/prysm/beacon-chain/blockchain"
"github.com/prysmaticlabs/prysm/beacon-chain/db"
"github.com/prysmaticlabs/prysm/beacon-chain/p2p"
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
pbrpc "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
"github.com/sirupsen/logrus"
"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 {
BeaconDB db.NoHeadAccessDatabase
GenesisTimeFetcher blockchain.TimeFetcher
StateGen *stategen.State
HeadFetcher blockchain.HeadFetcher
PeerManager p2p.PeerManager
PeersFetcher p2p.PeersProvider
}
// SetLoggingLevel of a beacon node according to a request type,
// either INFO, DEBUG, or TRACE.
func (_ *Server) SetLoggingLevel(_ context.Context, req *pbrpc.LoggingLevelRequest) (*empty.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(golog.LevelDebug)
// Geth specific logging.
glogger := gethlog.NewGlogHandler(gethlog.StreamHandler(os.Stderr, gethlog.TerminalFormat(true)))
glogger.Verbosity(gethlog.LvlTrace)
gethlog.Root().SetHandler(glogger)
}
return &empty.Empty{}, nil
}