Initial commit

data/server_manager.go

@@ -0,0 +1,54 @@
+package data
+
+import (
+	"github.com/ParadoxPixel/ThemePark-Websocket/objects"
+	"sync"
+)
+
+var serverMux sync.RWMutex
+var servers = make(map[string]*objects.Server)
+
+func AddServer(server *objects.Server) {
+	serverMux.Lock()
+	if _,ok := servers[server.ID]; !ok {
+		servers[server.ID] = server
+	}
+	serverMux.Unlock()
+}
+
+func CanServer(server string) bool {
+	if server == "" || server == "null" {
+		return false
+	}
+
+	if ok, _ := HasServer(server); ok {
+		return false
+	} else {
+		return true
+	}
+}
+
+func GetServer(server string) *objects.Server {
+	serverMux.Lock()
+	val,ok := servers[server]
+	serverMux.Unlock()
+	if ok {
+		return val
+	} else {
+		return nil
+	}
+}
+
+func HasServer(publicKey string) (bool, *objects.Server) {
+	serverMux.Lock()
+	val,ok := servers[publicKey]
+	serverMux.Unlock()
+
+	return ok, val
+}
+
+func RemoveServer(session string) {
+	serverMux.Lock()
+	delete(servers, session)
+	serverMux.Unlock()
+}

data/session_manager.go

@@ -0,0 +1,45 @@
+package data
+
+import (
+	"github.com/ParadoxPixel/ThemePark-Websocket/objects"
+	"sync"
+)
+
+var sessionMux sync.RWMutex
+var sessions = make(map[string]*objects.Session)
+
+func AddSession(session *objects.Session) {
+	sessionMux.Lock()
+	sessions[session.Token] = session
+	sessionMux.Unlock()
+}
+
+func GetSession(session string) *objects.Session {
+	sessionMux.Lock()
+	val := sessions[session]
+	sessionMux.Unlock()
+
+	return val
+}
+
+func HasSession(session string) bool {
+	sessionMux.Lock()
+	_,ok := sessions[session]
+	sessionMux.Unlock()
+
+	return ok
+}
+
+func CanSession(session string) bool {
+	if session == "" || session == "null" {
+		return false
+	}
+
+	return !HasSession(session)
+}
+
+func RemoveSession(session string) {
+	sessionMux.Lock()
+	delete(sessions, session)
+	sessionMux.Unlock()
+}