141 lines
4.9 KiB
C++
141 lines
4.9 KiB
C++
/*
|
|
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT-like license that can be found in the
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "PusherProxy.h"
|
|
|
|
using namespace toolkit;
|
|
using namespace std;
|
|
|
|
namespace mediakit {
|
|
|
|
PusherProxy::PusherProxy(const MediaSource::Ptr &src, int retry_count, const EventPoller::Ptr &poller)
|
|
: MediaPusher(src, poller) {
|
|
_retry_count = retry_count;
|
|
_on_close = [](const SockException &) {};
|
|
_weak_src = src;
|
|
_live_secs = 0;
|
|
_live_status = 1;
|
|
_republish_count = 0;
|
|
}
|
|
|
|
PusherProxy::~PusherProxy() {
|
|
_timer.reset();
|
|
}
|
|
|
|
void PusherProxy::setPushCallbackOnce(const function<void(const SockException &ex)> &cb) {
|
|
_on_publish = cb;
|
|
}
|
|
|
|
void PusherProxy::setOnClose(const function<void(const SockException &ex)> &cb) {
|
|
_on_close = cb;
|
|
}
|
|
|
|
void PusherProxy::publish(const string &dst_url) {
|
|
std::weak_ptr<PusherProxy> weak_self = shared_from_this();
|
|
std::shared_ptr<int> failed_cnt(new int(0));
|
|
|
|
setOnPublished([weak_self, dst_url, failed_cnt](const SockException &err) {
|
|
auto strong_self = weak_self.lock();
|
|
if (!strong_self) {
|
|
return;
|
|
}
|
|
|
|
if (strong_self->_on_publish) {
|
|
strong_self->_on_publish(err);
|
|
strong_self->_on_publish = nullptr;
|
|
}
|
|
|
|
auto src = strong_self->_weak_src.lock();
|
|
if (!err) {
|
|
// 推流成功 [AUTO-TRANSLATED:28ce6e56]
|
|
// Stream successfully pushed
|
|
strong_self->_live_ticker.resetTime();
|
|
strong_self->_live_status = 0;
|
|
*failed_cnt = 0;
|
|
InfoL << "Publish " << dst_url << " success";
|
|
} else if (src && (*failed_cnt < strong_self->_retry_count || strong_self->_retry_count < 0)) {
|
|
// 推流失败，延时重试推送 [AUTO-TRANSLATED:92b094ae]
|
|
// Stream failed, retry pushing with delay
|
|
strong_self->_republish_count++;
|
|
strong_self->_live_status = 1;
|
|
strong_self->rePublish(dst_url, (*failed_cnt)++);
|
|
} else {
|
|
// 如果媒体源已经注销, 或达到了最大重试次数，回调关闭 [AUTO-TRANSLATED:444adf27]
|
|
// If the media source has been deregistered, or the maximum retry count has been reached, callback to close
|
|
strong_self->_on_close(err);
|
|
}
|
|
});
|
|
|
|
setOnShutdown([weak_self, dst_url, failed_cnt](const SockException &err) {
|
|
auto strong_self = weak_self.lock();
|
|
if (!strong_self) {
|
|
return;
|
|
}
|
|
|
|
if (*failed_cnt == 0) {
|
|
// 第一次重推更新时长 [AUTO-TRANSLATED:5f778703]
|
|
// Update duration for the first re-push
|
|
strong_self->_live_secs += strong_self->_live_ticker.elapsedTime() / 1000;
|
|
strong_self->_live_ticker.resetTime();
|
|
TraceL << " live secs " << strong_self->_live_secs;
|
|
}
|
|
|
|
auto src = strong_self->_weak_src.lock();
|
|
// 推流异常中断，延时重试播放 [AUTO-TRANSLATED:e69e5a05]
|
|
// Stream abnormally interrupted, retry playing with delay
|
|
if (src && (*failed_cnt < strong_self->_retry_count || strong_self->_retry_count < 0)) {
|
|
strong_self->_republish_count++;
|
|
strong_self->rePublish(dst_url, (*failed_cnt)++);
|
|
} else {
|
|
// 如果媒体源已经注销, 或达到了最大重试次数，回调关闭 [AUTO-TRANSLATED:444adf27]
|
|
// If the media source has been deregistered, or the maximum retry count has been reached, callback to close
|
|
strong_self->_on_close(err);
|
|
}
|
|
});
|
|
|
|
MediaPusher::publish(dst_url);
|
|
}
|
|
|
|
void PusherProxy::rePublish(const string &dst_url, int failed_cnt) {
|
|
auto delay = MAX(2 * 1000, MIN(failed_cnt * 3000, 60 * 1000));
|
|
weak_ptr<PusherProxy> weak_self = shared_from_this();
|
|
_timer = std::make_shared<Timer>(
|
|
delay / 1000.0f,
|
|
[weak_self, dst_url, failed_cnt]() {
|
|
// 推流失败次数越多，则延时越长 [AUTO-TRANSLATED:bda77afe]
|
|
// The more times the stream fails, the longer the delay
|
|
auto strong_self = weak_self.lock();
|
|
if (!strong_self) {
|
|
return false;
|
|
}
|
|
WarnL << "推流重试[" << failed_cnt << "]:" << dst_url;
|
|
strong_self->MediaPusher::publish(dst_url);
|
|
return false;
|
|
},
|
|
getPoller());
|
|
}
|
|
|
|
int PusherProxy::getStatus() {
|
|
return _live_status.load();
|
|
}
|
|
uint64_t PusherProxy::getLiveSecs() {
|
|
if (_live_status == 0) {
|
|
return _live_secs + _live_ticker.elapsedTime() / 1000;
|
|
} else {
|
|
return _live_secs;
|
|
}
|
|
}
|
|
|
|
uint64_t PusherProxy::getRePublishCount() {
|
|
return _republish_count;
|
|
}
|
|
|
|
} /* namespace mediakit */
|