go语言实现高性能自定义ip管理模块(ip黑名单)
package main
import (
"fmt"
"net"
)
// Blacklist 定义黑名单结构体
type Blacklist struct {
ips map[string]bool // 使用map存储IP及其是否在黑名单中
}
// NewBlacklist 创建一个新的黑名单实例
func NewBlacklist() *Blacklist {
return &Blacklist{
ips: make(map[string]bool),
}
}
// Add 将IP添加到黑名单
func (bl *Blacklist) Add(ip string) {
bl.ips[ip] = true
}
// Remove 从黑名单移除IP
func (bl *Blacklist) Remove(ip string) {
delete(bl.ips, ip)
}
// Contains 检查IP是否在黑名单中
func (bl *Blacklist) Contains(ip string) bool {
return bl.ips[ip]
}
// IsIPv4 检查字符串是否为有效的IPv4地址
func IsIPv4(ip string) bool {
addr := net.ParseIP(ip)
return addr != nil && addr.To4() != nil
}
func main() {
blacklist := NewBlacklist()
blacklist.Add("192.168.1.1")
blacklist.Add("10.0.0.1")
if blacklist.Contains("192.168.1.1") {
fmt.Println("IP 192.168.1.1 is in the blacklist")
}
if !IsIPv4("123.45.67.89") {
fmt.Println("Invalid IP format")
}
}
这段代码定义了一个Blacklist
结构体,并实现了添加、移除和检查IP是否在黑名单中的方法。同时,提供了一个IsIPv4
函数来检查字符串是否为有效的IPv4地址。这个实现简洁且高效,并且提供了清晰的注释。
评论已关闭