diff --git a/src/main/scala/netty/ActorChannelConnector.scala b/src/main/scala/netty/ActorChannelConnector.scala index d7a8cf6..c0a6d47 100644 --- a/src/main/scala/netty/ActorChannelConnector.scala +++ b/src/main/scala/netty/ActorChannelConnector.scala @@ -34,12 +34,10 @@ final private class ActorChannelConnector( def apply(endpoint: Endpoint, channel: Channel): Unit = val clientPromise = Promise[Client]() channel.attr(key.client).set(clientPromise.future) - val channelEmit: ClientEmit = - emitToChannel(channel, withFlush = endpoint.alwaysFlush) - val monitoredEmit: ClientEmit = (msg: ipc.ClientIn) => + val channelEmit: ClientEmit = (msg: ipc.ClientIn) => endpoint.emitCounter.increment() - channelEmit(msg) - clients ! Clients.Control.Start(endpoint.behavior(monitoredEmit), clientPromise) + emitToChannel(channel, withFlush = endpoint.alwaysFlush) + clients ! Clients.Control.Start(endpoint.behavior(channelEmit), clientPromise) channel.closeFuture.addListener: new GenericFutureListener[NettyFuture[Void]]: def operationComplete(f: NettyFuture[Void]): Unit =