diff --git a/akka-remote/src/main/scala/akka/remote/artery/tcp/ArteryTcpTransport.scala b/akka-remote/src/main/scala/akka/remote/artery/tcp/ArteryTcpTransport.scala index 266d4359f4..f37e532571 100644 --- a/akka-remote/src/main/scala/akka/remote/artery/tcp/ArteryTcpTransport.scala +++ b/akka-remote/src/main/scala/akka/remote/artery/tcp/ArteryTcpTransport.scala @@ -171,10 +171,10 @@ private[remote] class ArteryTcpTransport(_system: ExtendedActorSystem, _provider Flow[EnvelopeBuffer] .map { env ⇒ - // TODO Possible performance improvement, could we reduce the copying of bytes? - afr.hiFreq(TcpOutbound_Sent, env.byteBuffer.limit) - val size = env.byteBuffer.limit + val size = env.byteBuffer.limit() + afr.hiFreq(TcpOutbound_Sent, size) + // TODO Possible performance improvement, could we reduce the copying of bytes? val bytes = ByteString(env.byteBuffer) bufferPool.release(env) diff --git a/akka-remote/src/test/scala/akka/remote/artery/tcp/TcpFramingSpec.scala b/akka-remote/src/test/scala/akka/remote/artery/tcp/TcpFramingSpec.scala index 5b63d7ea32..0b1361a309 100644 --- a/akka-remote/src/test/scala/akka/remote/artery/tcp/TcpFramingSpec.scala +++ b/akka-remote/src/test/scala/akka/remote/artery/tcp/TcpFramingSpec.scala @@ -87,8 +87,8 @@ class TcpFramingSpec extends AkkaSpec with ImplicitSender { val frames = Source.fromIterator(() ⇒ rechunk(bytes)).via(framingFlow).runWith(Sink.seq).futureValue frames.size should ===(numberOfFrames) frames.foreach { frame ⇒ - frame.byteBuffer.limit should ===(payload5.size) - val payload = new Array[Byte](frame.byteBuffer.limit) + frame.byteBuffer.limit() should ===(payload5.size) + val payload = new Array[Byte](frame.byteBuffer.limit()) frame.byteBuffer.get(payload) ByteString(payload) should ===(payload5) frame.streamId should ===(3)