LiveSource: Refine fetch for external exposed interface.

ossrs · Jan 12, 2022 · 2f41775 · 2f41775
1 parent c30aa47
commit 2f41775
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 19 deletions.
diff --git a/trunk/src/app/srs_app_rtc_source.cpp b/trunk/src/app/srs_app_rtc_source.cpp
@@ -256,6 +256,8 @@ srs_error_t SrsRtcSourceManager::fetch_or_create(SrsRequest* r, SrsRtcSource** p
 
     SrsRtcSource* source = NULL;
     if ((source = fetch(r)) != NULL) {
+        // update requst
+        source->update_auth(r);
         *pps = source;
         return err;
     }
@@ -285,17 +287,10 @@ SrsRtcSource* SrsRtcSourceManager::fetch(SrsRequest* r)
     SrsRtcSource* source = NULL;
 
     string stream_url = r->get_stream_url();
-    if (pool.find(stream_url) == pool.end()) {
-        return NULL;
+    if (pool.find(stream_url) != pool.end()) {
+        source = pool[stream_url];
     }
 
-    source = pool[stream_url];
-
-    // we always update the request of resource,
-    // for origin auth is on, the token in request maybe invalid,
-    // and we only need to update the token of request, it's simple.
-    source->update_auth(r);
-
     return source;
 }
 

diff --git a/trunk/src/app/srs_app_source.cpp b/trunk/src/app/srs_app_source.cpp
@@ -1713,6 +1713,8 @@ srs_error_t SrsLiveSourceManager::fetch_or_create(SrsRequest* r, ISrsLiveSourceH
 
     SrsLiveSource* source = NULL;
     if ((source = fetch(r)) != NULL) {
+        // update request
+        source->update_auth(r);
         *pps = source;
         return err;
     }
@@ -1745,17 +1747,10 @@ SrsLiveSource* SrsLiveSourceManager::fetch(SrsRequest* r)
     SrsLiveSource* source = NULL;
 
     string stream_url = r->get_stream_url();
-    if (pool.find(stream_url) == pool.end()) {
-        return NULL;
+    if (pool.find(stream_url) != pool.end()) {
+        source = pool[stream_url];
     }
-
-    source = pool[stream_url];
-
-    // we always update the request of resource,
-    // for origin auth is on, the token in request maybe invalid,
-    // and we only need to update the token of request, it's simple.
-    source->update_auth(r);
-
+
     return source;
 }