Golang 实战：用 WebSocket 创建一个即时聊天应用

2023-01-20 18:54:53

WebSocket：使用 Golang 构建实时聊天应用的终极指南

简介

在现代互联网世界中，实时通信已成为网络应用程序不可或缺的一部分。无论是社交媒体、游戏还是协作平台，能够即时交换数据至关重要。WebSocket 技术应运而生，满足了这一需求。本指南将带你一步步使用 Golang 构建一个实时聊天应用程序，让你亲身体验 WebSocket 的强大功能。

第一步：安装 Golang

踏入 Golang 之旅的第一步是安装其最新版本。前往官方网站并根据你的操作系统进行下载。

第二步：创建一个 Golang 项目

接下来，使用命令行创建一个新项目：

mkdir websocket-chat
cd websocket-chat
go mod init websocket-chat

第三步：添加依赖项

为了使用 WebSocket 库，我们需要使用 Go 模块导入它。在项目根目录下的 go.mod 文件中添加以下依赖项：

require github.com/gorilla/websocket v1.2.0

第四步：编写服务器代码

现在，让我们创建 server.go 文件并编写 WebSocket 服务器端代码：

package main

import (
	"github.com/gorilla/websocket"
	"log"
	"net/http"
)

var clients = make(map[*websocket.Conn]bool)

func main() {
	http.HandleFunc("/ws", handleWS)
	log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleWS(w http.ResponseWriter, r *http.Request) {
	conn, err := websocket.Upgrade(w, r, nil, 1024, 1024)
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	clients[conn] = true
	for {
		_, msg, err := conn.ReadMessage()
		if err != nil {
			log.Println(err)
			delete(clients, conn)
			break
		}
		for client := range clients {
			if client != conn {
				client.WriteMessage(websocket.TextMessage, msg)
			}
		}
	}
}

第五步：运行服务器

准备好让服务器启动了吗？在终端中输入：

go run server.go

第六步：编写客户端代码

是时候创建客户端了。在 client.go 文件中编写以下代码：

package main

import (
	"github.com/gorilla/websocket"
	"log"
	"time"
)

func main() {
	conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	go func() {
		for {
			_, msg, err := conn.ReadMessage()
			if err != nil {
				log.Println(err)
				return
			}
			log.Printf("Received message: %s", msg)
		}
	}()

	for {
		time.Sleep(1 * time.Second)
		err := conn.WriteMessage(websocket.TextMessage, []byte("Hello from client"))
		if err != nil {
			log.Println(err)
			return
		}
	}
}