watsonb8/fayec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Using one connection for multiple authenticated clients

Browse Source
This commit is contained in:
Brandon Watson
2023-09-19 20:40:52 -05:00
parent 832e02f42f
commit c4ebd5d6ff
3 changed files with 30 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/dispatcher/dispatcher.go
View File

@ -10,6 +10,7 @@ import (
"strconv"
"sync"
"sync/atomic"
"time"
)
type Dispatcher struct {
@ -116,7 +117,7 @@ func (d *Dispatcher) dispatchMessage(msg *message.Message) {
subsList := d.store.GetAll()
for i := range subsList {
sub := subsList[i]
d.Subscribe(sub.Name())
d.Subscribe(sub.Name(), sub.AuthToken())
}
case message.MetaSubscribe:
//handle MetaSubscribe resp
@ -187,13 +188,14 @@ func (d *Dispatcher) sendMessage(m *message.Message) error {
return d.transport.SendMessage(m)
}
func (d *Dispatcher) Subscribe(channel string) (*subscription.Subscription, error) {
func (d *Dispatcher) Subscribe(channel string, authToken string) (*subscription.Subscription, error) {
id := d.nextMsgID()
m := &message.Message{
Channel: message.MetaSubscribe,
ClientId: d.clientID,
Subscription: channel,
Id: id,
Ext: map[string]string{"access_token": authToken, "timestamp": string(time.Now().Unix())},
}
d.extensions.ApplyOutExtensions(m)
@ -208,7 +210,7 @@ func (d *Dispatcher) Subscribe(channel string) (*subscription.Subscription, erro
d.pendingSubs[id] = subscriptionConfirmation
d.pendingSubsMu.Unlock()
sub, err := subscription.NewSubscription(channel, d.Unsubscribe, inMsgCh)
sub, err := subscription.NewSubscription(channel, d.Unsubscribe, authToken, inMsgCh)
if err != nil {
return nil, err
}
@ -245,7 +247,7 @@ func (d *Dispatcher) Unsubscribe(sub *subscription.Subscription) error {
return nil
}
func (d *Dispatcher) Publish(subscription string, data message.Data) (err error) {
func (d *Dispatcher) Publish(subscription string, authToken string, data message.Data) (err error) {
id := d.nextMsgID()
m := &message.Message{
@ -253,6 +255,7 @@ func (d *Dispatcher) Publish(subscription string, data message.Data) (err error)
Data: data,
ClientId: d.clientID,
Id: id,
Ext: map[string]string{"access_token": authToken, "timestamp": string(time.Now().Unix())},
}
//ack from server