// Ranges returns range info for the server specified
func (s *statusServer) Ranges(ctx context.Context, req *serverpb.RangesRequest) (*serverpb.RangesResponse, error) {
nodeID, local, err := s.parseNodeID(req.NodeId)
if err != nil {
return nil, grpc.Errorf(codes.InvalidArgument, err.Error())
}
if !local {
status, err := s.dialNode(nodeID)
if err != nil {
return nil, err
}
return status.Ranges(ctx, req)
}
output := serverpb.RangesResponse{
Ranges: make([]serverpb.RangeInfo, 0, s.stores.GetStoreCount()),
}
err = s.stores.VisitStores(func(store *storage.Store) error {
// Use IterateRangeDescriptors to read from the engine only
// because it's already exported.
err := storage.IterateRangeDescriptors(store.Engine(),
func(desc roachpb.RangeDescriptor) (bool, error) {
rep, err := store.GetReplica(desc.RangeID)
if err != nil {
return true, err
}
status := rep.RaftStatus()
var raftState string
if status != nil {
// We can't put the whole raft.Status object in the json output
// because it contains a map with integer keys. Just extract
// the most interesting bit for now.
raftState = status.RaftState.String()
} else {
raftState = "StateDormant"
}
state := rep.State()
output.Ranges = append(output.Ranges, serverpb.RangeInfo{
Span: serverpb.PrettySpan{
StartKey: desc.StartKey.String(),
EndKey: desc.EndKey.String(),
},
RaftState: raftState,
State: state,
})
return false, nil
})
return err
})
if err != nil {
return nil, grpc.Errorf(codes.Internal, err.Error())
}
return &output, nil
}
func runDebugSplitKey(cmd *cobra.Command, args []string) error {
stopper := stop.NewStopper()
defer stopper.Stop()
if len(args) != 2 {
return errors.New("store and rangeID must be specified")
}
db, err := openStore(cmd, args[0], stopper)
if err != nil {
return err
}
rangeID, err := parseRangeID(args[1])
if err != nil {
return err
}
snap := db.NewSnapshot()
defer snap.Close()
var desc roachpb.RangeDescriptor
if err := storage.IterateRangeDescriptors(snap, func(descInside roachpb.RangeDescriptor) (bool, error) {
if descInside.RangeID == rangeID {
desc = descInside
return true, nil
}
return false, nil
}); err != nil {
return err
}
if desc.RangeID != rangeID {
return fmt.Errorf("range %d not found", rangeID)
}
if splitKey, err := engine.MVCCFindSplitKey(context.Background(), snap, rangeID,
desc.StartKey, desc.EndKey, func(msg string, args ...interface{}) {
fmt.Printf(msg+"\n", args...)
}); err != nil {
fmt.Println("No SplitKey found:", err)
} else {
fmt.Println("Computed SplitKey:", splitKey)
}
return nil
}
// Ranges returns range info for the server specified
func (s *statusServer) Ranges(ctx context.Context, req *serverpb.RangesRequest) (*serverpb.RangesResponse, error) {
nodeID, local, err := s.parseNodeID(req.NodeId)
if err != nil {
return nil, grpc.Errorf(codes.InvalidArgument, err.Error())
}
if !local {
status, err := s.dialNode(nodeID)
if err != nil {
return nil, err
}
return status.Ranges(ctx, req)
}
output := serverpb.RangesResponse{
Ranges: make([]serverpb.RangeInfo, 0, s.stores.GetStoreCount()),
}
convertRaftStatus := func(raftStatus *raft.Status) serverpb.RaftState {
var state serverpb.RaftState
if raftStatus == nil {
state.State = "StateDormant"
return state
}
state.ReplicaID = raftStatus.ID
state.HardState = raftStatus.HardState
state.Applied = raftStatus.Applied
// Grab Lead and State, which together form the SoftState.
state.Lead = raftStatus.Lead
state.State = raftStatus.RaftState.String()
state.Progress = make(map[uint64]serverpb.RaftState_Progress)
for id, progress := range raftStatus.Progress {
state.Progress[id] = serverpb.RaftState_Progress{
Match: progress.Match,
Next: progress.Next,
Paused: progress.Paused,
PendingSnapshot: progress.PendingSnapshot,
State: progress.State.String(),
}
}
return state
}
err = s.stores.VisitStores(func(store *storage.Store) error {
// Use IterateRangeDescriptors to read from the engine only
// because it's already exported.
err := storage.IterateRangeDescriptors(ctx, store.Engine(),
func(desc roachpb.RangeDescriptor) (bool, error) {
rep, err := store.GetReplica(desc.RangeID)
if err != nil {
return true, err
}
output.Ranges = append(output.Ranges, serverpb.RangeInfo{
Span: serverpb.PrettySpan{
StartKey: desc.StartKey.String(),
EndKey: desc.EndKey.String(),
},
RaftState: convertRaftStatus(rep.RaftStatus()),
State: rep.State(),
})
return false, nil
})
return err
})
if err != nil {
return nil, grpc.Errorf(codes.Internal, err.Error())
}
return &output, nil
}