custom error

und3f · und3f · commit 41555cd6d9a2 · 2016-01-17T16:04:56.000+02:00
diff --git a/lib/checkhost/client.rb b/lib/checkhost/client.rb
@@ -4,6 +4,19 @@
 require 'json'
 
 module CheckHost
+    class CheckHostError < RuntimeError
+        @@errors = {
+            "limit_exceeded" => "Limit of checks is exceeded. Try again in a 10 minutes."
+        }
+        def describe_error
+            puts @@errors.inspect()
+            if (@@errors.has_key?(message)) then
+                return @@errors[message]
+            end
+            return message
+        end
+    end
+
     class Client
         attr_reader :nodes
         
@@ -84,7 +97,7 @@ def query(action, data) # action: newcheck, results; data(hash): newcheck:: type
         
         def parse(json)
             res = JSON.parse(json)
-            raise "#{res['error']}" if res.has_key?('error')
+            raise CheckHostError.new(res['error']) if res.has_key?('error')
             return res
         end
 
