package com.codahale.metrics.jetty9;
import com.codahale.metrics.Timer;
import org.eclipse.jetty.io.Connection;
import org.eclipse.jetty.io.EndPoint;
import org.eclipse.jetty.server.ConnectionFactory;
import org.eclipse.jetty.server.Connector;
public class InstrumentedConnectionFactory implements ConnectionFactory {
private final ConnectionFactory connectionFactory;
private final Timer timer;
public InstrumentedConnectionFactory(ConnectionFactory connectionFactory, Timer timer) {
this.connectionFactory = connectionFactory;
this.timer = timer;
}
@Override
public String getProtocol() {
return connectionFactory.getProtocol();
}
@Override
public Connection newConnection(Connector connector, EndPoint endPoint) {
final Connection connection = connectionFactory.newConnection(connector, endPoint);
connection.addListener(new Connection.Listener() {
private Timer.Context context;
@Override
public void onOpened(Connection connection) {
this.context = timer.time();
}
@Override
public void onClosed(Connection connection) {
context.stop();
}
});
return connection;
}
}