我在 golang 中使用 gin-framework 創建了一個 Web 服務。在這個專案中，我還使用了一些來自特定主題的 kafka 訊息。我想要實作的是將我從主題收到的訊息倒入 websocket。所以通信只是一種方式，超過1個人可以連接到網路套接字并看到進來的訊息。我很難使用通道，所以在接收kafka訊息的函式里面我有這樣的東西：
ch <- KafkaMessage
在杜松子酒框架中，我創建了這樣的東西：
requestRouterRPCv1.GET("wf-live", wsWorkFlowLive)
func wsWorkFlowLive(c *gin.Context) {
ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
if err != nil {
log.Println("error get connection")
log.Fatal(err)
}
defer ws.Close()
err = ws.WriteJSON(<-ch)
if err != nil {
log.Println("error write message: " err.Error())
}
}
var upGrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}
這里是我用來測驗 websocket 的 html 片段：
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8" />
<title>index</title>
</head>
<body>
<h1>test websocket</h1>
<p id="message-json"></p>
<p id="message-text"></p>
<script>
function jsonWS() {
var ws = new WebSocket("ws://ws.acme.com/ws/v1/wf-live");
ws.onmessage = function (evt) {
console.log("Received Message: " evt.data);
document.getElementById("message-json").innerText = evt.data;
};
ws.onclose = function (evt) {
console.log("Connection closed.");
};
}
// Start websocket
jsonWS();
</script>
</body>
</html>
但是我想念一些東西，我是一個新手，因為一旦收到第一條 kafka 訊息，我就會出現以下錯誤行為：
- 僅顯示第一條訊息，在該連接快速關閉后
- 要查看第二個，我必須重繪 頁面，這不是 websocket 方式
- 因為連接關閉了通道它不是紅色的，所以它卡在 cosume 功能中，直到我讀到它。我不能有這種行為
- 為了避免第 3 點，我認為我必須有一種機制，僅當連接一個或多個 ws 時才將訊息發送到通道。
uj5u.com熱心網友回復：
Gorilla 聊天示例非常接近您的需要。該示例將從任何客戶端接收到的訊息廣播到所有連接的客戶端。執行以下操作以使代碼適應您的用例：
更改客戶端讀取泵以丟棄接收到的訊息，而不是將訊息發送到集線器。
func (c *Client) readPump() { defer func() { c.hub.unregister <- c c.conn.Close() }() c.conn.SetReadLimit(maxMessageSize) c.conn.SetReadDeadline(time.Now().Add(pongWait)) c.conn.SetPongHandler(func(string) error { c.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil }) for { if _, _, err := c.NextReader(); err != nil { break } } }更改您的 Kafka 讀取回圈以將訊息發送到Hub.broadcast頻道。
for { msg, err := c.ReadMessage(xxx) if err != nil { // handle error } hub.broadcast <- msg }洗掉將客戶端發送佇列中的訊息合并為單個 websocket 訊息的代碼，或調整客戶端以在單個 websocket 訊息中處理多個 Kafka 訊息。
uj5u.com熱心網友回復：
你很親密。只是少了兩件事。
1- 使用select陳述句不斷接收來自 kafka 頻道的新訊息。
2- 保持一個活躍的 websocket 連接。這個答案有更多細節
讓我知道這是否適合你。
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/447713.html