mirror of
https://github.com/openfaas/faas.git
synced 2025-06-18 03:56:37 +00:00
Re-vendor queue-worker publisher for reconnect
- re-vendor queue-worker for publisher via 0.6.0 - bump queue-worker version to 0.6.0 in docker-compose.yml for AMD64 - use new naming for NATS of nats -> NATS in variables where required - add default reconnect of 60 times, 2 seconds apart. Signed-off-by: Alex Ellis (VMware) <alexellis2@gmail.com>
This commit is contained in:
committed by
Alex Ellis
parent
e63150ef70
commit
b4a550327d
64
gateway/vendor/github.com/openfaas/nats-queue-worker/handler/handler.go
generated
vendored
64
gateway/vendor/github.com/openfaas/nats-queue-worker/handler/handler.go
generated
vendored
@ -1,38 +1,13 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/nats-io/go-nats-streaming"
|
||||
"github.com/openfaas/faas/gateway/queue"
|
||||
"regexp"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// NatsQueue queue for work
|
||||
type NatsQueue struct {
|
||||
nc stan.Conn
|
||||
}
|
||||
|
||||
type NatsConfig interface {
|
||||
GetClientID() string
|
||||
}
|
||||
|
||||
type DefaultNatsConfig struct {
|
||||
}
|
||||
|
||||
var supportedCharacters, _ = regexp.Compile("[^a-zA-Z0-9-_]+")
|
||||
|
||||
func (DefaultNatsConfig) GetClientID() string {
|
||||
val, _ := os.Hostname()
|
||||
return getClientId(val)
|
||||
}
|
||||
|
||||
// CreateNatsQueue ready for asynchronous processing
|
||||
func CreateNatsQueue(address string, port int, clientConfig NatsConfig) (*NatsQueue, error) {
|
||||
queue1 := NatsQueue{}
|
||||
// CreateNATSQueue ready for asynchronous processing
|
||||
func CreateNATSQueue(address string, port int, clientConfig NATSConfig) (*NATSQueue, error) {
|
||||
var err error
|
||||
natsURL := fmt.Sprintf("nats://%s:%d", address, port)
|
||||
log.Printf("Opening connection to %s\n", natsURL)
|
||||
@ -40,28 +15,17 @@ func CreateNatsQueue(address string, port int, clientConfig NatsConfig) (*NatsQu
|
||||
clientID := clientConfig.GetClientID()
|
||||
clusterID := "faas-cluster"
|
||||
|
||||
nc, err := stan.Connect(clusterID, clientID, stan.NatsURL(natsURL))
|
||||
queue1.nc = nc
|
||||
queue1 := NATSQueue{
|
||||
ClientID: clientID,
|
||||
ClusterID: clusterID,
|
||||
NATSURL: natsURL,
|
||||
Topic: "faas-request",
|
||||
maxReconnect: clientConfig.GetMaxReconnect(),
|
||||
reconnectDelay: clientConfig.GetReconnectDelay(),
|
||||
ncMutex: &sync.RWMutex{},
|
||||
}
|
||||
|
||||
err = queue1.connect()
|
||||
|
||||
return &queue1, err
|
||||
}
|
||||
|
||||
// Queue request for processing
|
||||
func (q *NatsQueue) Queue(req *queue.Request) error {
|
||||
var err error
|
||||
|
||||
fmt.Printf("NatsQueue - submitting request: %s.\n", req.Function)
|
||||
|
||||
out, err := json.Marshal(req)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
err = q.nc.Publish("faas-request", out)
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func getClientId(hostname string) string {
|
||||
return "faas-publisher-" + supportedCharacters.ReplaceAllString(hostname, "_")
|
||||
}
|
Reference in New Issue
Block a user