本文讲解Go中并发建立SSH连接时常见的死锁问题，重点剖析range遍历未关闭通道导致程序挂起的根本原因，并提供基于WaitGroup与通道协同的健壮解决方案。
本文讲解go中并发建立ssh连接时常见的死锁问题，重点剖析`range`遍历未关闭通道导致程序挂起的根本原因，并提供基于waitgroup与通道协同的健壮解决方案。
在Go中实现对多台服务器的并发SSH连通性探测（如通过TCP端口22检查）是一个典型并发任务，但初学者常因通道（channel）与同步原语（如sync.WaitGroup）的协作不当而陷入死锁。您提供的代码核心问题在于：主goroutine在for cr := range cnres处无限阻塞，因为该通道从未被关闭，而range语法仅在通道关闭且所有已发送值被接收后才退出。
更关键的是，原代码存在两个隐蔽缺陷：
- 变量捕获错误（Closure Bug）：goroutine中使用了外部循环变量host，但未将其作为参数传入闭包。由于for循环复用同一变量地址，所有goroutine实际共享最终迭代的host值，导致连接目标错乱；
- 关闭逻辑错位：close(cnres)放在range循环之后，而该循环本身永不终止，因此close()根本无法执行——形成典型的“先有鸡还是先有蛋”式死锁。
✅ 正确实现方案
以下为修复后的完整示例（已移除AWS SDK依赖以聚焦核心逻辑，您可按需重新集成）：
package main
import (
"fmt"
"net"
"sync"
)
type ConnectionResult struct {
Host string
Message string
Err error // 显式携带错误，便于调试
}
func main() {
// 创建带缓冲的通道，避免goroutine阻塞（缓冲大小建议 ≈ 任务数）
cnres := make(chan ConnectionResult, 100)
var wg sync.WaitGroup
// 模拟服务器列表（替换为您真实的EC2实例列表）
hosts := []string{"server1.example.com", "server2.example.com", "server3.example.com"}
// 启动goroutine消费结果
go func() {
for cr := range cnres {
if cr.Err != nil {
fmt.Printf("❌ Connection to %s failed: %v\n", cr.Host, cr.Err)
} else {
fmt.Printf("✅ Connection to %s %s\n", cr.Host, cr.Message)
}
}
}()
// 并发发起连接
for _, host := range hosts {
wg.Add(1)
go func(hostname string) { // 正确捕获当前host值
defer wg.Done()
conn, err := net.Dial("tcp", hostname+":22")
if err != nil {
cnres <- ConnectionResult{Host: hostname, Message: "failed", Err: err}
} else {
conn.Close() // 及时释放连接资源
cnres <- ConnectionResult{Host: hostname, Message: "succeeded", Err: nil}
}
}(host) // 立即传入当前host副本
}
// 等待所有goroutine完成
wg.Wait()
close(cnres) // 所有发送完成后关闭通道，消费者goroutine自然退出
}⚠️ 关键注意事项
- 通道缓冲很重要：若使用无缓冲通道（make(chan ConnectionResult)），当所有worker goroutine尝试发送结果而消费者尚未启动时，会立即阻塞。设置合理缓冲（如cap=100）可解耦生产/消费节奏；
- 显式关闭时机：close()必须在所有send操作完成后、且由单一协程执行，否则panic；本例中由主线程在wg.Wait()后调用，确保安全；
- 资源清理不可少：成功建立的TCP连接需显式conn.Close()，避免文件描述符泄漏；
- 错误处理要具体：将error字段加入ConnectionResult，比字符串标记更利于诊断网络超时、拒绝连接等具体原因；
- 扩展性建议：生产环境应引入连接超时（net.DialTimeout）、重试机制及限流（如semaphore控制并发数），防止对目标节点造成过大压力。
通过严格遵循“启动消费者 → 启动生产者 → 等待生产者完成 → 关闭通道”的四步模式，即可构建稳定、可预测的并发连接探测系统。
立即学习“go语言免费学习笔记（深入）”；
