tulkdan.dev
+27
-14
internal/pubsub.go
+27
-14
internal/pubsub.go
···
6
"fmt"
7
"net/http"
8
"slices"
9
"time"
10
)
11
···
16
17
type PubSub struct {
18
topics map[string][]*SubClients
19
}
20
21
func NewPubSub() *PubSub {
···
23
}
24
25
func (p *PubSub) getOrCreateTopic(topicName, requestId string) *SubClients {
26
clients, exists := p.topics[topicName]
27
if exists == false {
28
clients = []*SubClients{}
···
45
return currClient
46
}
47
48
func (p *PubSub) HandleSubscribe(w http.ResponseWriter, r *http.Request) {
49
topicName := r.URL.Query().Get("topic")
50
-
clientInTopic := p.getOrCreateTopic(topicName, r.Context().Value("REQUEST_ID").(string))
51
52
w.Header().Set("Content-Type", "text/event-stream")
53
w.Header().Set("Connection", "keep-alive")
···
59
case <-r.Context().Done():
60
fmt.Println("Client disconnected")
61
62
-
var index int
63
-
for i, client := range p.topics[topicName] {
64
-
if client.id == clientInTopic.id {
65
-
index = i
66
-
break
67
-
}
68
-
}
69
-
70
-
p.topics[topicName] = slices.Delete(p.topics[topicName], index, index+1)
71
-
72
-
if len(p.topics[topicName]) == 0 {
73
-
delete(p.topics, topicName)
74
-
}
75
76
return
77
case data := <-clientInTopic.channel:
···
6
"fmt"
7
"net/http"
8
"slices"
9
+
"sync"
10
"time"
11
)
12
···
17
18
type PubSub struct {
19
topics map[string][]*SubClients
20
+
mu sync.RWMutex
21
}
22
23
func NewPubSub() *PubSub {
···
25
}
26
27
func (p *PubSub) getOrCreateTopic(topicName, requestId string) *SubClients {
28
+
p.mu.Lock()
29
+
defer p.mu.Unlock()
30
+
31
clients, exists := p.topics[topicName]
32
if exists == false {
33
clients = []*SubClients{}
···
50
return currClient
51
}
52
53
+
func (p *PubSub) clientDisconnect(topicName, requestId string) {
54
+
p.mu.Lock()
55
+
defer p.mu.Unlock()
56
+
57
+
var index int
58
+
for i, client := range p.topics[topicName] {
59
+
if client.id == requestId {
60
+
index = i
61
+
break
62
+
}
63
+
}
64
+
65
+
p.topics[topicName] = slices.Delete(p.topics[topicName], index, index+1)
66
+
67
+
if len(p.topics[topicName]) == 0 {
68
+
delete(p.topics, topicName)
69
+
}
70
+
}
71
+
72
func (p *PubSub) HandleSubscribe(w http.ResponseWriter, r *http.Request) {
73
topicName := r.URL.Query().Get("topic")
74
+
requestId := r.Context().Value("REQUEST_ID").(string)
75
+
clientInTopic := p.getOrCreateTopic(topicName, requestId)
76
77
w.Header().Set("Content-Type", "text/event-stream")
78
w.Header().Set("Connection", "keep-alive")
···
84
case <-r.Context().Done():
85
fmt.Println("Client disconnected")
86
87
+
p.clientDisconnect(topicName, requestId)
88
89
return
90
case data := <-clientInTopic.channel: