fix: handle connection error

electron · May 10, 2019 · 94f64a7 · 94f64a7
1 parent fed93cf
commit 94f64a7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/atom/browser/net/node_stream_loader.cc b/atom/browser/net/node_stream_loader.cc
@@ -23,6 +23,10 @@ NodeStreamLoader::NodeStreamLoader(network::ResourceResponseHead head,
       isolate_(isolate),
       emitter_(isolate, emitter),
       weak_factory_(this) {
+  binding_.set_connection_error_handler(
+      base::BindOnce(&NodeStreamLoader::NotifyComplete,
+                     weak_factory_.GetWeakPtr(), net::ERR_FAILED));
+
   // PostTask since it might destruct.
   base::SequencedTaskRunnerHandle::Get()->PostTask(
       FROM_HERE, base::BindOnce(&NodeStreamLoader::Start,

diff --git a/atom/browser/net/node_stream_loader.h b/atom/browser/net/node_stream_loader.h
@@ -6,6 +6,7 @@
 #define ATOM_BROWSER_NET_NODE_STREAM_LOADER_H_
 
 #include <map>
+#include <memory>
 #include <string>
 #include <vector>