Fix lack of timeout support in KafkaClient and KafkaConnection