Github user vanzin commented on a diff in the pull request: https://github.com/apache/spark/pull/21451#discussion_r198653427 --- Diff: core/src/main/scala/org/apache/spark/network/netty/NettyBlockRpcServer.scala --- @@ -73,10 +73,34 @@ class NettyBlockRpcServer( } val data = new NioManagedBuffer(ByteBuffer.wrap(uploadBlock.blockData)) val blockId = BlockId(uploadBlock.blockId) + logInfo(s"Receiving replicated block $blockId with level ${level} " + + s"from ${client.getSocketAddress}") blockManager.putBlockData(blockId, data, level, classTag) responseContext.onSuccess(ByteBuffer.allocate(0)) } } + override def receiveStream( + client: TransportClient, + messageHeader: ByteBuffer, + responseContext: RpcResponseCallback): StreamCallbackWithID = { + val message = BlockTransferMessage.Decoder.fromByteBuffer(messageHeader) + message match { + case uploadBlockStream: UploadBlockStream => + val (level: StorageLevel, classTag: ClassTag[_]) = { --- End diff -- Indentation is off here. Using `.asInstanceOf[UploadBlockStream]` would achieve the same goal here with less indentation, just with a different exception...
--- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org