Merge pull request #46 from akhilrex/websockets
Akhil Gupta
4 years ago
committed by
GitHub
commit
0f802b3553
@ -0,0 +1,148 @@
|
|||||||
|
package controllers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/gorilla/websocket"
|
||||||
|
)
|
||||||
|
|
||||||
|
type EnqueuePayload struct {
|
||||||
|
ItemId string `json:"itemId"`
|
||||||
|
PodcastId string `json:"podcastId"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var wsupgrader = websocket.Upgrader{
|
||||||
|
ReadBufferSize: 1024,
|
||||||
|
WriteBufferSize: 1024,
|
||||||
|
}
|
||||||
|
|
||||||
|
var activePlayers = make(map[*websocket.Conn]string)
|
||||||
|
var allConnections = make(map[*websocket.Conn]string)
|
||||||
|
|
||||||
|
var broadcast = make(chan Message) // broadcast channel
|
||||||
|
|
||||||
|
type Message struct {
|
||||||
|
Identifier string `json:"identifier"`
|
||||||
|
MessageType string `json:"messageType"`
|
||||||
|
Payload string `json:"payload"`
|
||||||
|
Connection *websocket.Conn `json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func Wshandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
conn, err := wsupgrader.Upgrade(w, r, nil)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Failed to set websocket upgrade: %+v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
for {
|
||||||
|
var mess Message
|
||||||
|
err := conn.ReadJSON(&mess)
|
||||||
|
if err != nil {
|
||||||
|
// fmt.Println("Socket Error")
|
||||||
|
//fmt.Println(err.Error())
|
||||||
|
isPlayer := activePlayers[conn] != ""
|
||||||
|
if isPlayer {
|
||||||
|
delete(activePlayers, conn)
|
||||||
|
broadcast <- Message{
|
||||||
|
MessageType: "PlayerRemoved",
|
||||||
|
Identifier: mess.Identifier,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
delete(allConnections, conn)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
mess.Connection = conn
|
||||||
|
allConnections[conn] = mess.Identifier
|
||||||
|
broadcast <- mess
|
||||||
|
// conn.WriteJSON(mess)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func HandleWebsocketMessages() {
|
||||||
|
for {
|
||||||
|
// Grab the next message from the broadcast channel
|
||||||
|
msg := <-broadcast
|
||||||
|
//fmt.Println(msg)
|
||||||
|
|
||||||
|
switch msg.MessageType {
|
||||||
|
case "RegisterPlayer":
|
||||||
|
activePlayers[msg.Connection] = msg.Identifier
|
||||||
|
for connection, _ := range allConnections {
|
||||||
|
connection.WriteJSON(Message{
|
||||||
|
Identifier: msg.Identifier,
|
||||||
|
MessageType: "PlayerExists",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
fmt.Println("Player Registered")
|
||||||
|
case "PlayerRemoved":
|
||||||
|
for connection, _ := range allConnections {
|
||||||
|
connection.WriteJSON(Message{
|
||||||
|
Identifier: msg.Identifier,
|
||||||
|
MessageType: "NoPlayer",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
fmt.Println("Player Registered")
|
||||||
|
case "Enqueue":
|
||||||
|
var payload EnqueuePayload
|
||||||
|
fmt.Println(msg.Payload)
|
||||||
|
err := json.Unmarshal([]byte(msg.Payload), &payload)
|
||||||
|
if err == nil {
|
||||||
|
items := getItemsToPlay(payload.ItemId, payload.PodcastId)
|
||||||
|
var player *websocket.Conn
|
||||||
|
for connection, id := range activePlayers {
|
||||||
|
|
||||||
|
if msg.Identifier == id {
|
||||||
|
player = connection
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if player != nil {
|
||||||
|
payloadStr, err := json.Marshal(items)
|
||||||
|
if err == nil {
|
||||||
|
player.WriteJSON(Message{
|
||||||
|
Identifier: msg.Identifier,
|
||||||
|
MessageType: "Enqueue",
|
||||||
|
Payload: string(payloadStr),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
fmt.Println(err.Error())
|
||||||
|
}
|
||||||
|
case "Register":
|
||||||
|
var player *websocket.Conn
|
||||||
|
for connection, id := range activePlayers {
|
||||||
|
|
||||||
|
if msg.Identifier == id {
|
||||||
|
player = connection
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if player == nil {
|
||||||
|
fmt.Println("Player Not Exists")
|
||||||
|
msg.Connection.WriteJSON(Message{
|
||||||
|
Identifier: msg.Identifier,
|
||||||
|
MessageType: "NoPlayer",
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
msg.Connection.WriteJSON(Message{
|
||||||
|
Identifier: msg.Identifier,
|
||||||
|
MessageType: "PlayerExists",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Send it out to every client that is currently connected
|
||||||
|
// for client := range clients {
|
||||||
|
// err := client.WriteJSON(msg)
|
||||||
|
// if err != nil {
|
||||||
|
// log.Printf("error: %v", err)
|
||||||
|
// client.Close()
|
||||||
|
// delete(clients, client)
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue