为什么ChatGPT选择了SSE，而不是WebSocket？

我在探索ChatGPT的使用过程中，发现了一个有趣的现象：ChatGPT在实现流式返回的时候，选择了SSE（Server-Sent Events），而非WebSocket。

那么问题来了：为什么ChatGPT选择了SSE，而不是WebSocket呢。

SSE是什么？

SSE，全称Server-Sent Events，译为服务器发送事件，它是一种Web技术，允许服务器端实时地向客户端推送信息。SSE运行在HTTP协议之上，它利用持久化的HTTP连接，以事件流（Event Stream）的形式将数据发送给客户端，由客户端监听后获取。服务器端会定期向这个连接发送更新，这些更新被封装在一个或多个HTTP包中，每个包含有更新的事件流。这样，当有新的更新时，服务器就不需要等待客户端的请求，而是直接将数据推送给客户端。当连接被关闭或出现故障时，客户端会自动重新发送请求，重新建立连接。这确保了数据传输的连续性和实时性。

那么，SSE有什么优点呢?

单向通信：SSE只支持从服务器到客户端的单向通信，服务器可以主动发送数据，用户只能接收。
高效实时：因使用持久化连接，服务器可以实时地将数据推送给客户端，而无需客户端频繁发起请求。

什么是WebSocket？

WebSocket是一种网络通信协议，它最早被提出来是为了解决HTTP连接的一大限制：HTTP协议中，一个客户端发送给服务端的请求必须由服务端返回一个响应，这使得服务端无法主动向客户端推送数据。WebSocket的通信过程如下：

客户端通过发送一个特殊的HTTP请求向服务器请求建立WebSocket连接。这个请求类似于：GET /chat HTTP/1.1 Upgrade: websocket Connection: Upgrade
服务器响应这个请求，确认建立WebSocket连接。这个响应类似于：HTTP/1.1 101 Switching Protocols Upgrade: websocket Connection: Upgrade
之后，客户端和服务器就可以通过这个常开的连接自由地发送或接收消息。

SSE与WebSocket的比较

你可能疑问，为什么不直接使用WebSocket，它似乎更为通用，也同样支持实时数据推送。这就是我们需要对比两者的理由。

通信模式：SSE只支持服务器向客户端的单向通信，而WebSocket支持全双工通信，即服务器和客户端可以互相发送数据。对于ChatGPT这样的应用来说，大多数情况下，用户的请求是稀疏的，而服务器的响应是密集的，因此，SSE的单向通信模式更为合适。
网络协议：SSE运行在HTTP协议上，因此，它可以提供更高的兼容性和灵活性。举个例子，如果你的产品已经部署在Web服务器上，那么你大概率无需做任何改动，就可以使用SSE技术。而WebSocket则需要单独的服务器和端口。

除此之外，SSE和WebSocket在消息大小、连接数量、跨域支持等方面都有一些细微的差别，我们在具体设计时需要根据实际需求和制约因素做出选择。

使用Golang和React实践SSE

首先，我们需要创建一个Golang服务器。这个服务器将监听8000端口，等待客户端的SSE连接请求，并定时向连接发送消息。为了简单起见，这里我们假设服务器每秒产生一条新消息。以下是Golang服务器的代码：

package main

import ("fmt""log""net/http""time"
)

func main() {http.HandleFunc("/events", func(w http.ResponseWriter, r *http.Request) {w.Header().Set("Content-Type", "text/event-stream")w.Header().Set("Connection", "keep-alive")w.Header().Set("Cache-Control", "no-cache")w.Header().Set("Access-Control-Allow-Origin", "*")
for {fmt.Fprintf(w, "data: Message at %s\n\n", time.Now())if flusher, ok := w.(http.Flusher); ok {flusher.Flush()} else {log.Println("Streaming unsupported!")return}time.Sleep(time.Second)}})http.Handle("/", http.FileServer(http.Dir("static")))log.Println("Listening on localhost:8000")http.ListenAndServe("localhost:8000", nil)
}

然后，我们需要创建一个React客户端，这个客户端会建立与Golang服务器的SSE连接，并在收到新消息时更新页面。以下是React客户端的代码：

import React, { useEffect, useState } from 'react';

function App() {const [events, setEvents] = useState([]);
useEffect(() => {const eventSource = new EventSource('http://localhost:8000/events');eventSource.onmessage = (event) => {setEvents((prevState) => [...prevState, event.data]);};}, []);
return (<div className="App"><h1>Live updates from server</h1>{events.map((event, i) => <p key={i}>{event}</p>)}</div>);
}
export default App;

这个应用程序将从服务器接收一个每秒更新一次的实时数据流，并在客户端将这些更新显示出来。