in common/src/main/java/org/apache/uniffle/common/netty/protocol/Message.java [156:185]
public static Message decode(Type msgType, ByteBuf in) {
switch (msgType) {
case RPC_RESPONSE:
return RpcResponse.decode(in, false);
case SEND_SHUFFLE_DATA_REQUEST:
return SendShuffleDataRequestV1.decode(in);
case GET_LOCAL_SHUFFLE_DATA_REQUEST:
return GetLocalShuffleDataRequest.decode(in);
case GET_LOCAL_SHUFFLE_DATA_V2_REQUEST:
return GetLocalShuffleDataV2Request.decode(in);
case GET_LOCAL_SHUFFLE_DATA_RESPONSE:
return GetLocalShuffleDataResponse.decode(in, true);
case GET_LOCAL_SHUFFLE_INDEX_REQUEST:
return GetLocalShuffleIndexRequest.decode(in);
case GET_LOCAL_SHUFFLE_INDEX_RESPONSE:
return GetLocalShuffleIndexResponse.decode(in, true);
case GET_LOCAL_SHUFFLE_INDEX_V2_RESPONSE:
return GetLocalShuffleIndexV2Response.decode(in, true);
case GET_MEMORY_SHUFFLE_DATA_REQUEST:
return GetMemoryShuffleDataRequest.decode(in);
case GET_MEMORY_SHUFFLE_DATA_RESPONSE:
return GetMemoryShuffleDataResponse.decode(in, true);
case GET_SORTED_SHUFFLE_DATA_REQUEST:
return GetSortedShuffleDataRequest.decode(in);
case GET_SORTED_SHUFFLE_DATA_RESPONSE:
return GetSortedShuffleDataResponse.decode(in, true);
default:
throw new IllegalArgumentException("Unexpected message type: " + msgType);
}
}