/
ws.go
103 lines (84 loc) · 2.75 KB
/
ws.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package controllers
import (
"fmt"
api_model "github.com/FurkanSamaraz/Golang-Chat/internal/pkg/model"
api_structure "github.com/FurkanSamaraz/Golang-Chat/internal/pkg/structures"
"github.com/gofiber/websocket/v2"
)
type WsController struct {
Scv api_model.RedisService
}
var clients = make(map[string]*api_structure.Client)
var rooms = make(map[string]*api_structure.Room)
// WsHandler handles WebSocket connections and message sending
// @Summary WebSocket Handler
// @Description Handles WebSocket connections and message sending
// @Tags WebSocket
// @Router /ws [get]
func (redisModel *WsController) WsHandler(c *websocket.Conn) {
user := c.Locals("user").(*api_structure.Claims)
// Kullanıcı listesine kullanıcı ekle
client := &api_structure.Client{Conn: c, Username: user.Name}
clients[user.Name] = client
fmt.Println("clients", len(clients), clients, c.RemoteAddr())
// Gelen yeni mesajlar için süresiz dinle
// Tanımladığımız WebSocket noktamız üzerinden
for {
var msg api_structure.Chat
err := client.Conn.ReadJSON(&msg)
if err != nil {
fmt.Println("error reading json", err)
break
}
fmt.Println("received message", msg)
// Mesajın bir hedef odası olup olmadığını kontrol edin
if msg.Target != nil {
// Odayı al veya oluştur
roomID := msg.Target.ID
room, exists := rooms[roomID]
if !exists {
room = &api_structure.Room{
ID: roomID,
Name: msg.Target.Name,
Clients: make(map[string]*api_structure.Client),
}
rooms[roomID] = room
}
// Odaya kullanıcıyı ekle
room.Clients[client.Username] = client
// Odadaki kullanıcılara mesajı yayınlayın
for _, c := range room.Clients {
if c.Username != client.Username {
err = c.Conn.WriteJSON(msg)
if err != nil {
fmt.Println("error writing json", err)
break
}
// Gönderici ve alıcıyı birbirinin kişi listelerine ekleyin
redisModel.Scv.AddToContactList(c.Username, client.Username)
redisModel.Scv.AddToContactList(client.Username, c.Username)
}
}
// Mesajı Redis'e kaydet
redisModel.Scv.SaveChatHistory(msg)
} else {
// Tüm bağlı istemcilere mesaj yayınlayın (Broadcast)
for _, c := range clients {
if c.Username != client.Username {
err = c.Conn.WriteJSON(msg)
if err != nil {
fmt.Println("error writing json", err)
break
}
// Gönderici ve alıcıyı birbirinin kişi listelerine ekleyin
redisModel.Scv.AddToContactList(c.Username, client.Username)
redisModel.Scv.AddToContactList(client.Username, c.Username)
}
}
// Mesajı Redis'e kaydet
redisModel.Scv.SaveChatHistory(msg)
}
}
fmt.Println("exiting", c.RemoteAddr().String())
delete(clients, user.Name)
}