Merge pull request #146 from Shopify/enhance_throttle_for_ns1

Add API rate limit for NS1 provider and bump up the version to 5.7.1

CHANGELOG.md

@@ -1,4 +1,7 @@
 # CHANGELOG
+## 5.7.1
+- add API rate limit for NS1 provider [FEATURE]
+
 ## 5.7.0
 - add OCI library as runtime dependency [FEATURE]
 

lib/record_store/provider/ns1.rb

@@ -1,4 +1,5 @@
 require_relative 'ns1/client'
+require_relative 'ns1/patch_api_header'
 
 module RecordStore
   class Provider::NS1 < Provider

lib/record_store/provider/ns1/patch_api_header.rb

@@ -0,0 +1,21 @@
+require 'net/http'
+
+# Patch the method which retrieves headers for API rate limit dynamically
+module NS1::Transport
+  class NetHttp
+    def process_response(response)
+      sleep(response.to_hash["x-ratelimit-period"].first.to_i /
+        [1, response.to_hash["x-ratelimit-remaining"].first.to_i].max.to_f)
+
+      body = JSON.parse(response.body)
+      case response
+      when Net::HTTPOK
+        NS1::Response::Success.new(body, response.code.to_i)
+      else
+        NS1::Response::Error.new(body, response.code.to_i)
+      end
+    rescue JSON::ParserError
+      raise NS1::Transport::ResponseParseError
+    end
+  end
+end

lib/record_store/version.rb

@@ -1,3 +1,3 @@
 module RecordStore
-  VERSION = '5.7.0'.freeze
+  VERSION = '5.7.1'.freeze
 end