首页 > 后端开发 > Golang > 正文

IP地址设置操作使用NET/NETIP

心靈之曲
发布: 2025-02-01 08:12:30
原创
1059人浏览过

ip地址设置操作使用net/netip

本文承接前文,深入探讨net/netip包的高级应用——IP地址集合处理。虽然net/netip本身不直接支持集合类型,但我们可以通过巧妙的抽象来高效管理IP范围和集合。

IP集合的必要性

IP集合在以下场景中至关重要:

  • 管理允许/拒绝IP列表
  • 处理网络ACL(访问控制列表)
  • 高效利用IP范围
  • 执行集合操作(并集、交集、差集)

构建IP集合类型

我们构建一个灵活的IP集合实现:

<code class="go">package ipset

import (
    "fmt"
    "net/netip"
    "sort"
)

// range 表示连续的IP地址范围
type range struct {
    first netip.Addr
    last  netip.Addr
}

// ipset 使用范围表示IP地址集合
type ipset struct {
    ranges []range
}

func newipset() *ipset {
    return &ipset{}
}

// add 向集合添加单个IP地址
func (s *ipset) add(ip netip.Addr) {
    s.addrange(range{first: ip, last: ip})
}

// addrange 向集合添加IP范围
func (s *ipset) addrange(r range) {
    if r.first.Compare(r.last) > 0 {
        r.first, r.last = r.last, r.first
    }

    idx := sort.Search(len(s.ranges), func(i int) bool {
        return s.ranges[i].first.Compare(r.first) > 0
    })

    s.ranges = append(s.ranges, range{})
    copy(s.ranges[idx+1:], s.ranges[idx:])
    s.ranges[idx] = r

    s.optimize()
}

// optimize 合并重叠或相邻的范围
func (s *ipset) optimize() {
    if len(s.ranges) < 2 {
        return
    }

    merged := []range{}
    current := s.ranges[0]
    for _, next := range s.ranges[1:] {
        if current.last.Compare(next.first) >= 0 {
            // 合并重叠范围
            last := current.last
            if next.last.Compare(last) > 0 {
                last = next.last
            }
            current.last = last
        } else {
            merged = append(merged, current)
            current = next
        }
    }
    merged = append(merged, current)
    s.ranges = merged
}

// contains 检查集合中是否包含某个IP地址
func (s *ipset) contains(ip netip.Addr) bool {
    idx := sort.Search(len(s.ranges), func(i int) bool {
        return s.ranges[i].last.Compare(ip) >= 0
    })

    if idx < len(s.ranges) && s.ranges[idx].first.Compare(ip) <= 0 {
        return true
    }
    return false
}
</code>
登录后复制

集合操作

接下来实现基本的集合操作:

<code class="go">// union 返回一个新集合,包含两个集合的所有IP
func (s *ipset) union(other *ipset) *ipset {
    result := newipset()
    for _, r := range s.ranges {
        result.addrange(r)
    }
    for _, r := range other.ranges {
        result.addrange(r)
    }
    return result
}

// intersection 返回一个新集合,包含两个集合中都存在的IP
func (s *ipset) intersection(other *ipset) *ipset {
    result := newipset()
    for _, r1 := range s.ranges {
        for _, r2 := range other.ranges {
            first := r1.first
            if r2.first.Compare(first) > 0 {
                first = r2.first
            }
            last := r1.last
            if r2.last.Compare(last) < 0 {
                last = r2.last
            }
            if first.Compare(last) <= 0 {
                result.addrange(range{first: first, last: last})
            }
        }
    }
    return result
}

// difference 返回一个新集合,包含在第一个集合中但在第二个集合中不存在的IP
func (s *ipset) difference(other *ipset) *ipset {
    result := newipset()
    for _, r1 := range s.ranges {
        for _, r2 := range other.ranges {
            // 复杂的范围差集计算,略去详细代码,需要考虑多种情况
            // ...
        }
    }
    return result
}</code>
登录后复制

(此处省略了 difference 函数的复杂实现,它需要处理各种范围重叠和非重叠情况)

现实世界应用示例

  1. IP地址列表管理: 一个访问管理器,管理允许和阻止的IP列表。

  2. 网络范围计算器: 一个工具,用于处理IP范围和子网

  3. 防火墙规则优化器: 一个优化重叠防火墙规则的工具。

(此处省略了这三个应用示例的完整代码,它们基于上述ipset类型和函数)

性能考虑

  • 范围优化: 合并重叠范围以提高效率。
  • 二分查找: 在范围内查找IP时使用二分查找。
  • 内存使用: 存储范围而不是单个IP以减少内存消耗。

最佳实践

  • 始终验证输入。
  • 分别处理IPv4和IPv6。
  • 使用高效的集合操作。

后续

下一篇文章将探讨net/netip包中内置的IP地址处理方法。 虽然本文构建了自己的集合实现,但了解net/netip的内置功能对于高效的网络编程至关重要。

记住,IP集合处理可能很复杂,请务必彻底测试您的实现。

以上就是IP地址设置操作使用NET/NETIP的详细内容,更多请关注php中文网其它相关文章!

相关标签:
最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号