EasyDarwin自动停止推流

原文转自：http://blog.csdn.net/ss00_2012/article/details/51441753

我们使用EasyDarwin的推流转发来进行媒体直播的时候，有时会有这样一个需求，当推流端推送的媒体没有客户端观看时，就希望推流端不再推送,也就是实现按需推送，当推流端使用4G网络时这个需求会更加强烈。

需求明确，那么说下实现。实现分为两步：触发和通知。

触发，EasyDarwin为每一路推流转发维护一个ReflectorSession对象，该对象的fNumOutputs属性用来指示当前拉流客户端的数量。当客户端停止拉流时会调用QTSSReflectorModule::DestroySession()->QTSSReflectorModule::RemoveOutput()->ReflectorSession::RemoveOutput(),其中ReflectorSession::RemoveOutput()代码如下：

[cpp] view
plain copy

1. void    ReflectorSession::RemoveOutput(ReflectorOutput* inOutput, Bool16 isClient)
2. {
3. (void)atomic_sub(&fNumOutputs, 1);
4. for (UInt32 y = 0; y < fSourceInfo->GetNumStreams(); y++)
5. {
6. fStreamArray[y]->RemoveOutput(inOutput);
7. if (isClient)
8. fStreamArray[y]->DecEyeCount();
9. }
10. }

其中(void)atomic_sub(&fNumOutputs, 1);用来减少当前的客户端数量，我们可以在此判断fNumOutputs是否为0，为0就表示满足我们的触发条件，修改后的代码如下：

[cpp] view
plain copy

1. void    ReflectorSession::RemoveOutput(ReflectorOutput* inOutput, Bool16 isClient)
2. {
3. (void)atomic_sub(&fNumOutputs, 1);
4. for (UInt32 y = 0; y < fSourceInfo->GetNumStreams(); y++)
5. {
6. fStreamArray[y]->RemoveOutput(inOutput);
7. if (isClient)
8. fStreamArray[y]->DecEyeCount();
9. }
10. //移除客户端之后判断fNumOutputs是否为0,add
11. if(fNumOutputs == 0)
12. {
13. //调用角色，停止推流
14. qtss_printf("客户端都不看啦\n");
15. QTSS_RoleParams theParams;
16. theParams.easyStreamStopParams.inSerial = fSerial;
17. theParams.easyStreamStopParams.inChannel= fChannel;
18. QTSSModule* theModule = QTSServerInterface::GetModule(QTSSModule::kStreamStopRole, 0);
19. (void)theModule->CallDispatch(Easy_StreamStop_Role, &theParams);
20. }
21. //
22. }

触发之后我们就调用Easy_StreamStop_Role角色来通知完成停止推流。

上述触发并不完美，对于推流之后并没有客户端拉流这种情况无法处理。因此我们又增加了循环检测的机制，通过将ReflectorSession继承Task，然后在其Run函数里循环检测fNumOutputs是否为0可以有效的解决这种情况，ReflectorSession::Run()代码如下：

[cpp] view
plain copy

1. SInt64 ReflectorSession::Run()
2. {
3. if(fIfFirstRun)
4. fIfFirstRun = false;//第一次的时候还没有拉流，就不要进行处理了;客户端拉流不要过慢。
5. else
6. {
7. if(fNumOutputs == 0)
8. {
9. //调用角色，停止推流
10. qtss_printf("没有客户端观看当前转发媒体\n");
11. QTSS_RoleParams theParams;
12. theParams.easyStreamStopParams.inSerial = fSerial;
13. theParams.easyStreamStopParams.inChannel= fChannel;
14. QTSSModule* theModule = QTSServerInterface::GetModule(QTSSModule::kStreamStopRole, 0);
15. (void)theModule->CallDispatch(Easy_StreamStop_Role, &theParams);
16. }
17. }
18. return 15*1000;
19. }

另外不要忘了在ReflectorSession的构造函数中调用this->Signal(Task::kStartEvent);来开始循环检测。

通知，OK，到此触发就搞定了，下面我们就需要把停止推流这个信息告诉推流端就OK了，由谁告诉？对于推流端的管理是由EasyCMS来进行的，因此我们需要先将停止推流信息告诉EasyCMS,然后让EasyCMS告诉指定的推流端。为此我们创建了EasyCMSSession来和EasyCMS进行通讯，创建了EasyCMSModule为外面调用角色提供接口。详细代码请参考https://github.com/EasyDarwin/EasyDarwin。