提高PSRtpSender对象的线程安全性

src/Rtp/PSRtpSender.cpp

@@ -29,35 +29,37 @@ PSRtpSender::~PSRtpSender() {
     InfoL << this << " " << printSSRC(_rtp_encoder->getSsrc());
 }
 
-void PSRtpSender::onPS(uint32_t stamp, void *packet, size_t bytes) {
-    //此函数在其他线程执行
-    _rtp_encoder->inputFrame(std::make_shared<FrameFromPtr>((char *) packet, bytes, stamp));
-}
-
 void PSRtpSender::startSend(const string &dst_url, uint16_t dst_port, bool is_udp, const function<void(const SockException &ex)> &cb){
     _is_udp = is_udp;
-    //确保Socket对象的线程安全
+    _socket = std::make_shared<Socket>(_poller, false);
-    _socket = std::make_shared<Socket>(_poller, true);
     _dst_url = dst_url;
     _dst_port = dst_port;
     weak_ptr<PSRtpSender> weak_self = shared_from_this();
     if (is_udp) {
         _socket->bindUdpSock(0);
-        WorkThreadPool::Instance().getPoller()->async([cb, dst_url, dst_port, weak_self]() {
+        auto poller = _poller;
-            //切换线程目的是为了dns解析放在后台线程执行
+        WorkThreadPool::Instance().getPoller()->async([cb, dst_url, dst_port, weak_self, poller]() {
             struct sockaddr addr;
+            //切换线程目的是为了dns解析放在后台线程执行
             if (!SockUtil::getDomainIP(dst_url.data(), dst_port, addr)) {
+                poller->async([dst_url, cb]() {
+                    //切回自己的线程
                     cb(SockException(Err_dns, StrPrinter << "dns解析域名失败:" << dst_url));
+                });
                 return;
             }
+
+            //dns解析成功
+            poller->async([addr, weak_self, cb]() {
+                //切回自己的线程
                 cb(SockException());
                 auto strong_self = weak_self.lock();
                 if (strong_self) {
-                //dns解析成功
                     strong_self->_socket->setSendPeerAddr(&addr);
                     strong_self->onConnect();
                 }
             });
+        });
     } else {
         _socket->connect(dst_url, dst_port, [cb, weak_self](const SockException &err) {
             cb(err);
@@ -90,29 +92,47 @@ void PSRtpSender::onConnect(){
     InfoL << "开始发送 ps rtp:" << _socket->get_peer_ip() << ":" << _socket->get_peer_port() << ", 是否为udp方式:" << _is_udp;
 }
 
-void PSRtpSender::onRtp(const RtpPacket::Ptr &rtp, bool) {
 //此函数在其他线程执行
-    if(!_is_connect){
+void PSRtpSender::inputFrame(const Frame::Ptr &frame) {
-        return;
+    if (_is_connect) {
+        //连接成功后才做实质操作(节省cpu资源)
+        PSEncoder::inputFrame(frame);
+    }
 }
 
+//此函数在其他线程执行
+void PSRtpSender::onPS(uint32_t stamp, void *packet, size_t bytes) {
+    _rtp_encoder->inputFrame(std::make_shared<FrameFromPtr>((char *) packet, bytes, stamp));
+}
+
+//此函数在其他线程执行
+void PSRtpSender::onRtp(const RtpPacket::Ptr &rtp, bool) {
     //开启合并写提高发送性能
     PacketCache<RtpPacket>::inputPacket(true, rtp, false);
 }
 
-void PSRtpSender::onFlush(shared_ptr<List<RtpPacket::Ptr>> &rtp_list, bool key_pos) {
 //此函数在其他线程执行
+void PSRtpSender::onFlush(shared_ptr<List<RtpPacket::Ptr>> &rtp_list, bool) {
+    if(!_is_connect){
+        //连接成功后才能发送数据
+        return;
+    }
+
+    auto is_udp = _is_udp;
+    auto socket = _socket;
+    _poller->async([rtp_list, is_udp, socket]() {
         int i = 0;
         int size = rtp_list->size();
         rtp_list->for_each([&](const RtpPacket::Ptr &packet) {
-        if (_is_udp) {
+            if (is_udp) {
                 //udp模式，rtp over tcp前4个字节可以忽略
-            _socket->send(std::make_shared<BufferRtp>(packet, 4), nullptr, 0, ++i == size);
+                socket->send(std::make_shared<BufferRtp>(packet, 4), nullptr, 0, ++i == size);
             } else {
                 //tcp模式, rtp over tcp前2个字节可以忽略,只保留后续rtp长度的2个字节
-            _socket->send(std::make_shared<BufferRtp>(packet, 2), nullptr, 0, ++i == size);
+                socket->send(std::make_shared<BufferRtp>(packet, 2), nullptr, 0, ++i == size);
             }
         });
+    });
 }
 
 void PSRtpSender::onErr(const SockException &ex, bool is_connect) {
@@ -143,4 +163,3 @@ void PSRtpSender::onErr(const SockException &ex, bool is_connect) {
 }
 
 }//namespace mediakit
-

src/Rtp/PSRtpSender.h

@@ -54,6 +54,11 @@ public:
      */
     void startSend(const string &dst_url, uint16_t dst_port, bool is_udp, const function<void(const SockException &ex)> &cb);
 
+    /**
+     * 输入帧数据
+     */
+    void inputFrame(const Frame::Ptr &frame) override;
+
 protected:
     //mpeg-ps回调
     void onPS(uint32_t stamp, void *packet, size_t bytes) override;
@@ -65,7 +70,6 @@ protected:
      */
     void onFlush(std::shared_ptr<List<RtpPacket::Ptr> > &rtp_list, bool key_pos) override;
 
-
 private:
     //rtp打包后回调
     void onRtp(const RtpPacket::Ptr &in, bool is_key);