TerraOceanPlugin/GoBot/server/server.go

48 lines
825 B
Go

2019-10-12 14:07:40 +07:00
package server
import (
"fmt"
"log"
"net/http"
"strconv"
"strings"
2019-10-12 14:07:40 +07:00
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return strings.Contains(r.Host, "localhost")
},
2019-10-12 14:07:40 +07:00
}
func newConnection(w http.ResponseWriter, r *http.Request) {
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
fmt.Fprintf(w, "An error has occured: %s.\n", err)
return
2019-10-12 14:07:40 +07:00
}
defer ws.Close()
newSession(ws)
for {
_, p, err := ws.ReadMessage()
if err != nil {
disconnectSession(ws)
break
}
msg := string(p)
handleIncomingMessage(ws, msg)
}
}
2019-10-12 15:49:28 +07:00
func StartServer(port int) {
2019-10-12 14:07:40 +07:00
hostLocation := ":" + strconv.Itoa(port)
http.HandleFunc("/ws", newConnection)
log.Fatal(http.ListenAndServe(hostLocation, nil))
}