Advanced Techniques in Go Programming/

...

/

The Implementation of the Server

The Implementation of the Server

Let’s learn how to implement the WebSocket server with Go.

We'll cover the following...

This lesson presents the implementation of the WebSocket server that implements the echo service, which can be really handy when testing network connections.

The GitHub repository used for keeping the code can be found here.

Note: The GitHub repository contains a Dockerfile file for producing a Docker image from the WebSocket server source file.

Coding example

The implementation of the WebSocket server can be found in ws.go, which contains the next code:

Press + to interact
package main
import (
	"fmt"
	"log"
	"net/http"
	"os"
	"time"
	"github.com/gorilla/websocket"
)

This is the external package used for working with the WebSocket protocol.

Press + to interact
var PORT = ":1234"
var upgrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

This is where the parameters of websocket.Upgrader are defined. They are going to be used shortly.

Press + to interact
func rootHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "Welcome!\n")
	fmt.Fprintf(w, "Please use /ws for WebSocket!")
}

This is a regular HTTP handler function.

Press + to interact
func wsHandler(w http.ResponseWriter, r *http.Request) {
	log.Println("Connection from:", r.Host)
	ws, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("upgrader.Upgrade:", err)
		return
	}
	defer ws.Close()
...