websocket: move the server handler to ServerOptions

1 files changed, 4 insertions, 6 deletions

diff --git a/lib/websocket/examples/cmd/server/main.go b/lib/websocket/examples/cmd/server/main.go
index ad593c36..2258718f 100644
--- a/lib/websocket/examples/cmd/server/main.go
+++ b/lib/websocket/examples/cmd/server/main.go
@@ -25,16 +25,14 @@ func main() {
 
 	opts := &websocket.ServerOptions{
 		Address: ":9001",
+		// Register the authentication handler.
+		HandleAuth:         handleAuth,
+		HandleClientAdd:    handleClientAdd,
+		HandleClientRemove: handleClientRemove,
 	}
 
 	server = websocket.NewServer(opts)
 
-	// Register the authentication handler.
-	server.HandleAuth = handleAuth
-
-	server.HandleClientAdd = handleClientAdd
-	server.HandleClientRemove = handleClientRemove
-
 	// Register the message handler
 	err := server.RegisterTextHandler(http.MethodPost, "/message", handlePostMessage)
 	if err != nil {