本文介绍在 Go 中高效判断一个或多个 IP 地址是否属于预定义集合的多种方法，重点推荐基于 map[string]bool 的哈希查找方案，并提供可直接复用的 net.IP 键转换函数与完整示例代码。
本文介绍在 go 中高效判断一个或多个 ip 地址是否属于预定义集合的多种方法，重点推荐基于 `map[string]bool` 的哈希查找方案，并提供可直接复用的 `net.ip` 键转换函数与完整示例代码。
Go 语言没有内置的 Set 类型，但可通过 map[T]bool（如 map[string]bool）优雅实现集合语义：键存在即表示元素在集合中，值仅为占位符（通常设为 true）。这种结构支持平均 O(1) 时间复杂度的成员查找，远优于遍历切片的 O(n) 方案，尤其适合高频查询场景（如网络请求鉴权、黑名单校验）。
✅ 推荐方案：使用 map[string]bool + 标准化 IP 键
由于 net.IP 是 []byte 类型（不可比较，不能直接作 map 键），需将其转换为可比较且唯一确定的键。最简洁高效的方式是统一转为 IPv6 格式字节数组，再转为 string：
import (
"net"
"fmt"
)
// Key 将 net.IP 转换为唯一、可比较的字符串键（使用 IPv6 表示，兼容 IPv4）
func Key(ip net.IP) string {
return string(ip.To16())
}
// 示例：构建禁止访问的 IP 集合
forbiddenIPs := map[string]bool{
Key(net.ParseIP("192.168.1.100")): true,
Key(net.ParseIP("2001:db8::1")): true,
}
// 检查单个 IP
ip := net.ParseIP("192.168.1.100")
if forbiddenIPs[Key(ip)] {
fmt.Println("拒绝访问：IP 在黑名单中")
} else {
fmt.Println("允许访问")
}
// 检查 net.LookupIP 返回的多个 IP（典型网络应用流程）
host := "example.com"
ips, err := net.LookupIP(host)
if err != nil {
panic(err)
}
for _, ip := range ips {
if forbiddenIPs[Key(ip)] {
fmt.Printf("警告：主机 %s 的解析 IP %s 被禁止\n", host, ip.String())
// 可中断、记录日志或返回错误
}
}⚠️ 注意事项：
- ip.To16() 自动将 IPv4 映射为 IPv6（如 192.168.1.1 → ::ffff:192.168.1.1），确保同一逻辑 IP 总生成相同键；
- 直接 string([]byte) 安全有效：Go 的 string 可容纳任意字节序列，无需 UTF-8 编码；
- 不建议用 ip.String() 作键（因格式不唯一，如 ::ffff:192.168.1.1 和 192.168.1.1 可能同时存在）；
- 初始化 map 时务必使用 ip.To16()，避免键不一致导致漏查。
? 替代方案：固定长度数组键（零分配、更省内存）
若追求极致性能与内存效率（避免字符串分配），可用 [16]byte 数组作键（数组可比较）：
func KeyArray(ip net.IP) [16]byte {
var a [16]byte
copy(a[:], ip.To16())
return a
}
forbiddenIPs := map[[16]byte]bool{
KeyArray(net.ParseIP("192.168.1.100")): true,
}
// 使用方式相同
if forbiddenIPs[KeyArray(ip)] { /* ... */ }该方案避免字符串内存分配，在大规模 IP 集合（数万+）且 GC 敏感场景下更具优势。
? 扩展场景：IP 段匹配（CIDR 或范围）
若需支持 CIDR 网段（如 192.168.0.0/16）或连续 IP 区间，map 方案不再适用。此时应改用 排序切片 + 二分查找，配合 net.IPNet.Contains() 或自定义区间比较：
import "sort"
type IPRange struct {
Start, End net.IP
}
type IPRangeSlice []IPRange
func (s IPRangeSlice) Len() int { return len(s) }
func (s IPRangeSlice) Less(i, j int) bool { return s[i].Start.Compare(s[j].Start) < 0 }
func (s IPRangeSlice) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
// 查找 IP 是否落在任一范围内
func (s IPRangeSlice) Contains(ip net.IP) bool {
i := sort.Search(len(s), func(k int) bool {
return s[k].Start.Compare(ip) > 0
})
if i > 0 {
r := s[i-1]
return r.Start.Compare(ip) <= 0 && r.End.Compare(ip) >= 0
}
return false
}此方案时间复杂度为 O(log n)，支持灵活的网段管理，适用于防火墙规则、地理 IP 库等场景。
✅ 总结
| 方案 | 时间复杂度 | 适用场景 | 备注 |
|---|---|---|---|
| map[string]bool + ip.To16() | O(1) 平均 | 单 IP 精确匹配（推荐默认方案） | 简洁、安全、易维护 |
| map[[16]byte]bool | O(1) | 高频、大集合、低 GC 压力 | 零分配，稍冗长 |
| 排序切片 + 二分查找 | O(log n) | IP 网段/CIDR 匹配 | 支持范围，扩展性强 |
选择依据：优先用 map[string]bool + Key(ip net.IP) string；若需网段匹配，则切换至排序切片方案。两者可共存——例如用 map 存精确黑名单，用切片存白名单网段，组合构建健壮的网络访问控制层。
