mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-21 04:52:35 +02:00
b427ff1f88
- fixed access log logic - implement acl at connection level - acl logging - ip/cidr blocking - geoblocking with MaxMind database
39 lines
692 B
Go
39 lines
692 B
Go
package acl
|
|
|
|
import (
|
|
"github.com/puzpuzpuz/xsync/v3"
|
|
acl "github.com/yusing/go-proxy/internal/acl/types"
|
|
"go.uber.org/atomic"
|
|
)
|
|
|
|
var cityCache = xsync.NewMapOf[string, *acl.City]()
|
|
var numCachedLookup atomic.Uint64
|
|
|
|
func (cfg *MaxMindConfig) lookupCity(ip *acl.IPInfo) (*acl.City, bool) {
|
|
if ip.City != nil {
|
|
return ip.City, true
|
|
}
|
|
|
|
if cfg.db.Reader == nil {
|
|
return nil, false
|
|
}
|
|
|
|
city, ok := cityCache.Load(ip.Str)
|
|
if ok {
|
|
numCachedLookup.Inc()
|
|
return city, true
|
|
}
|
|
|
|
cfg.db.RLock()
|
|
defer cfg.db.RUnlock()
|
|
|
|
city = new(acl.City)
|
|
err := cfg.db.Lookup(ip.IP, city)
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
cityCache.Store(ip.Str, city)
|
|
ip.City = city
|
|
return city, true
|
|
}
|