From 1223039a813fa836b3a7616225fc2d488b396249 Mon Sep 17 00:00:00 2001 From: Kali Kaneko Date: Thu, 13 Dec 2018 08:45:27 -0500 Subject: geolocate gateways and kd-tree during initialization, we fetch the eip-config.json file from the configured provider. we geolocate the gateways (using a golang package that has some cities missing, hence the workaround) and initialize a KD-Tree with the gateways. using the KD-Tree, it is very cheap to calculate the nearest gateway for every request, which is provider as a filed in the json to the client - as a suggestion to be used or not in the gateway selection process. --- geo.go | 40 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 geo.go (limited to 'geo.go') diff --git a/geo.go b/geo.go new file mode 100644 index 0000000..e004fbb --- /dev/null +++ b/geo.go @@ -0,0 +1,40 @@ +package main + +import ( + "github.com/hongshibao/go-kdtree" +) + +type EuclideanPoint struct { + kdtree.Point + Vec []float64 +} + +func (p EuclideanPoint) Dim() int { + return len(p.Vec) +} + +func (p EuclideanPoint) GetValue(dim int) float64 { + return p.Vec[dim] +} + +func (p EuclideanPoint) Distance(other kdtree.Point) float64 { + var ret float64 + for i := 0; i < p.Dim(); i++ { + tmp := p.GetValue(i) - other.GetValue(i) + ret += tmp * tmp + } + return ret +} + +func (p EuclideanPoint) PlaneDistance(val float64, dim int) float64 { + tmp := p.GetValue(dim) - val + return tmp * tmp +} + +func NewEuclideanPoint(vals ...float64) *EuclideanPoint { + ret := &EuclideanPoint{} + for _, val := range vals { + ret.Vec = append(ret.Vec, val) + } + return ret +} -- cgit v1.2.3