func (h *rpcHandler) onGet(req *kvrpcpb.CmdGetRequest) *kvrpcpb.CmdGetResponse { if !h.keyInRegion(req.Key) { panic("onGet: key not in region") } val, err := h.mvccStore.Get(req.Key, req.GetVersion()) if err != nil { return &kvrpcpb.CmdGetResponse{ Error: convertToKeyError(err), } } return &kvrpcpb.CmdGetResponse{ Value: val, } }
// One normally `Send`. func (s *testClientSuite) TestSendBySelf(c *C) { l := startServer(":61234", c, handleRequest) defer l.Close() cli := newRPCClient() req := new(pb.Request) req.Type = pb.MessageType_CmdGet getReq := new(pb.CmdGetRequest) getReq.Key = []byte("a") ver := uint64(0) getReq.Version = ver req.CmdGetReq = getReq resp, err := cli.SendKVReq(":61234", req, readTimeoutShort) c.Assert(err, IsNil) c.Assert(req.GetType(), Equals, resp.GetType()) }