tidwall 6257ddba78 Faster point in polygon / GeoJSON updates
The big change is that the GeoJSON package has been completely
rewritten to fix a few of geometry calculation bugs, increase
performance, and to better follow the GeoJSON spec RFC 7946.

GeoJSON updates

- A LineString now requires at least two points.
- All json members, even foreign, now persist with the object.
- The bbox member persists too but is no longer used for geometry
  calculations. This is change in behavior. Previously Tile38 would
  treat the bbox as the object's physical rectangle.
- Corrections to geometry intersects and within calculations.

Faster spatial queries

- The performance of Point-in-polygon and object intersect operations
  are greatly improved for complex polygons and line strings. It went
  from O(n) to roughly O(log n).
- The same for all collection types with many children, including
  FeatureCollection, GeometryCollection, MultiPoint, MultiLineString,
  and MultiPolygon.

Codebase changes

- The pkg directory has been renamed to internal
- The GeoJSON internal package has been moved to a seperate repo at
  https://github.com/tidwall/geojson. It's now vendored.

Please look out for higher memory usage for datasets using complex
shapes. A complex shape is one that has 64 or more points. For these
shapes it's expected that there will be increase of least 54 bytes per
point.
2018-10-13 04:30:48 -07:00

114 lines
2.1 KiB
Go

package endpoint
import (
"crypto/tls"
"crypto/x509"
"fmt"
"io/ioutil"
"sync"
"time"
paho "github.com/eclipse/paho.mqtt.golang"
)
const (
mqttExpiresAfter = time.Second * 30
)
// MQTTConn is an endpoint connection
type MQTTConn struct {
mu sync.Mutex
ep Endpoint
conn paho.Client
ex bool
t time.Time
}
// Expired returns true if the connection has expired
func (conn *MQTTConn) Expired() bool {
conn.mu.Lock()
defer conn.mu.Unlock()
if !conn.ex {
if time.Now().Sub(conn.t) > mqttExpiresAfter {
conn.close()
conn.ex = true
}
}
return conn.ex
}
func (conn *MQTTConn) close() {
if conn.conn != nil {
if conn.conn.IsConnected() {
conn.conn.Disconnect(250)
}
conn.conn = nil
}
}
// Send sends a message
func (conn *MQTTConn) Send(msg string) error {
conn.mu.Lock()
defer conn.mu.Unlock()
if conn.ex {
return errExpired
}
conn.t = time.Now()
if conn.conn == nil {
uri := fmt.Sprintf("tcp://%s:%d", conn.ep.MQTT.Host, conn.ep.MQTT.Port)
ops := paho.NewClientOptions()
if conn.ep.MQTT.CertFile != "" || conn.ep.MQTT.KeyFile != "" ||
conn.ep.MQTT.CACertFile != "" {
var config tls.Config
if conn.ep.MQTT.CertFile != "" || conn.ep.MQTT.KeyFile != "" {
cert, err := tls.LoadX509KeyPair(conn.ep.MQTT.CertFile,
conn.ep.MQTT.KeyFile)
if err != nil {
return err
}
config.Certificates = append(config.Certificates, cert)
}
if conn.ep.MQTT.CACertFile != "" {
// Load CA cert
caCert, err := ioutil.ReadFile(conn.ep.MQTT.CACertFile)
if err != nil {
return err
}
caCertPool := x509.NewCertPool()
caCertPool.AppendCertsFromPEM(caCert)
config.RootCAs = caCertPool
}
ops = ops.SetTLSConfig(&config)
}
ops = ops.SetClientID("tile38").AddBroker(uri)
c := paho.NewClient(ops)
if token := c.Connect(); token.Wait() && token.Error() != nil {
return token.Error()
}
conn.conn = c
}
t := conn.conn.Publish(conn.ep.MQTT.QueueName, conn.ep.MQTT.Qos,
conn.ep.MQTT.Retained, msg)
t.Wait()
if t.Error() != nil {
conn.close()
return t.Error()
}
return nil
}
func newMQTTConn(ep Endpoint) *MQTTConn {
return &MQTTConn{
ep: ep,
t: time.Now(),
}
}