Solution 2: Working with TCP/IP and WebSocket

package main

import (
	"bufio"
	"fmt"
	"math/rand"
	"net"
	"strconv"
	"strings"
	"time"
)

func handleConnection(conn net.Conn) {
    defer conn.Close()
    r := bufio.NewReader(conn)
    for {
        fmt.Fprint(conn, ">> Enter a number or 'quit' to quit \n")

        line, err := r.ReadString('\n')
        if err != nil {
            break
        }
        line = strings.TrimSpace(line)
        if line == "quit" {
            break
        }
        n, err := strconv.Atoi(line)
        if err != nil {
            fmt.Println("Invalid number")
            continue
        }
        fmt.Println("Generating random number in range 0 to", n)
        rand.Seed(time.Now().UnixNano())
        fmt.Fprintln(conn, "Random number generated:", rand.Intn(n))
    }
}


func main() {
	ln, err := net.Listen("tcp", ":1234")
	if err != nil {
		panic(err)
	}
	defer ln.Close()
	for {
		conn, err := ln.Accept()
		if err != nil {
			panic(err)
		}
		go handleConnection(conn)
	}
}