EasyDarwin自动停止推流

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

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

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

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

void    ReflectorSession::RemoveOutput(ReflectorOutput* inOutput, Bool16 isClient)
{
    (void)atomic_sub(&fNumOutputs, 1);
    for (UInt32 y = 0; y < fSourceInfo->GetNumStreams(); y++)
    {
        fStreamArray[y]->RemoveOutput(inOutput);
        if (isClient)
            fStreamArray[y]->DecEyeCount();
    }
}

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

void    ReflectorSession::RemoveOutput(ReflectorOutput* inOutput, Bool16 isClient)
{
    (void)atomic_sub(&fNumOutputs, 1);
    for (UInt32 y = 0; y < fSourceInfo->GetNumStreams(); y++)
    {
        fStreamArray[y]->RemoveOutput(inOutput);
        if (isClient)
            fStreamArray[y]->DecEyeCount();
    }
 //移除客户端之后判断fNumOutputs是否为0,add
 if(fNumOutputs == 0)
 {
  //调用角色，停止推流
  qtss_printf("客户端都不看啦
");
  QTSS_RoleParams theParams;
  theParams.easyStreamStopParams.inSerial = fSerial;
  theParams.easyStreamStopParams.inChannel= fChannel;
  QTSSModule* theModule = QTSServerInterface::GetModule(QTSSModule::kStreamStopRole, 0);
  (void)theModule->CallDispatch(Easy_StreamStop_Role, &theParams);
 }
 //
}

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

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

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

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

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