6257ddba78
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.
124 lines
2.7 KiB
Go
124 lines
2.7 KiB
Go
package endpoint
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/sqs"
|
|
"github.com/streadway/amqp"
|
|
)
|
|
|
|
var errCreateQueue = errors.New("Error while creating queue")
|
|
|
|
const (
|
|
sqsExpiresAfter = time.Second * 30
|
|
)
|
|
|
|
// SQSConn is an endpoint connection
|
|
type SQSConn struct {
|
|
mu sync.Mutex
|
|
ep Endpoint
|
|
session *session.Session
|
|
svc *sqs.SQS
|
|
channel *amqp.Channel
|
|
ex bool
|
|
t time.Time
|
|
}
|
|
|
|
func (conn *SQSConn) generateSQSURL() string {
|
|
return "https://sqs." + conn.ep.SQS.Region + "amazonaws.com/" + conn.ep.SQS.QueueID + "/" + conn.ep.SQS.QueueName
|
|
}
|
|
|
|
// Expired returns true if the connection has expired
|
|
func (conn *SQSConn) Expired() bool {
|
|
conn.mu.Lock()
|
|
defer conn.mu.Unlock()
|
|
if !conn.ex {
|
|
if time.Now().Sub(conn.t) > sqsExpiresAfter {
|
|
conn.ex = true
|
|
conn.close()
|
|
}
|
|
}
|
|
return conn.ex
|
|
}
|
|
|
|
func (conn *SQSConn) close() {
|
|
if conn.svc != nil {
|
|
conn.svc = nil
|
|
conn.session = nil
|
|
}
|
|
}
|
|
|
|
// Send sends a message
|
|
func (conn *SQSConn) Send(msg string) error {
|
|
conn.mu.Lock()
|
|
defer conn.mu.Unlock()
|
|
|
|
if conn.ex {
|
|
return errExpired
|
|
}
|
|
conn.t = time.Now()
|
|
|
|
if conn.svc == nil && conn.session == nil {
|
|
credPath := conn.ep.SQS.CredPath
|
|
credProfile := conn.ep.SQS.CredProfile
|
|
var sess *session.Session
|
|
if credPath != "" && credProfile != "" {
|
|
sess = session.Must(session.NewSession(&aws.Config{
|
|
Region: aws.String(conn.ep.SQS.Region),
|
|
Credentials: credentials.NewSharedCredentials(credPath, credProfile),
|
|
MaxRetries: aws.Int(5),
|
|
}))
|
|
} else if credPath != "" {
|
|
sess = session.Must(session.NewSession(&aws.Config{
|
|
Region: aws.String(conn.ep.SQS.Region),
|
|
Credentials: credentials.NewSharedCredentials(credPath, "default"),
|
|
MaxRetries: aws.Int(5),
|
|
}))
|
|
} else {
|
|
sess = session.Must(session.NewSession(&aws.Config{
|
|
Region: aws.String(conn.ep.SQS.Region),
|
|
MaxRetries: aws.Int(5),
|
|
}))
|
|
}
|
|
// Create a SQS service client.
|
|
svc := sqs.New(sess)
|
|
|
|
svc.CreateQueue(&sqs.CreateQueueInput{
|
|
QueueName: aws.String(conn.ep.SQS.QueueName),
|
|
Attributes: map[string]*string{
|
|
"DelaySeconds": aws.String("60"),
|
|
"MessageRetentionPeriod": aws.String("86400"),
|
|
},
|
|
})
|
|
conn.session = sess
|
|
conn.svc = svc
|
|
}
|
|
|
|
queueURL := conn.generateSQSURL()
|
|
// Send message
|
|
sendParams := &sqs.SendMessageInput{
|
|
MessageBody: aws.String(msg),
|
|
QueueUrl: aws.String(queueURL),
|
|
}
|
|
_, err := conn.svc.SendMessage(sendParams)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func newSQSConn(ep Endpoint) *SQSConn {
|
|
return &SQSConn{
|
|
ep: ep,
|
|
t: time.Now(),
|
|
}
|
|
}
|