just1689/entity-sync

package esweb

import (
    "github.com/gorilla/mux"
    "github.com/gorilla/websocket"
    "github.com/just1689/entity-sync/es/shared"
    "log"
    "net/http"
    "time"
)

const (
    // Time allowed to write a message to the peer.
    writeWait = 10 * time.Second
    // Time allowed to read the next pong message from the peer.
    pongWait = 60 * time.Second
    // Send pings to peer with this period. Must be less than pongWait.
    pingPeriod = (pongWait * 9) / 10
    // Maximum message size allowed from peer.
    maxMessageSize = 512
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func SetupMuxBridge(mux *mux.Router, bridgeClientBuilder shared.ByteSecretHandlingRemoteProxy, msgPassThrough shared.SecretByteHandler) {
    itemHub := newHub(bridgeClientBuilder, msgPassThrough)
    go itemHub.run()
    mux.HandleFunc("/ws/entity-sync/", func(w http.ResponseWriter, r *http.Request) {
        serveWs(itemHub, w, r)
    })

}

// serveWs handles websocket requests from the peer.
func serveWs(hub *hub, w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }

    c := &client{
        hub:  hub,
        conn: conn,
        send: make(chan []byte, 256),
        bridgeProxy: bridgeProxy{
            entityKeyHandlers: make(map[shared.Action]shared.EntityKeyHandler),
        },
        MsgPassThrough: hub.MsgPassThrough,
    }
    c.hub.register <- c

    c.bridgeProxy.entityKeyHandlers[shared.ActionSubscribe],
        c.bridgeProxy.entityKeyHandlers[shared.ActionUnSubscribe],
        c.bridgeProxy.queueDCNotify =
        hub.bridgeClientBuilder(
            func(barr []byte) {
                c.send <- barr
            }, func() string {
                return c.secret
            })

    go c.writePump()
    go c.readPump()
}