// handleRequest receive Request then send empty Response back fill with same Type.
func handleRequest(conn net.Conn, c *C) {
c.Assert(conn, NotNil)
defer conn.Close()
var msg msgpb.Message
msgID, err := util.ReadMessage(conn, &msg)
c.Assert(err, IsNil)
c.Assert(msgID, Greater, uint64(0))
c.Assert(msg.GetMsgType(), Equals, msgpb.MessageType_KvReq)
req := msg.GetKvReq()
c.Assert(req, NotNil)
var resp pb.Response
resp.Type = req.Type
msg = msgpb.Message{
MsgType: msgpb.MessageType_KvResp,
KvResp: &resp,
}
err = util.WriteMessage(conn, msgID, &msg)
c.Assert(err, IsNil)
}
// errMismatch if response mismatches request return error.
func errMismatch(resp *pb.Response, req *pb.Request) error {
return errors.Errorf("message type mismatches, response[%s] request[%s]",
resp.GetType(), req.GetType())
}
func (h *rpcHandler) handleRequest(req *kvrpcpb.Request) *kvrpcpb.Response {
var resp kvrpcpb.Response
if err := h.checkContext(req.GetContext()); err != nil {
resp.RegionError = err
return &resp
}
switch req.GetType() {
case kvrpcpb.MessageType_CmdGet:
resp.CmdGetResp = h.onGet(req.CmdGetReq)
case kvrpcpb.MessageType_CmdScan:
resp.CmdScanResp = h.onScan(req.CmdScanReq)
case kvrpcpb.MessageType_CmdPrewrite:
resp.CmdPrewriteResp = h.onPrewrite(req.CmdPrewriteReq)
case kvrpcpb.MessageType_CmdCommit:
resp.CmdCommitResp = h.onCommit(req.CmdCommitReq)
case kvrpcpb.MessageType_CmdCleanup:
resp.CmdCleanupResp = h.onCleanup(req.CmdCleanupReq)
case kvrpcpb.MessageType_CmdBatchGet:
resp.CmdBatchGetResp = h.onBatchGet(req.CmdBatchGetReq)
case kvrpcpb.MessageType_CmdScanLock:
resp.CmdResolveLockResp = h.onResolveLock(req.CmdResolveLockReq)
case kvrpcpb.MessageType_CmdResolveLock:
resp.CmdResolveLockResp = h.onResolveLock(req.CmdResolveLockReq)
case kvrpcpb.MessageType_CmdRawGet:
resp.CmdRawGetResp = h.onRawGet(req.CmdRawGetReq)
case kvrpcpb.MessageType_CmdRawPut:
resp.CmdRawPutResp = h.onRawPut(req.CmdRawPutReq)
case kvrpcpb.MessageType_CmdRawDelete:
resp.CmdRawDeleteResp = h.onRawDelete(req.CmdRawDeleteReq)
}
resp.Type = req.Type
return &resp
}