Improve riemann tcp connection reading