构建一个即时消息应用(五):实时消息

开发 后端
对于实时消息,我们将使用 服务器发送事件。这是一个打开的连接,我们可以在其中传输数据流。我们会有个端点,用户会在其中订阅发送给他的所有消息。

[[345206]]

本文是该系列的第五篇。

对于实时消息,我们将使用 服务器发送事件Server-Sent Events。这是一个打开的连接,我们可以在其中传输数据流。我们会有个端点,用户会在其中订阅发送给他的所有消息。

消息户端

在 HTTP 部分之前,让我们先编写一个映射map ,让所有客户端都监听消息。 像这样全局初始化:

  1. type MessageClient struct { 
  2.     Messages chan Message 
  3.     UserID   string 
  4.  
  5. var messageClients sync.Map 

已创建的新消息

还记得在 上一篇文章 中,当我们创建这条消息时,我们留下了一个 “TODO” 注释。在那里,我们将使用这个函数来调度一个 goroutine。

  1. go messageCreated(message) 

把这行代码插入到我们留注释的位置。

  1. func messageCreated(message Message) error { 
  2.     if err := db.QueryRow(` 
  3.         SELECT user_id FROM participants 
  4.         WHERE user_id != $1 and conversation_id = $2 
  5.     `, message.UserID, message.ConversationID). 
  6.     Scan(&message.ReceiverID); err != nil { 
  7.         return err 
  8.     } 
  9.  
  10.     go broadcastMessage(message) 
  11.  
  12.     return nil 
  13.  
  14. func broadcastMessage(message Message) { 
  15.     messageClients.Range(func(key, _ interface{}) bool { 
  16.         client := key.(*MessageClient) 
  17.         if client.UserID == message.ReceiverID { 
  18.             client.Messages <- message 
  19.         } 
  20.         return true 
  21.     }) 

该函数查询接收者 ID(其他参与者 ID),并将消息发送给所有客户端。

订阅消息

让我们转到 main() 函数并添加以下路由:

  1. router.HandleFunc("GET""/api/messages", guard(subscribeToMessages)) 

此端点处理 /api/messages 上的 GET 请求。请求应该是一个 EventSource 连接。它用一个事件流响应,其中的数据是 JSON 格式的。

  1. func subscribeToMessages(w http.ResponseWriter, r *http.Request) { 
  2.     if a := r.Header.Get("Accept"); !strings.Contains(a, "text/event-stream") { 
  3.         http.Error(w, "This endpoint requires an EventSource connection", http.StatusNotAcceptable) 
  4.         return 
  5.     } 
  6.  
  7.     f, ok := w.(http.Flusher) 
  8.     if !ok { 
  9.         respondError(w, errors.New("streaming unsupported")) 
  10.         return 
  11.     } 
  12.  
  13.     ctx := r.Context() 
  14.     authUserID := ctx.Value(keyAuthUserID).(string) 
  15.  
  16.     h := w.Header() 
  17.     h.Set("Cache-Control""no-cache"
  18.     h.Set("Connection""keep-alive"
  19.     h.Set("Content-Type""text/event-stream"
  20.  
  21.     messages := make(chan Message) 
  22.     defer close(messages) 
  23.  
  24.     client := &MessageClient{Messages: messages, UserID: authUserID} 
  25.     messageClients.Store(client, nil) 
  26.     defer messageClients.Delete(client) 
  27.  
  28.     for { 
  29.         select { 
  30.         case <-ctx.Done(): 
  31.             return 
  32.         case message := <-messages: 
  33.             if b, err := json.Marshal(message); err != nil { 
  34.                 log.Printf("could not marshall message: %v\n", err) 
  35.                 fmt.Fprintf(w, "event: error\ndata: %v\n\n", err) 
  36.             } else { 
  37.                 fmt.Fprintf(w, "data: %s\n\n", b) 
  38.             } 
  39.             f.Flush() 
  40.         } 
  41.     } 

首先,它检查请求头是否正确,并检查服务器是否支持流式传输。我们创建一个消息通道,用它来构建一个客户端,并将其存储在客户端映射中。每当创建新消息时,它都会进入这个通道,因此我们可以通过 for-select 循环从中读取。

服务器发送事件Server-Sent Events使用以下格式发送数据:

  1. data: some data here\n\n 

我们以 JSON 格式发送:

  1. data: {"foo":"bar"}\n\n 

我们使用 fmt.Fprintf() 以这种格式写入响应写入器writter,并在循环的每次迭代中刷新数据。

这个循环会一直运行,直到使用请求上下文关闭连接为止。我们延迟了通道的关闭和客户端的删除,因此,当循环结束时,通道将被关闭,客户端不会收到更多的消息。

注意,服务器发送事件Server-Sent Events(EventSource)的 JavaScript API 不支持设置自定义请求头😒,所以我们不能设置 Authorization: Bearer <token>。这就是为什么 guard() 中间件也会从 URL 查询字符串中读取令牌的原因。


实时消息部分到此结束。我想说的是,这就是后端的全部内容。但是为了编写前端代码,我将再增加一个登录端点:一个仅用于开发的登录。

 

 

责任编辑:庞桂玉 来源: Linux中国
相关推荐

2020-10-09 12:45:19

创建消息即时消息编程语言

2019-09-29 15:25:13

CockroachDBGoJavaScript

2020-03-31 12:21:20

JSON即时消息编程语言

2019-10-28 20:12:40

OAuthGuard中间件编程语言

2020-10-16 14:40:20

即时消息Home页面编程语言

2020-10-12 09:20:13

即时消息Access页面编程语言

2020-10-19 16:20:38

即时消息Conversatio编程语言

2020-10-10 20:51:10

即时消息编程语言

2023-08-14 08:01:12

websocket8g用户

2021-03-25 08:29:33

SpringBootWebSocket即时消息

2015-03-18 15:37:19

社交APP场景

2011-10-19 09:30:23

jQuery

2024-02-01 12:54:00

RustWebSocket消息代理

2023-11-30 08:34:29

批量消息消息队列

2023-03-27 08:33:32

2024-04-24 11:42:21

Redis延迟消息数据库

2021-05-10 15:05:18

消息通信本地网络

2021-12-03 00:02:01

通讯工具即时

2010-05-24 09:51:37

System Cent

2022-08-30 11:41:53

网络攻击木马
点赞
收藏

51CTO技术栈公众号