瀏覽代碼

Remind #176: channel->isWriteComplete

ithewei 3 年之前
父節點
當前提交
54fdda5069
共有 2 個文件被更改,包括 12 次插入8 次删除
  1. 6 4
      evpp/TcpClient_test.cpp
  2. 6 4
      examples/websocket_server_test.cpp

+ 6 - 4
evpp/TcpClient_test.cpp

@@ -30,10 +30,12 @@ int main(int argc, char* argv[]) {
             // send(time) every 3s
             setInterval(3000, [channel](TimerID timerID){
                 if (channel->isConnected()) {
-                    char str[DATETIME_FMT_BUFLEN] = {0};
-                    datetime_t dt = datetime_now();
-                    datetime_fmt(&dt, str);
-                    channel->write(str);
+                    if (channel->isWriteComplete()) {
+                        char str[DATETIME_FMT_BUFLEN] = {0};
+                        datetime_t dt = datetime_now();
+                        datetime_fmt(&dt, str);
+                        channel->write(str);
+                    }
                 } else {
                     killTimer(timerID);
                 }

+ 6 - 4
examples/websocket_server_test.cpp

@@ -62,10 +62,12 @@ int main(int argc, char** argv) {
         MyContext* ctx = channel->newContext<MyContext>();
         // send(time) every 1s
         ctx->timerID = setInterval(1000, [channel](TimerID id) {
-            char str[DATETIME_FMT_BUFLEN] = {0};
-            datetime_t dt = datetime_now();
-            datetime_fmt(&dt, str);
-            channel->send(str);
+            if (channel->isConnected() && channel->isWriteComplete()) {
+                char str[DATETIME_FMT_BUFLEN] = {0};
+                datetime_t dt = datetime_now();
+                datetime_fmt(&dt, str);
+                channel->send(str);
+            }
         });
     };
     ws.onmessage = [](const WebSocketChannelPtr& channel, const std::string& msg) {