mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-12-02 00:12:33 +08:00
109 lines
3.4 KiB
C++
109 lines
3.4 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT 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.
|
||
*/
|
||
|
||
#ifndef SRC_RTMP_RtmpPlayer_H_
|
||
#define SRC_RTMP_RtmpPlayer_H_
|
||
|
||
#include <memory>
|
||
#include <string>
|
||
#include <functional>
|
||
#include "amf.h"
|
||
#include "Rtmp.h"
|
||
#include "RtmpProtocol.h"
|
||
#include "Player/PlayerBase.h"
|
||
#include "Util/TimeTicker.h"
|
||
#include "Network/Socket.h"
|
||
#include "Network/TcpClient.h"
|
||
|
||
namespace mediakit {
|
||
|
||
//实现了rtmp播放器协议部分的功能，及数据接收功能
|
||
class RtmpPlayer : public PlayerBase, public toolkit::TcpClient, public RtmpProtocol {
|
||
public:
|
||
using Ptr = std::shared_ptr<RtmpPlayer>;
|
||
RtmpPlayer(const toolkit::EventPoller::Ptr &poller);
|
||
~RtmpPlayer() override;
|
||
|
||
void play(const std::string &strUrl) override;
|
||
void pause(bool bPause) override;
|
||
void speed(float speed) override;
|
||
void teardown() override;
|
||
|
||
protected:
|
||
virtual bool onMetadata(const AMFValue &val) = 0;
|
||
virtual void onRtmpPacket(RtmpPacket::Ptr chunk_data) = 0;
|
||
uint32_t getProgressMilliSecond() const;
|
||
void seekToMilliSecond(uint32_t ms);
|
||
|
||
protected:
|
||
void onMediaData_l(RtmpPacket::Ptr chunk_data);
|
||
//在获取config帧后才触发onPlayResult_l(而不是收到play命令回复)，所以此时所有track都初始化完毕了
|
||
void onPlayResult_l(const toolkit::SockException &ex, bool handshake_done);
|
||
|
||
//form Tcpclient
|
||
void onRecv(const toolkit::Buffer::Ptr &buf) override;
|
||
void onConnect(const toolkit::SockException &err) override;
|
||
void onError(const toolkit::SockException &ex) override;
|
||
//from RtmpProtocol
|
||
void onRtmpChunk(RtmpPacket::Ptr chunk_data) override;
|
||
void onStreamDry(uint32_t stream_index) override;
|
||
void onSendRawData(toolkit::Buffer::Ptr buffer) override {
|
||
send(std::move(buffer));
|
||
}
|
||
|
||
template<typename FUNC>
|
||
void addOnResultCB(const FUNC &func) {
|
||
_map_on_result.emplace(_send_req_id, func);
|
||
}
|
||
template<typename FUNC>
|
||
void addOnStatusCB(const FUNC &func) {
|
||
_deque_on_status.emplace_back(func);
|
||
}
|
||
|
||
void onCmd_result(AMFDecoder &dec);
|
||
void onCmd_onStatus(AMFDecoder &dec);
|
||
void onCmd_onMetaData(AMFDecoder &dec);
|
||
|
||
void send_connect();
|
||
void send_createStream();
|
||
void send_play();
|
||
void send_pause(bool pause);
|
||
|
||
private:
|
||
std::string _app;
|
||
std::string _stream_id;
|
||
std::string _tc_url;
|
||
|
||
bool _paused = false;
|
||
bool _metadata_got = false;
|
||
//是否为性能测试模式
|
||
bool _benchmark_mode = false;
|
||
|
||
//播放进度控制
|
||
uint32_t _seek_ms = 0;
|
||
uint32_t _fist_stamp[2] = {0, 0};
|
||
uint32_t _now_stamp[2] = {0, 0};
|
||
toolkit::Ticker _now_stamp_ticker[2];
|
||
std::deque<std::function<void(AMFValue &dec)> > _deque_on_status;
|
||
std::unordered_map<int, std::function<void(AMFDecoder &dec)> > _map_on_result;
|
||
|
||
//rtmp接收超时计时器
|
||
toolkit::Ticker _rtmp_recv_ticker;
|
||
//心跳发送定时器
|
||
std::shared_ptr<toolkit::Timer> _beat_timer;
|
||
//播放超时定时器
|
||
std::shared_ptr<toolkit::Timer> _play_timer;
|
||
//rtmp接收超时定时器
|
||
std::shared_ptr<toolkit::Timer> _rtmp_recv_timer;
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
#endif /* SRC_RTMP_RtmpPlayer_H_ */
|