TRKJ/EdgeCommon
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeCommon.git synced 2025-12-08 08:00:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

实现IP库制品管理API、自动更新程序

Browse Source
This commit is contained in:
GoEdgeLab
2022-08-23 14:54:53 +08:00
parent 5b14b02df8
commit f230cf410f
11 changed files with 5157 additions and 3303 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
pkg/iplibrary/default_ip_library.go
View File

@@ -12,18 +12,23 @@ import (
//go:embed internal-ip-library.db
var ipLibraryData []byte
var library = NewIPLibrary()
var defaultLibrary = NewIPLibrary()
func Init() error {
return library.Init()
func DefaultIPLibraryData() []byte {
return ipLibraryData
}
func InitDefault() error {
defaultLibrary.reader = nil
return defaultLibrary.InitFromData(ipLibraryData)
}
func Lookup(ip net.IP) *QueryResult {
return library.Lookup(ip)
return defaultLibrary.Lookup(ip)
}
func LookupIP(ip string) *QueryResult {
return library.LookupIP(ip)
return defaultLibrary.LookupIP(ip)
}
type IPLibrary struct {
@@ -34,11 +39,15 @@ func NewIPLibrary() *IPLibrary {
return &IPLibrary{}
}
func (this *IPLibrary) Init() error {
if len(ipLibraryData) == 0 || this.reader != nil {
func NewIPLibraryWithReader(reader *Reader) *IPLibrary {
return &IPLibrary{reader: reader}
}
func (this *IPLibrary) InitFromData(data []byte) error {
if len(data) == 0 || this.reader != nil {
return nil
}
var reader = bytes.NewReader(ipLibraryData)
var reader = bytes.NewReader(data)
gzipReader, err := gzip.NewReader(reader)
if err != nil {
return err