add cache
This commit is contained in:
@@ -0,0 +1,44 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "sync"
|
||||||
|
|
||||||
|
var (
|
||||||
|
mu sync.RWMutex
|
||||||
|
ChatGroups = make(map[uint32]ChatGroup)
|
||||||
|
Clients = make(map[uint32]Client)
|
||||||
|
ClientsMap = make(map[uint32]map[uint32]*Client)
|
||||||
|
)
|
||||||
|
|
||||||
|
func AddGroupToCache(chatGroup *ChatGroup) {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
ChatGroups[chatGroup.Id] = *chatGroup
|
||||||
|
ClientsMap[chatGroup.Id] = make(map[uint32]*Client)
|
||||||
|
}
|
||||||
|
|
||||||
|
func RemoveGroupFromCache(chatGroup *ChatGroup) {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
delete(ChatGroups, chatGroup.Id)
|
||||||
|
delete(ClientsMap, chatGroup.Id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func AddUserToCache(client *Client) {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
for _, groupIn := range client.Groups {
|
||||||
|
if clients, ok := ClientsMap[groupIn.Id]; ok {
|
||||||
|
clients[client.Id] = client
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func RemoveClientFromCache(client *Client) {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
for _, groupIn := range client.Groups {
|
||||||
|
if clients, ok := ClientsMap[groupIn.Id]; ok {
|
||||||
|
delete(clients, client.Id)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,21 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "github.com/coder/websocket"
|
||||||
|
|
||||||
|
type Client struct {
|
||||||
|
Password string
|
||||||
|
Name string
|
||||||
|
Pronouns string
|
||||||
|
Groups [12]*ChatGroup
|
||||||
|
Connection *websocket.Conn
|
||||||
|
Id uint32
|
||||||
|
Color [3]byte
|
||||||
|
IsAuthenticated bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type ChatGroup struct {
|
||||||
|
Name string
|
||||||
|
Members [32]*Client
|
||||||
|
Id uint32
|
||||||
|
Color [3]byte
|
||||||
|
}
|
||||||
+3
-1
@@ -38,5 +38,7 @@ func ServeConnection(responseWriter http.ResponseWriter, request *http.Request)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleUnauthenticatedMessage() {}
|
func handleUnauthenticatedMessage() {
|
||||||
|
|
||||||
|
}
|
||||||
func handleAuthenticatedMessage() {}
|
func handleAuthenticatedMessage() {}
|
||||||
|
|||||||
Reference in New Issue
Block a user