diff options
Diffstat (limited to 'internal/server')
| -rw-r--r-- | internal/server/api/api.go | 4 | ||||
| -rw-r--r-- | internal/server/server.go | 7 |
2 files changed, 8 insertions, 3 deletions
diff --git a/internal/server/api/api.go b/internal/server/api/api.go index 4dfa172..df29448 100644 --- a/internal/server/api/api.go +++ b/internal/server/api/api.go @@ -669,7 +669,7 @@ func SetUserData(ctx context.Context, sess *session.Session, request *packet.Set return &ErrInternalError } - return &packet.GetUserData{ + return &packet.SetUserData{ Data: request.Data, } } @@ -683,7 +683,7 @@ func GetUserData(ctx context.Context, sess *session.Session, request *packet.Get return &ErrInternalError } - return &packet.GetUserData{ + return &packet.SetUserData{ Data: data, } } diff --git a/internal/server/server.go b/internal/server/server.go index 9abf51f..327eaa8 100644 --- a/internal/server/server.go +++ b/internal/server/server.go @@ -241,13 +241,18 @@ func (server *server) handleConnection(conn net.Conn) { }() // Send initial packets - payload, err := api.GetNetworksInfo(ctx, sess) + payload := api.GetUserData(ctx, sess, &packet.GetUserData{}) + dataPacket := packet.NewPacket(packet.NewMsgPackEncoder(payload)) + sess.Write(ctx, dataPacket) + + payload, err = api.GetNetworksInfo(ctx, sess) if err != nil { return // closes the connection } infoPacket := packet.NewPacket(packet.NewMsgPackEncoder(payload)) sess.Write(ctx, infoPacket) + // Infinite read loop buffer := make([]byte, 512) for { n, err := conn.Read(buffer) |
