HyperNet/Nexus
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Provide external apis for ability of auth

Browse Source
This commit is contained in:
LittleSheep
2024-10-22 23:23:58 +08:00
parent 0f4112e08b
commit a1412acc90
7 changed files with 58 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/internal/http/ws/connections.go
View File

@@ -2,10 +2,10 @@ package ws
import (
"git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
"math/rand"
"sync"
"git.solsynth.dev/hypernet/nexus/pkg/internal/models"
"github.com/gofiber/contrib/websocket"
)
@@ -14,7 +14,7 @@ var (
wsConn = make(map[uint]map[uint64]*websocket.Conn)
)
func ClientRegister(user models.Account, conn *websocket.Conn) uint64 {
func ClientRegister(user sec.UserInfo, conn *websocket.Conn) uint64 {
wsMutex.Lock()
if wsConn[user.ID] == nil {
wsConn[user.ID] = make(map[uint64]*websocket.Conn)
@@ -31,7 +31,7 @@ func ClientRegister(user models.Account, conn *websocket.Conn) uint64 {
return clientId
}
func ClientUnregister(user models.Account, id uint64) {
func ClientUnregister(user sec.UserInfo, id uint64) {
wsMutex.Lock()
if wsConn[user.ID] == nil {
wsConn[user.ID] = make(map[uint64]*websocket.Conn)

4
pkg/internal/http/ws/ws.go
View File

@@ -1,8 +1,8 @@
package ws
import (
"git.solsynth.dev/hypernet/nexus/pkg/internal/models"
"git.solsynth.dev/hypernet/nexus/pkg/nex"
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
"github.com/gofiber/contrib/websocket"
jsoniter "github.com/json-iterator/go"
"github.com/rs/zerolog/log"
@@ -10,7 +10,7 @@ import (
)
func Listen(c *websocket.Conn) {
user := c.Locals("user").(models.Account)
user := c.Locals("nex_user").(sec.UserInfo)
// Push connection
clientId := ClientRegister(user, c)