WebSocketController

顾名思义，WebSocketController用于处理websocket逻辑。websocket是基于HTTP的一种长连接方案，在websocket建立之初，有一次HTTP格式的请求和应答交换，建立完成后，所有的消息在websocket上传输，消息由固定的格式包装，但消息的内容和收发次序没有任何要求，完全由用户定义。

可以由drogon_ctl工具快速生成WebSocketController的源文件，命令格式如下：

  1. drogon_ctl create controller -w <[namespace::]class_name>

假设我们要通过websocket实现一个简单的回声功能，即服务端只是简单的把客户端发来的消息再发回去，通过drogon_ctl创建WebSocketController的实现类EchoWebsock，如下：

  1. drogon_ctl create controller -w EchoWebsock

该命令会生成EchoWebsock.h和EchoWebsock.cc两个文件，

  1. //EchoWebsock.h
  2. #pragma once
  3. #include <drogon/WebSocketController.h>
  4. using namespace drogon;
  5. class EchoWebsock:public drogon::WebSocketController<EchoWebsock>
  6. {
  7.   public:
  8.     void handleNewMessage(const WebSocketConnectionPtr&,
  9.                           std::string &&,
  10.                           const WebSocketMessageType &) override;
  11.     void handleNewConnection(const HttpRequestPtr &,
  12.                              const WebSocketConnectionPtr&) override;
  13.     void handleConnectionClosed(const WebSocketConnectionPtr&) override;
  14.     WS_PATH_LIST_BEGIN
  15.     //list path definitions here;
  16.     WS_PATH_LIST_END
  17. };
  1. #include "EchoWebsock.h"
  2. void EchoWebsock::handleNewMessage(const WebSocketConnectionPtr &wsConnPtr,std::string &&message)
  3. {
  4.     //write your application logic here
  5. }
  6. void EchoWebsock::handleNewConnection(const HttpRequestPtr &req,const WebSocketConnectionPtr &wsConnPtr)
  7. {
  8.     //write your application logic here
  9. }
  10. void EchoWebsock::handleConnectionClosed(const WebSocketConnectionPtr &wsConnPtr)
  11. {
  12.     //write your application logic here
  13. }

编辑后内容如下：

  1. //EchoWebsock.h
  2. #pragma once
  3. #include <drogon/WebSocketController.h>
  4. using namespace drogon;
  5. class EchoWebsock:public drogon::WebSocketController<EchoWebsock>
  6. {
  7. public:
  8.     virtual void handleNewMessage(const WebSocketConnectionPtr&,
  9.                                   std::string &&,
  10.                                   const WebSocketMessageType &)override;
  11.     virtual void handleNewConnection(const HttpRequestPtr &,
  12.                                      const WebSocketConnectionPtr&)override;
  13.     virtual void handleConnectionClosed(const WebSocketConnectionPtr&)override;
  14.     WS_PATH_LIST_BEGIN
  15.     //list path definitions here;
  16.     WS_PATH_ADD("/echo");
  17.     WS_PATH_LIST_END
  18. };
  1. //EchoWebsock.cc
  2. #include "EchoWebsock.h"
  3. void EchoWebsock::handleNewMessage(const WebSocketConnectionPtr &wsConnPtr,std::string &&message)
  4. {
  5.     //write your application logic here
  6.     wsConnPtr->send(message);
  7. }
  8. void EchoWebsock::handleNewConnection(const HttpRequestPtr &req,const WebSocketConnectionPtr &wsConnPtr)
  9. {
  10.     //write your application logic here
  11. }
  12. void EchoWebsock::handleConnectionClosed(const WebSocketConnectionPtr &wsConnPtr)
  13. {
  14.     //write your application logic here
  15. }

首先，在这个例子中，通过WS_PATH_ADD宏把这个控制器注册到了/echo路径上，WS_PATH_ADD宏的用法跟之前介绍的其他控制器的宏类似，也可以注册路径并且附带若干过滤器Filter。由于websocket在框架中单独处理，所以它可以和前两种控制器的路径重复而不会相互影响。

其次，本例中三个虚函数的实现，只有handleNewMessage有实质内容，只是简单的把收到的消息通过send接口发回客户端。把这个控制器编译进框架，就可以看到效果，请各位自己试验吧。

注意：和通常的HTTP协议一样，http的websocket可以被旁路还原，如果需要安全保障，应由https提供加密功能，当然，用户自己在服务端和客户端完成加密和解密也是可以的，只是https更方便，底层都由drogon处理，用户只需关心业务逻辑。

用户自定义的WebSocketController类继承自drogon::WebSocketController类模板，模板参数是子类类型，用户需自己实现如下三个虚函数来对websocket的建立、关闭和消息进行处理：

  1. virtual void handleNewConnection(const HttpRequestPtr &req,const WebSocketConnectionPtr &wsConn);
  2. virtual void handleNewMessage(const WebSocketConnectionPtr &wsConn,std::string &&message,
  3. const WebSocketMessageType &);
  4. virtual void handleConnectionClosed(const WebSocketConnectionPtr &wsConn);

容易知道：

  • handleNewConnection在websocket建立之后被调用，req是客户端发来的建立请求，这时候框架已经返回了response，用户可以做的是通过req获得一些额外信息，比如token之类。wsConn是这个websocket对象的智能指针，常用的接口后面再谈。
  • handleNewMessage在websocket收到新的消息之后被调用，消息存储在message变量里，注意这个message是完整的消息净荷，框架已经做完了消息的解封包和解码等工作，用户直接处理消息本身即可。
  • handleConnectionClosed在websocket连接关闭之后调用，用户可以做一些收尾工作。

WebSocketConnection

WebSocketConnection对象常用接口如下：

  1. //发送websocket消息，消息的编码和封包都由框架负责，这里直接发送消息的净荷
  2. void send(const char *msg,uint64_t len);
  3. void send(const std::string &msg);
  5. //本websocket的本端和远端地址
  6. const trantor::InetAddress &localAddr() const;
  7. const trantor::InetAddress &peerAddr() const;
  9. //本weosocket的连接状态
  10. bool connected() const;
  11. bool disconnected() const;
  13. //关闭本websocket
  14. void shutdown();//close write
  15. void forceClose();//close
  17. //设置和获取本websocket的上下文，由用户存入一些业务数据，
  18. //any类型意味着可以存取任意类型的对象。
  19. void setContext(const any &context);
  20. const any &getContext() const;
  21. any *getMutableContext();

05 过滤器