基於live555開源流媒體服務器開發的EasyRTSPServer實現主動關閉指定通道session的方法

我們在基於live555研發音視頻流媒體服務器時，在某些特定場景下, 需要主動關閉某一通道對應的所有連接，即要立即close所有觀看當前視頻通道的所有session；

解決方案

1.主動關閉通道API

Easy_API Easy_I32 Easy_APICALL EasyRtspServer_ResetChannel(EASY_CHANNEL_HANDLE channelHandle);

//在API函數中, 僅置標誌位爲復位(0x01)

int LiveRtspServer::ResetChannel(RTSP_CHANNEL_HANDLE channelHandle)
{
    if (channelHandle < 0)            return RTSP_SERVER_ERR_BadArgument;
    if (NULL == pLiveChannel)        return RTSP_SERVER_ERR_NotInitialized;


    LockMutex(&mutexChannel);        //Lock

    LIVE_CHANNEL_OBJ_T    *pStreamChannel = NULL;
    LIVE_CHANNEL_OBJ_T    *pChannel = pLiveChannel;
    while (NULL != pChannel)
    {
        if (pChannel == channelHandle)
        {
            pStreamChannel = pChannel;
            break;
        }

        pChannel = pChannel->next;
    }

    if (NULL == pStreamChannel)
    {
        UnlockMutex(&mutexChannel);        //Unlock
        return RTSP_SERVER_ERR_OPERATE;
    }


    ServerMediaSession* sms = RTSPServer::lookupServerMediaSession(NULL, 0, NULL, pStreamChannel->streamName);
    if (NULL != sms)
    {
        pStreamChannel->resetFlag = 0x01;
    }
    else
    {
        memset(&pStreamChannel->mediainfo, 0x00, sizeof(RTSP_MEDIA_INFO_T));
    }

    UnlockMutex(&mutexChannel);        //Unlock

    return RTSP_SERVER_ERR_NoErr;
}

2.修改工作線程

BasicTaskScheduler0 *pTaskScheduler0 = NULL;

if (NULL != pThreadTask->pSubEnv && (pTaskScheduler0 = (BasicTaskScheduler0 *)&pThreadTask->pSubEnv->taskScheduler()))
{
    if (pThreadTask->pSubEnv->IsResetChannel() > 0x00)                //以下代碼爲主動關閉當前通道及所對應的客戶端
    {
        if (pThreadTask->pSubEnv->IsResetChannel() == 0x01)
        {
            ServerMediaSession* sms = (ServerMediaSession*)pThreadTask->pSubEnv->GetTmpServerMediaSession();
            char streamName[512] = {0};
            if (NULL != sms)
            {
                strcpy(streamName, sms->streamName());

                UsageEnvironment *pEnv = &sms->envir();

                pThreadTask->pSubEnv->ResetChannel(0x02, NULL);

                if (NULL != pEnv)
                {
                    pThis->removeServerMediaSession(pEnv, streamName, False);
                }

                pThis->closeAllClientSessionsForServerMediaSession(sms);
                pThis->CloseAllConnections(streamName);
            }


            pThreadTask->pSubEnv->ResetChannel(0x00, NULL);
            pTaskScheduler0->ClearDelayQueue();            //清空隊列
        }

        Sleep(10);
        continue;
    }

    int ret = pTaskScheduler0->SingleStep(0, pThreadTask->id, pThreadTask->pSubEnv->GetEnvirName());
    if (ret < 0)
    {
        break;
    }