112 lines
3.7 KiB
Go
112 lines
3.7 KiB
Go
package writer
|
|
|
|
import (
|
|
"github.com/iMDT/bbb-graphql-middleware/internal/msgpatch"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/iMDT/bbb-graphql-middleware/internal/common"
|
|
log "github.com/sirupsen/logrus"
|
|
"nhooyr.io/websocket/wsjson"
|
|
)
|
|
|
|
// HasuraConnectionWriter
|
|
// process messages (middleware to hasura)
|
|
func HasuraConnectionWriter(hc *common.HasuraConnection, fromBrowserChannel chan interface{}, wg *sync.WaitGroup) {
|
|
log := log.WithField("_routine", "HasuraConnectionWriter")
|
|
|
|
browserConnection := hc.Browserconn
|
|
|
|
log = log.WithField("browserConnectionId", browserConnection.Id).WithField("hasuraConnectionId", hc.Id)
|
|
|
|
defer wg.Done()
|
|
defer hc.ContextCancelFunc()
|
|
defer log.Infof("finished")
|
|
|
|
RangeLoop:
|
|
for {
|
|
select {
|
|
case <-hc.Context.Done():
|
|
break RangeLoop
|
|
case fromBrowserMessage := <-fromBrowserChannel:
|
|
{
|
|
if fromBrowserMessage == nil {
|
|
continue
|
|
}
|
|
|
|
var fromBrowserMessageAsMap = fromBrowserMessage.(map[string]interface{})
|
|
|
|
if fromBrowserMessageAsMap["type"] == "start" {
|
|
var queryId = fromBrowserMessageAsMap["id"].(string)
|
|
|
|
//Identify type based on query string
|
|
messageType := common.Query
|
|
payload := fromBrowserMessageAsMap["payload"].(map[string]interface{})
|
|
query, ok := payload["query"].(string)
|
|
if ok {
|
|
if strings.HasPrefix(query, "subscription") {
|
|
messageType = common.Subscription
|
|
|
|
if strings.Contains(query, "_stream(") && strings.Contains(query, "cursor: {") {
|
|
messageType = common.Streaming
|
|
}
|
|
|
|
if strings.Contains(query, "_aggregate") && strings.Contains(query, "aggregate {") {
|
|
messageType = common.SubscriptionAggregate
|
|
}
|
|
}
|
|
|
|
if strings.HasPrefix(query, "mutation") {
|
|
messageType = common.Mutation
|
|
}
|
|
}
|
|
|
|
//Identify if the client that requested this subscription expects to receive json-patch
|
|
//Client append `Patched_` to the query operationName to indicate that it supports
|
|
jsonPatchSupported := false
|
|
operationName, ok := payload["operationName"].(string)
|
|
if ok && strings.HasPrefix(operationName, "Patched_") {
|
|
jsonPatchSupported = true
|
|
}
|
|
|
|
browserConnection.ActiveSubscriptionsMutex.Lock()
|
|
browserConnection.ActiveSubscriptions[queryId] = common.GraphQlSubscription{
|
|
Id: queryId,
|
|
Message: fromBrowserMessage,
|
|
LastSeenOnHasuraConnetion: hc.Id,
|
|
JsonPatchSupported: jsonPatchSupported,
|
|
Type: messageType,
|
|
}
|
|
// log.Tracef("Current queries: %v", browserConnection.ActiveSubscriptions)
|
|
browserConnection.ActiveSubscriptionsMutex.Unlock()
|
|
}
|
|
|
|
if fromBrowserMessageAsMap["type"] == "stop" {
|
|
var queryId = fromBrowserMessageAsMap["id"].(string)
|
|
browserConnection.ActiveSubscriptionsMutex.RLock()
|
|
jsonPatchSupported := browserConnection.ActiveSubscriptions[queryId].JsonPatchSupported
|
|
browserConnection.ActiveSubscriptionsMutex.RUnlock()
|
|
if jsonPatchSupported {
|
|
msgpatch.RemoveConnSubscriptionCacheFile(browserConnection, queryId)
|
|
}
|
|
browserConnection.ActiveSubscriptionsMutex.Lock()
|
|
delete(browserConnection.ActiveSubscriptions, queryId)
|
|
// log.Tracef("Current queries: %v", browserConnection.ActiveSubscriptions)
|
|
browserConnection.ActiveSubscriptionsMutex.Unlock()
|
|
}
|
|
|
|
if fromBrowserMessageAsMap["type"] == "connection_init" {
|
|
browserConnection.ConnectionInitMessage = fromBrowserMessage
|
|
}
|
|
|
|
log.Tracef("sending to hasura: %v", fromBrowserMessage)
|
|
err := wsjson.Write(hc.Context, hc.Websocket, fromBrowserMessage)
|
|
if err != nil {
|
|
log.Errorf("error on write (we're disconnected from hasura): %v", err)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|