SSE客户端C++实现(使用libcurl)
1、SSE数据包格式如下所示,一条SSE消息中可以有一个或多个message,每个message由\n\n分隔,一个message也可以由一个或多个filed组成,每个filed由\n分隔,filed有data、id、event、retry四种。
data表示消息数据,如"data:value\n"。
id表示数据包编号,当连接断开重连的时候,客户端应该发送一个 HTTP 头,里面包含一个特殊的Last-Event-ID头信息,值为收到的最后一条数据的id。
event表示自定义的事件类型。
retry表示浏览器重新发起连接的间隔时间,当时间间隔到期,客户端应该重新发起连接。
还可以有仅以冒号开头的行,表示注释。通常服务器每隔一段时间就会向浏览器发送一个注释,保持连接不中断。
包含一条message的消息:
data: {"username": "bobby", "time": "02:33:48"} 包含两条message的消息:
data:this is message A
data:this is
data:message B 包含三条message的消息:
:explan textdata:this is message A
data:this is
data:message B2、SSE客户端实现
#include <thread>
#include <functional>
size_t dataCallback(void* ptr, size_t size, size_t nmemb, void* param)
{
if (ptr == NULL || size == 0 || param == NULL)
return 0;
int s = size * nmemb;
std::string strMsg((char*)ptr, s);
CSSEClient* client = static_cast<CSSEClient*>(param);
if (client && client->onMessage)
client->onMessage(strMsg);
return s;
}
int progressCallback(void* param, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow) {
CSSEClient* client = static_cast<CSSEClient*>(param);
return client->isExit() ? 1 : 0;// 返回非0值会中断curl_easy_perform()
}
class CSSEClient
{
public:
CSSEClient(const std::string& strURL) {
curl_global_init(CURL_GLOBAL_DEFAULT);
_strURL = strURL;
_thread = std::thread(&CSSEClient::run, this);
}
virtual ~CSSEClient() {
_bExit = true;
_thread.join();
curl_global_cleanup();
}
bool isExit() { return _bExit; }
std::function<void(const std::string&)> onMessage;
private:
void run() {
while (!_bExit) {
subscribeSSE();
if (_bExit)
break;
std::this_thread::sleep_for(std::chrono::seconds(1));
}
}
void subscribeSSE() {
CURL* curl = curl_easy_init();
if (!curl) return;
struct curl_slist* headers = NULL;
headers = curl_slist_append(headers, "Accept: text/event-stream");
headers = curl_slist_append(headers, "Cache-Control: no-cache");
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
curl_easy_setopt(curl, CURLOPT_URL, _strURL.c_str());
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, dataCallback);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, this);
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 10L);
curl_easy_setopt(curl, CURLOPT_TIMEOUT, 0L);
curl_easy_setopt(curl, CURLOPT_TCP_KEEPALIVE, 1L); //开启TCP心跳
curl_easy_setopt(curl, CURLOPT_TCP_KEEPIDLE, 60L); //空闲60秒后探测
curl_easy_setopt(curl, CURLOPT_TCP_KEEPINTVL, 10L); //探测间隔10秒
curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0L);// 开启传输进度(已下载/上传数据量、总数据量等)回调
curl_easy_setopt(curl, CURLOPT_XFERINFOFUNCTION, progressCallback); //设置传输进度回调方法,即使没有数据传输该方法也会被调用(通常每秒1次),所以可以在该方法中判断是否需要退出
curl_easy_setopt(curl, CURLOPT_XFERINFODATA, this); //设置传输回调方法的第一个参数
CURLcode res = curl_easy_perform(curl);
if (res == CURLE_ABORTED_BY_CALLBACK) {
//SSE connection aborted by user;
}
else {
//error
}
curl_easy_cleanup(curl);
}
std::thread _thread;
std::string _strURL;
std::atomic_bool _bExit = false;
}; 比如,客户端订阅用户相关数据的变化:
CSSEClient _sseClient("http://test-api.baidu.com.cn:8080/SSE/subscribe/userID") 对于需要认证的SSE服务,需添加CURLOPT_USERPWD或OAuth头。
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
页:
[1]