external/srs
1
0
Fork 0
mirror of https://github.com/ossrs/srs.git synced 2025-02-15 04:42:04 +00:00
Code Issues Releases Wiki Activity

DVR: Copy req from publish (#2714)

Browse source 
* DVR: copy req from publish

* DVR: copy req from publish
This commit is contained in:
Haibo Chen 2021-11-07 16:59:33 +08:00 committed by GitHub
parent f4e87b17a3
commit 3974b319d1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 23 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
trunk/src/app/srs_app_dvr.cpp
View file

@ -598,8 +598,12 @@ srs_error_t SrsDvrPlan::initialize(SrsOriginHub* h, SrsDvrSegmenter* s, SrsReque
return err; return err;
} }
srs_error_t SrsDvrPlan::on_publish() srs_error_t SrsDvrPlan::on_publish(SrsRequest* r)
{ {
// @see https://github.com/ossrs/srs/issues/1613#issuecomment-960623359
srs_freep(req);
req = r->copy();
return srs_success; return srs_success;
} }
@ -687,11 +691,11 @@ SrsDvrSessionPlan::~SrsDvrSessionPlan()
{ {
} }
srs_error_t SrsDvrSessionPlan::on_publish() srs_error_t SrsDvrSessionPlan::on_publish(SrsRequest* r)
{ {
srs_error_t err = srs_success; srs_error_t err = srs_success;
if ((err = SrsDvrPlan::on_publish()) != srs_success) { if ((err = SrsDvrPlan::on_publish(r)) != srs_success) {
return err; return err;
} }
@ -762,11 +766,11 @@ srs_error_t SrsDvrSegmentPlan::initialize(SrsOriginHub* h, SrsDvrSegmenter* s, S
return srs_success; return srs_success;
} }
srs_error_t SrsDvrSegmentPlan::on_publish() srs_error_t SrsDvrSegmentPlan::on_publish(SrsRequest* r)
{ {
srs_error_t err = srs_success; srs_error_t err = srs_success;
if ((err = SrsDvrPlan::on_publish()) != srs_success) { if ((err = SrsDvrPlan::on_publish(r)) != srs_success) {
return err; return err;
} }
@ -913,13 +917,14 @@ SrsDvr::~SrsDvr()
_srs_config->unsubscribe(this); _srs_config->unsubscribe(this);
srs_freep(plan); srs_freep(plan);
srs_freep(req);
} }
srs_error_t SrsDvr::initialize(SrsOriginHub* h, SrsRequest* r) srs_error_t SrsDvr::initialize(SrsOriginHub* h, SrsRequest* r)
{ {
srs_error_t err = srs_success; srs_error_t err = srs_success;
req = r; req = r->copy();
hub = h; hub = h;
SrsConfDirective* conf = _srs_config->get_dvr_apply(r->vhost); SrsConfDirective* conf = _srs_config->get_dvr_apply(r->vhost);
@ -945,7 +950,7 @@ srs_error_t SrsDvr::initialize(SrsOriginHub* h, SrsRequest* r)
return err; return err;
} }
srs_error_t SrsDvr::on_publish() srs_error_t SrsDvr::on_publish(SrsRequest* r)
{ {
srs_error_t err = srs_success; srs_error_t err = srs_success;
@ -954,10 +959,13 @@ srs_error_t SrsDvr::on_publish()
return err; return err;
} }
if ((err = plan->on_publish()) != srs_success) { if ((err = plan->on_publish(r)) != srs_success) {
return srs_error_wrap(err, "publish"); return srs_error_wrap(err, "publish");
} }
srs_freep(req);
req = r->copy();
return err; return err;
} }

8
trunk/src/app/srs_app_dvr.hpp
View file

@ -167,7 +167,7 @@ public:
virtual ~SrsDvrPlan(); virtual ~SrsDvrPlan();
public: public:
virtual srs_error_t initialize(SrsOriginHub* h, SrsDvrSegmenter* s, SrsRequest* r); virtual srs_error_t initialize(SrsOriginHub* h, SrsDvrSegmenter* s, SrsRequest* r);
virtual srs_error_t on_publish(); virtual srs_error_t on_publish(SrsRequest* r);
virtual void on_unpublish(); virtual void on_unpublish();
virtual srs_error_t on_meta_data(SrsSharedPtrMessage* shared_metadata); virtual srs_error_t on_meta_data(SrsSharedPtrMessage* shared_metadata);
virtual srs_error_t on_audio(SrsSharedPtrMessage* shared_audio, SrsFormat* format); virtual srs_error_t on_audio(SrsSharedPtrMessage* shared_audio, SrsFormat* format);
@ -187,7 +187,7 @@ public:
SrsDvrSessionPlan(); SrsDvrSessionPlan();
virtual ~SrsDvrSessionPlan(); virtual ~SrsDvrSessionPlan();
public: public:
virtual srs_error_t on_publish(); virtual srs_error_t on_publish(SrsRequest* r);
virtual void on_unpublish(); virtual void on_unpublish();
}; };
@ -203,7 +203,7 @@ public:
virtual ~SrsDvrSegmentPlan(); virtual ~SrsDvrSegmentPlan();
public: public:
virtual srs_error_t initialize(SrsOriginHub* h, SrsDvrSegmenter* s, SrsRequest* r); virtual srs_error_t initialize(SrsOriginHub* h, SrsDvrSegmenter* s, SrsRequest* r);
virtual srs_error_t on_publish(); virtual srs_error_t on_publish(SrsRequest* r);
virtual void on_unpublish(); virtual void on_unpublish();
virtual srs_error_t on_audio(SrsSharedPtrMessage* shared_audio, SrsFormat* format); virtual srs_error_t on_audio(SrsSharedPtrMessage* shared_audio, SrsFormat* format);
virtual srs_error_t on_video(SrsSharedPtrMessage* shared_video, SrsFormat* format); virtual srs_error_t on_video(SrsSharedPtrMessage* shared_video, SrsFormat* format);
@ -237,7 +237,7 @@ public:
// publish stream event, // publish stream event,
// when encoder start to publish RTMP stream. // when encoder start to publish RTMP stream.
// @param fetch_sequence_header whether fetch sequence from source. // @param fetch_sequence_header whether fetch sequence from source.
virtual srs_error_t on_publish(); virtual srs_error_t on_publish(SrsRequest* r);
// the unpublish event., // the unpublish event.,
// when encoder stop(unpublish) to publish RTMP stream. // when encoder stop(unpublish) to publish RTMP stream.
virtual void on_unpublish(); virtual void on_unpublish();

5
trunk/src/app/srs_app_source.cpp
View file

@ -1138,7 +1138,8 @@ srs_error_t SrsOriginHub::on_publish()
return srs_error_wrap(err, "dash publish"); return srs_error_wrap(err, "dash publish");
} }
if ((err = dvr->on_publish()) != srs_success) { // @see https://github.com/ossrs/srs/issues/1613#issuecomment-961657927
if ((err = dvr->on_publish(req)) != srs_success) {
return srs_error_wrap(err, "dvr publish"); return srs_error_wrap(err, "dvr publish");
} }
@ -1401,7 +1402,7 @@ srs_error_t SrsOriginHub::on_reload_vhost_dvr(string vhost)
} }
// start to publish by new plan. // start to publish by new plan.
if ((err = dvr->on_publish()) != srs_success) { if ((err = dvr->on_publish(req)) != srs_success) {
return srs_error_wrap(err, "dvr publish failed"); return srs_error_wrap(err, "dvr publish failed");
} }