Merge pull request #260 from firebase/force-reuse

FirebaseHttpClient: add forceReuse

Author: proppy

Diff for: src/FirebaseHttpClient_Esp8266.cpp

@@ -15,12 +15,32 @@
 #define USE_ESP_ARDUINO_CORE_2_0_0
 #endif
 
+// Firebase now returns `Connection: close` after REST streaming redirection.
+//
+// Override the built-in ESP8266HTTPClient to *not* close the
+// connection if forceReuse it set to `true`.
+class ForceReuseHTTPClient : public HTTPClient {
+public:
+  void end() {
+    if (_forceReuse) {
+      _canReuse = true;
+    }
+    HTTPClient::end();
+  }
+  void forceReuse(bool forceReuse) {
+    _forceReuse = forceReuse;
+  }
+protected:
+  bool _forceReuse = false;
+};
+
 class FirebaseHttpClientEsp8266 : public FirebaseHttpClient {
  public:
   FirebaseHttpClientEsp8266() {}
 
   void setReuseConnection(bool reuse) override {
     http_.setReuse(reuse);
+    http_.forceReuse(reuse);
   }
 
   void begin(const std::string& url) override {
@@ -64,7 +84,7 @@ class FirebaseHttpClientEsp8266 : public FirebaseHttpClient {
   }
 
  private:
-  HTTPClient http_;
+  ForceReuseHTTPClient http_;
 };
 
 FirebaseHttpClient* FirebaseHttpClient::create() {