Question

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

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

Answer 1

使用“set”

构建我们的集合

Go中没有内置Set类型，但您可以优雅地使用map[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

注意：我们利用了以下事实：如果某个键不在地图中，则为地图建立索引会导致zero value的值类型，false为bool ，正确地告诉该元素不在地图中（设置）。

在Go Playground上尝试。

^{注意＃2 ：有一些技巧可以让代码将地图设置得更短，您可以在这个答案中查看它们：Check if a value is in a list。< / SUP>}

在集合

中使用net.IP

现在我们只需要一个代表net.IP的类型，它可以用作地图中的键类型（请参阅此问题，了解地图键类型的构成：How can I prevent a type being used as a map key?）。

不幸的是net.IP本身没有资格，因为它是一个片段：

type IP []byte

切片无法比较。有关详细信息，请参阅此问题：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（这是Go将string存储在内存中的方式），但Go中的string值表示任意字节序列，因此以下也可以工作，更简单，效率更高：

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范围，但此解决方案可能是：您可以在O(log2(n))时间内找到覆盖IP地址的范围。

Answer 2

您可以使用func (ip IP) Equal(x IP) bool包中的net：

Equal报告ip和x是否是相同的IP地址。一个IPv4 地址和IPv6格式的相同地址被认为是相同的。

喜欢这个工作样本：

package main

import (
    "fmt"
    "net"
)

func main() {
    ip := net.ParseIP("127.0.0.1")
    ips, err := net.LookupIP("localhost")
    if err != nil {
        panic(err)
    }
    for _, v := range ips {
        if v.Equal(ip) {
            fmt.Println(v)
        }
    }
}

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

2 个答案:

使用“set”

构建我们的集合

在集合

替代密钥类型

替代

排序切片