How can i implement a websocket using httprouter

[dbhys4github]

As the title：How can i implement a websocket using httprouter？

[han-so1omon]

I got it working with something like the code below. I'm using gorilla/websocket for websocket connection management and messaging.

//////////////////////////// server.go
package main

import (
	"github.com/gorilla/websocket"
	"github.com/julienschmidt/httprouter"
	"log"
	"net/http"
)

var upgrader = websocket.Upgrader{}

func internalError(ws *websocket.Conn, msg string, err error) {
	log.Println(msg, err)
	ws.WriteMessage(websocket.TextMessage, []byte("Internal server error."))
}

func sendStuff(ws *websocket.Conn, /* stuff */) {
	ws.WriteMessage(websocket.TextMessage, []byte(/* stuff */)
}

func Connect(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	// Setup websocket
	ws, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("upgrade:", err)
		return
	}
        
        // Do some preprocessing here...

	defer ws.Close()

	// Do some more processing if you want...
	done := false
	for {
		select {
		case /* <-appDone */:
			done = true
		case /* <-appUpdated */:
			go sendStuff(ws, /*stuff*/)
		}
		if done {
			break
		}
	}
}
func NewRouter() *httprouter.Router {
	router := httprouter.New()
	router.GET("/ws", Connect)

	return router
}

//////////////////////////// main.go
package main

import (
	"log"
	"net/http"
)

func main() {
	log.Println("Starting app")
	// Get new router
	router := NewRouter()
	// Serve routes
	http.ListenAndServe(":8900", router)
}

Here's my client skeleton code as well in Python.

import os
import sys
import websocket

try:
    import thread
except ImportError:
    import _thread as thread

import time

def process_message(message):
    print(message)

def on_message(ws, message):
    process_message(message)

def on_error(ws, error):
    print("error")
    print(error)

def on_close(ws):
    print("### end ###")

def on_open(ws):
    def run(*args):
        while True:
            time.sleep(1)
        time.sleep(1)
        ws.close()
        print("terminating websocket...")
    thread.start_new_thread(run, ())

if __name__ == "__main__":
    websocket.enableTrace(True)
    ws = websocket.WebSocketApp(
            "ws://localhost:8900/ws",
            on_message = on_message,
            on_error = on_error,
            on_close = on_close)
    ws.on_open = on_open
    ws.run_forever()

This confused me a bit as well, as I wasn't sure which http method to use with a websocket. I settled on the GET method, as that's what I've seen in some other places. Everything else as shown, substituting POST instead of GET yields 405 Method Not Allowed.

[han-so1omon]

@julienschmidt I recommend closing this issue and labeling as question