ws_server.go

package main

import (
	"github.com/better-go/pkg/log"
	"github.com/better-go/pkg/net/ws"
	"net/http"
)

func main() {
	// TODO: need absolute path, change here! ! !
	homePage := "/Users/henry/Documents/iSpace/better-go/pkg/net/ws/example/ws.html"

	log.Infof("ws html file path: %v", homePage)

	s := ws.NewWebSocketServer(
		func(receivedMessage []byte) (responseMessage []byte, err error) {
			log.Infof("ws server: receive message: %v", string(receivedMessage))
			resp := string(receivedMessage) + " >>> echo from server"
			responseMessage = []byte(resp)
			return responseMessage, nil
		},
		func(apiKey string) bool {
			//return false // TODO: test ws auth
			return true
		},
	)

	// ws echo endpoint:
	http.Handle("/echo", s.DispatchWithAuth())

	// home page:
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, homePage)
	})

	log.Infof("ready to run, please check web browser: localhost:8080")

	// http server:
	http.ListenAndServe(":8080", nil)
}