Golang development: using Websocket to implement real-time chat application-Golang-php.cn

Home

Backend Development

Golang

Golang development: using Websocket to implement real-time chat application

WBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWB

Sep 20, 2023 pm 01:15 PM

golangwebsocketLive chat application

Golang development: using Websocket to implement real-time chat application

Golang development: Real-time chat application using Websocket

In today's Internet era, real-time communication has become an indispensable part of people's lives. Whether it is instant messaging, real-time chat or real-time updates, an efficient and stable communication method is required to achieve it. Websocket is a protocol that is very suitable for real-time communication.

Golang is a simple and efficient programming language with excellent concurrency performance and is very suitable for developing real-time communication applications. This article will introduce how to use Golang's Websocket library to implement a real-time chat application and provide corresponding code examples.

First, we need to create a Golang project and introduce Golang's Websocket library.

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

Next, we need to define some global variables and structures.

var clients = make(map[*websocket.Conn]bool) // 客户端集合
var broadcast = make(chan Message)           // 广播消息通道

// 消息结构体
type Message struct {
    Username string `json:"username"`
    Message  string `json:"message"`
}

Then, we need to implement a function that handles Websocket connections.

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为Websocket连接
    upgrader := websocket.Upgrader{}
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }

    // 将连接添加到客户端集合中
    clients[conn] = true

    // 关闭连接时，从客户端集合中移除
    defer func() {
        delete(clients, conn)
        conn.Close()
    }()

    // 循环接收和发送消息
    for {
        var message Message
        err := conn.ReadJSON(&message)
        if err != nil {
            log.Printf("error: %v", err)
            delete(clients, conn)
            break
        }
        broadcast <- message
    }
}

Next, we need to implement a function that broadcasts the message to all clients.

func broadcastMessages() {
    for {
        // 从广播通道中接收消息
        message := <-broadcast

        // 将消息发送给所有客户端
        for client := range clients {
            err := client.WriteJSON(message)
            if err != nil {
                log.Printf("error: %v", err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

Finally, we need to start the Websocket server in the main function.

func main() {
    // 设置静态文件目录
    fs := http.FileServer(http.Dir("public"))
    http.Handle("/", fs)

    // 设置Websocket路由
    http.HandleFunc("/ws", handleWebSocket)

    // 启动Websocket服务器
    go broadcastMessages()

    // 开始监听端口
    log.Println("Server started on :8080")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal(err)
    }
}

Next, we can create an HTML page to implement the user interface.

<!DOCTYPE html>
<html>

<head>
  <meta charset="UTF-8">
  <title>实时聊天应用</title>
</head>

<body>
  <h1 id="实时聊天应用">实时聊天应用</h1>
  
  <div id="message-container"></div>
  
  <form>
    <input type="text" id="username" placeholder="请输入用户名">
    <input type="text" id="message" placeholder="请输入消息">
    <button type="button" onclick="sendMessage()">发送</button>
  </form>

  <script>
    // 创建Websocket连接
    var socket = new WebSocket("ws://localhost:8080/ws");
    
    // 监听连接事件
    socket.onopen = function() {
      console.log("连接成功");
    };
    
    // 监听消息事件
    socket.onmessage = function(event) {
      var message = JSON.parse(event.data);
      var username = message.username;
      var content = message.message;
      var container = document.getElementById("message-container");
      container.innerHTML += "<p><b>" + username + "：</b>" + content + "</p>";
    };
    
    // 监听错误事件
    socket.onerror = function(error) {
      console.log("出现错误：" + error);
    }
    
    // 监听关闭事件
    socket.onclose = function(event) {
      console.log("连接关闭");
    }
    
    // 发送消息
    function sendMessage() {
      var username = document.getElementById("username").value;
      var message = document.getElementById("message").value;
      if (username && message) {
        var data = {
          username: username,
          message: message
        };
        socket.send(JSON.stringify(data));
        document.getElementById("message").value = "";
      }
    }
  </script>
</body>

</html>

So far, we have completed an example of using Golang's Websocket library to implement a real-time chat application. Through this example, we can deeply understand the working principle of Websocket and learn how to use Golang to develop real-time communication applications.

Of course, there are many details and security considerations in actual projects, such as verifying messages, using SSL encryption, etc. But with this example, you already have a good starting point for building your own live chat application. Happy coding!

The above is the detailed content of Golang development: using Websocket to implement real-time chat application. For more information, please follow other related articles on the PHP Chinese website!

Statement

The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

go语言有没有缩进Dec 01, 2022 pm 06:54 PM

go语言有缩进。在go语言中，缩进直接使用gofmt工具格式化即可（gofmt使用tab进行缩进）；gofmt工具会以标准样式的缩进和垂直对齐方式对源代码进行格式化，甚至必要情况下注释也会重新格式化。

聊聊Golang中的几种常用基本数据类型Jun 30, 2022 am 11:34 AM

本篇文章带大家了解一下golang 的几种常用的基本数据类型，如整型，浮点型，字符，字符串，布尔型等，并介绍了一些常用的类型转换操作。

go语言为什么叫goNov 28, 2022 pm 06:19 PM

go语言叫go的原因：想表达这门语言的运行速度、开发速度、学习速度（develop）都像gopher一样快。gopher是一种生活在加拿大的小动物，go的吉祥物就是这个小动物，它的中文名叫做囊地鼠，它们最大的特点就是挖洞速度特别快，当然可能不止是挖洞啦。

一文详解Go中的并发【20 张动图演示】Sep 08, 2022 am 10:48 AM

Go语言中各种并发模式看起来是怎样的？下面本篇文章就通过20 张动图为你演示 Go 并发，希望对大家有所帮助！

tidb是go语言么Dec 02, 2022 pm 06:24 PM

是，TiDB采用go语言编写。TiDB是一个分布式NewSQL数据库；它支持水平弹性扩展、ACID事务、标准SQL、MySQL语法和MySQL协议，具有数据强一致的高可用特性。TiDB架构中的PD储存了集群的元信息，如key在哪个TiKV节点；PD还负责集群的负载均衡以及数据分片等。PD通过内嵌etcd来支持数据分布和容错；PD采用go语言编写。