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

✨ Implement websocket gateway

Browse Source
This commit is contained in:
LittleSheep 2024-11-02 13:09:52 +08:00
parent 6bb8eab3fc
commit ea150608b4
1 changed files with 34 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,11 +1,16 @@
package ws package ws
import ( import (
"context"
"fmt"
"git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
"git.solsynth.dev/hypernet/nexus/pkg/nex" "git.solsynth.dev/hypernet/nexus/pkg/nex"
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec" "git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
"git.solsynth.dev/hypernet/nexus/pkg/proto"
"github.com/gofiber/contrib/websocket" "github.com/gofiber/contrib/websocket"
jsoniter "github.com/json-iterator/go" jsoniter "github.com/json-iterator/go"
"github.com/rs/zerolog/log" "github.com/rs/zerolog/log"
"github.com/samber/lo"
"github.com/spf13/viper" "github.com/spf13/viper"
) )
@ -42,38 +47,36 @@ func Listen(c *websocket.Conn) {
packet.Endpoint = val packet.Endpoint = val
} }
/* service := directory.GetServiceInstanceByType(packet.Endpoint)
service := directory.GetServiceInstanceByType(packet.Endpoint) if service == nil {
if service == nil { _ = c.WriteMessage(mt, nex.WebSocketPackage{
_ = c.WriteMessage(mt, nex.NetworkPackage{ Action: "error",
Action: "error", Message: "service not found",
Message: "service not found", }.Marshal())
}.Marshal()) continue
continue }
} pc, err := service.GetGrpcConn()
pc, err := service.GetGrpcConn() if err != nil {
if err != nil { _ = c.WriteMessage(mt, nex.WebSocketPackage{
_ = c.WriteMessage(mt, nex.NetworkPackage{ Action: "error",
Action: "error", Message: fmt.Sprintf("unable to connect to service: %v", err.Error()),
Message: fmt.Sprintf("unable to connect to service: %v", err.Error()), }.Marshal())
}.Marshal()) continue
continue }
}
sc := proto.NewStreamServiceClient(pc) sc := proto.NewStreamServiceClient(pc)
_, err = sc.EmitStreamEvent(context.Background(), &proto.StreamEventRequest{ _, err = sc.PushStream(context.Background(), &proto.PushStreamRequest{
Event: packet.Action, UserId: lo.ToPtr(uint64(user.ID)),
UserId: uint64(user.ID), ClientId: lo.ToPtr(clientId),
ClientId: uint64(clientId), Body: packet.Marshal(),
Payload: packet.RawPayload(), })
}) if err != nil {
if err != nil { _ = c.WriteMessage(mt, nex.WebSocketPackage{
_ = c.WriteMessage(mt, nex.NetworkPackage{ Action: "error",
Action: "error", Message: fmt.Sprintf("unable send message to service: %v", err.Error()),
Message: fmt.Sprintf("unable send message to service: %v", err.Error()), }.Marshal())
}.Marshal()) continue
continue }
}*/
} }
// Pop connection // Pop connection