重构连接等待代码