Xây dựng ứng dụng chat cực kỳ đơn giản với golang và websockets

func runServer() {
	app := iris.New()

   //Khai báo instance websocket, server và các client sẽ kết nối đến namespace v1, và cùng bắt sự kiện "chat"
	var ws = neffos.New(websocket.DefaultGorillaUpgrader, neffos.Namespaces{
		"v1": neffos.Events{
			"chat": serverReceived,
		},
	})

	app.Get("/websocket_endpoint", websocket.Handler(ws))
	log.Println("Serving websockets on http://localhost:8080/websocket_endpoint")
	log.Fatal(app.Listen(":8080"))
}

//Khi server tiếp nhận message tại event "chat" sẽ truyền phát đến các client thuộc cùng 1 namespace, 1 room
func serverReceived(c *neffos.NSConn, msg neffos.Message) error {
	c.Conn.Server().Broadcast(nil, neffos.Message{
		Namespace: msg.Namespace,
		Room:      msg.Room,
		Event:     msg.Event,
		Body:      msg.Body,
	})
	return nil
}

//khởi chạy client, xác định room để join, tự đặt tên và có thể chọn màu để highlight trên console
func runClient(room, full_name, colors string) {
	ctx := context.Background()
	client, err := neffos.Dial(ctx, gorilla.DefaultDialer, "ws://localhost:8080/websocket_endpoint", neffos.Namespaces{
		"v1": neffos.Events{
			"chat": clientReceive,
		},
	})
	if err != nil {
		panic(err)
	}

	//Khởi tạo instance client kết nối đến namespace cụ thể, ở đây là v1
	c, err := client.Connect(ctx, "v1")
	if err != nil {
		panic(err)
	}
	//Khởi tạo instance client join room
	r, err := c.JoinRoom(ctx, room)
	if err != nil {
		panic(err)
	}
	fmt.Println("Hello, I'm " + color.Colorize(colors, full_name) + "!")

out:
	for {
		select {
		//Trường hợp mất kết nối websocket
		case <-client.NotifyClose:
			log.Println(color.Colorize(color.Red, "Websocket connection closed"))
			break out
		//Trường hợp kết nối socket thành công
		default:
			//Nhập message
			fmt.Print("Enter message: ")
			scanner := bufio.NewScanner(os.Stdin)
			scanner.Scan() // use `for scanner.Scan()` to keep reading
			msg := scanner.Text()
			cursor.ClearLinesUp(1)
   
			//Client bắn message và gửi đến cho server tại room và namespace xác định và tại event "chat"
			r.Emit("chat", []byte(color.Colorize(colors, full_name + ": ") + msg))
			break
		}
	}

}

//Client thực hiện lệnh khi nhận được tín hiệu từ server tại event "chat"
func clientReceive(c *neffos.NSConn, msg neffos.Message) error {
	fmt.Print("\r")
	log.Println(string(msg.Body))
	fmt.Print("Enter message: ")
	return nil
}

//Khởi tạo server và client
	switch side {
	case "server":
		runServer()
	//client1 và client2 tại room1
	case "client1":
		runClient("room1", "client1", color.Yellow)
	case "client2":
		runClient("room1", "client2", color.Blue)
	//client3 và client4 tại room2
	case "client3":
		runClient("room2", "client3", color.Green)
	case "client4":
		runClient("room2", "client4", color.Cyan)
  }