Allow disabling persistent connections

Author: drbrain

README.md

@@ -770,6 +770,7 @@ found in `lib/influxdb/config.rb` for the source of truth.
 |                 | `:open_timeout`         | 5             | socket timeout
 |                 | `:read_timeout`         | 300           | socket timeout
 |                 | `:auth_method`          | "params"      | "params", "basic_auth" or "none"
+|                 | `:persistent`           | true          | set to false to disable persistent connections
 | Retry           | `:retry`                | -1            | max. number of retry attempts (reading and writing)
 |                 | `:initial_delay`        | 0.01          | initial wait time (doubles every retry attempt)
 |                 | `:max_delay`            | 30            | max. wait time when retrying

lib/influxdb/client/http.rb

@@ -136,6 +136,8 @@ def generate_cert_store
     end
 
     def get_http
+      return build_http config.next_host unless config.persistent
+
       @https ||=
         begin
           https = config.hosts.map { |host|

lib/influxdb/config.rb

@@ -18,6 +18,7 @@ module InfluxDB
     auth_method:          nil,
     proxy_addr:           nil,
     proxy_port:           nil,
+    persistent:           true,
 
     # SSL options
     use_ssl:              false,

spec/influxdb/client_spec.rb

@@ -60,6 +60,24 @@
     end
   end
 
+  describe "#get_http" do
+    it "returns an existing connection with persistence enabled" do
+      first  = subject.send :get_http
+      second = subject.send :get_http
+
+      expect(first).to equal(second)
+    end
+
+    it "returns a new connection with persistence disabled" do
+      subject.config.persistent = false
+
+      first  = subject.send :get_http
+      second = subject.send :get_http
+
+      expect(first).to_not equal(second)
+    end
+  end
+
   describe "GET #ping" do
     it "returns OK" do
       stub_request(:get, "http://influxdb.test:9999/ping")

spec/influxdb/config_spec.rb

@@ -14,6 +14,7 @@
     specify { expect(conf.port).to eq 8086 }
     specify { expect(conf.username).to eq "root" }
     specify { expect(conf.password).to eq "root" }
+    specify { expect(conf.persistent).to be_truthy }
     specify { expect(conf.use_ssl).to be_falsey }
     specify { expect(conf.time_precision).to eq "s" }
     specify { expect(conf.auth_method).to eq "params" }