在Golang中检查IP地址片中IP的有效方法

本篇文章向大家介绍《在Golang中检查IP地址片中IP的有效方法》，主要包括golang，具有一定的参考价值，需要的朋友可以参考一下。

问题内容

我正在用 Golang 开发一个网络应用程序。我有一部分 IP 地址。每次收到请求时，我都会使用net.LookupIP(host)主机的 IP 地址来查找返回切片的net.IP. 比较这些的最佳方法是什么？

顺便说一句，在 Python 中，我们有一个set数据结构，它使上述问题很容易解决，但是 Go 呢？

正确答案

带“套”

构建我们的集合

Go中没有内置Set类型，但您可以优雅地使用 amap[Type]bool作为集合，例如：

// Create a set with 2 values in it: [1, 2]
m := map[int]bool{1: true, 2: true}

// Test an element:
fmt.Println(m[1]) // true
fmt.Println(m[3]) // false

// Set an element:
m[3] = true
fmt.Println(m[3]) // true

// Delete an element:
delete(m, 1)
fmt.Println(m[1]) // false

注意：我们利用了这样一个事实，即如果键不在映射中，索引映射会导致值类型的值为零false，在 的情况下bool，正确地告诉元素不在映射（集合）中。

在Go Playground上尝试一下。

注意＃2 ：有一些技巧可以使将地图作为一组处理的代码更短，您可以在以下答案中检查它们：检查值是否在列表中。

net.IP在集合中使用

现在我们只需要一个表示 a 的类型net.IP，它可以用作映射中的键类型（请参阅这个关于什么构成映射键类型的问题：[如何防止类型被用作映射键？](https://stackoverflow.com/questions/37916647/how- can-i-prevent-a-type-being-used-as-a-map-key)）。

不幸的是net.IP，它本身不符合条件，因为它是一个切片：

type IP []byte

和切片没有可比性。有关详细信息，请参阅此问题：[Hash with key as an array type](https://stackoverflow.com/questions/29175247/hash-with-key-as-an-array- type)和：Why have arrays in Go？

一种简单的方法是将其转换为规范string值，我们就完成了。为此，我们可以简单地将 IP 的字节转换为十六进制string。但是 IPv4 地址可能会显示为 IPv6，所以我们应该先将其转换为 IPv6：

func Key(ip net.IP) string {
    return hex.EncodeToString(ip.To16())
}

注意：IP 地址的字节可能不是有效的 UTF-8 编码string（这是 Gostring在内存中存储 s 的方式），但stringGo 中的值表示任意字节序列，因此以下方法也有效，更简单，效率更高：

func Key(ip net.IP) string {
    return string(ip.To16())  // Simple []byte => string conversion
}

我们可以使用这样的 IP 字符串作为键。使用 IP 填充您的地图以检查：

// Populate forbidden IPs:
forbIPs := map[string]bool{
    Key(ip1): true,
    Key(ip2): true,
}

// Now check a single IP:
ipToCheck := ...
if forbIPs[Key(ipToCheck)] {
    fmt.Println("Forbidden!")
} else {
    fmt.Println("Allowed.")
}

如果您要检查多个 IP（由 返回net.LookupIP()），则这是一个for循环：

ips, err := net.LookupIP(host)
// Check err
for _, ip := range ips {
    if forbIPs[Key(ip)] {
        // FORBIDDEN!
    }
}

备用键类型

请注意，“上面提到的”切片不可比较，但数组可以。所以我们也可以使用数组作为键。这就是它的样子：

func Key(ip net.IP) (a [16]byte) {
    copy(a[:], ip)
    return
}

// And the IP set:
forbIPs := map[[16]byte]bool{
    // ...
}

备择方案

排序切片

或者，我们可以简单地将禁止的 IP 存储在一个切片[]net.IP中，并 对其进行排序 。如果它是排序的，我们可以使用二进制搜索在其中找到一个IP（标准库sort.Search()）。

是的，与上面的（散列）映射解决方案O(log2(n))的复杂性相比，二分搜索具有复杂性。O(1)但是这个替代方案还有另一个优点：

枚举单个 IP 并不总是可行的。有时（通常）列出 IP 范围更容易。第一种解决方案对于处理 IP 范围是不可行的，但这种解决方案可能是：您可以及时找到覆盖 IP 地址的范围O(log2(n))。

理论要掌握，实操不能落！以上关于《在Golang中检查IP地址片中IP的有效方法》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！