dns_one 0.4.29 → 0.4.30

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8ad077556ec78e89b5d4f8a88827dc77bf8397e8
-  data.tar.gz: 8b7916cd71ad40672f8fca7a5baf41b13129c2a8
+  metadata.gz: 5bd90da7f26c043c1061527a8d6c317b323a1d63
+  data.tar.gz: 594f84fea313f5092620752731e463bb5b40562c
 SHA512:
-  metadata.gz: cc2559f45f107f09ffcf56579890d057104ca176e8c27e9737a793ef754ac2b0a732dd1a84fe324e4b6d8e663c6d5e32426961b631fe92db24233acf9e81fb9a
-  data.tar.gz: 51fe6c09207f7026008300afade6e95675ea93b79d7d2b7506eb1bdba85eee7ad82372ef4edc5b7a82ecb14507a4326c62920a86a8118bf8aba51042a5a14879
+  metadata.gz: aafba801a40a381693c2fb347668dae193323084ce98af5fc6e424f65ba2a50d37e9f8cba27174970a765ec0d67e4ecdb07a6ecbf3c799f2b7287219cd2c82b3
+  data.tar.gz: d1f868608bd2267eb2451aee3af246f56f2b958a7d15cbe7773eaf2a6bf412892ce621033cd23560daf554c13b99fda7889d30b646b761ccead708287e964f69

data/lib/dns_one/cli.rb

@@ -1,8 +1,9 @@
-require "thor"
 
+require "thor"
+require 'sqlite3'
 require "dns_one"
 require "dns_one/setup"
-require 'sqlite3'
+require "dns_one/stat"
 
 class DnsOne::CLI < Thor  
 

data/lib/dns_one/stat.rb

@@ -0,0 +1,142 @@
+module DnsOne; class Stat
+
+    def initialize conf = {}
+        @conf = conf
+        ensure_db
+    end
+
+    def save rcode, req_resource, cache
+        @db.execute(
+            "INSERT INTO responses (time, rcode, req_resource, cache) VALUES (?, ?, ?, ?)", 
+            [
+                Time.now.to_i, 
+                Resolv::DNS::RCode.const_get(rcode), 
+                req_resource::TypeValue, 
+                (cache ? 1 : 0)
+            ]
+        )
+    rescue => e
+        Log.e e
+    end
+
+    def get_counts counter, from = nil
+        validate_counter counter
+        validate_from from
+
+        from ||= (Time.now - 5 * 60).to_i
+
+        s = <<-SQL
+            select #{counter}, count(*) 
+            from responses 
+            where time > #{from}
+            group by #{counter}
+        SQL
+
+        counts = {}
+
+        @db.execute(s) do |row|
+            _counter, count = row
+            counts[_counter] = count
+        end
+
+        counts
+    end
+
+    private
+
+    def validate_counter counter
+        unless [:rcode, :req_resource, :cache].include? counter
+            raise "invalid arg #{counter}"
+        end
+    end
+
+    def validate_from from
+        case from
+        when nil, Integer
+        when String
+            unless from =~ /^\d+$/
+                raise "invalid arg #{from}"
+            end
+        else
+            raise  "invalid arg class #{from.class}"
+        end
+    end
+
+    def ensure_db
+        new_db = !File.exists?(db_file)
+        @db = SQLite3::Database.new db_file
+        create_tables if new_db
+    end
+    
+    def create_tables
+        sqls = []
+
+        sqls << <<-SQL
+            create table responses (
+                time int,
+                rcode int,
+                req_resource int,
+                cache int
+            );
+        SQL
+        
+        sqls <<  <<-SQL
+            CREATE INDEX responses_time         ON responses (time);
+        SQL
+
+        sqls <<  <<-SQL
+            CREATE INDEX responses_rcode        ON responses (rcode);
+        SQL
+
+        sqls <<  <<-SQL
+            CREATE INDEX responses_req_resource ON responses (req_resource);
+        SQL
+
+        sqls <<  <<-SQL
+            CREATE INDEX responses_cache        ON responses (cache);
+        SQL
+
+        sqls.each{|sql| @db.execute sql }
+    end
+
+
+    def db_file
+        @conf[:db_file]
+    end
+
+    def request_resources
+        unless defined? @@request_resources
+            @@request_resources = {}
+            %w(A AAAA ANY CNAME HINFO MINFO MX NS PTR SOA TXT WKS).each do |res|
+                val = Object.const_get("Resolv::DNS::Resource::IN::#{res}")::TypeValue
+                @@request_resources[val] = res.downcase
+            end
+        end
+        @@request_resources
+    end
+
+    def const_underscore name
+        name = name.to_s.dup
+        name.gsub!('::', '/')
+        name.gsub!(/([A-Z\d]+)([A-Z][a-z])/,'\1_\2')
+        name.gsub!(/([a-z\d])([A-Z])/,'\1_\2')
+        name.tr!("-", "_")
+        name.downcase!
+        name
+    end
+
+    def rcodes
+        unless @@rcodes
+            @@rcodes = Hash[ 
+                Resolv::DNS::RCode.constants.map{|c| 
+                    [
+                        Resolv::DNS::RCode.const_get(c), 
+                        const_underscore(c)
+                    ] 
+                } 
+            ]
+        end
+        @@rcodes
+    end
+end; end
+

data/lib/dns_one/version.rb

@@ -1,3 +1,3 @@
 module DnsOne
-    VERSION = "0.4.29"
+    VERSION = "0.4.30"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dns_one
 version: !ruby/object:Gem::Version
-  version: 0.4.29
+  version: 0.4.30
 platform: ruby
 authors:
 - Tom Lobato
@@ -171,6 +171,7 @@ files:
 - lib/dns_one/log.rb
 - lib/dns_one/server.rb
 - lib/dns_one/setup.rb
+- lib/dns_one/stat.rb
 - lib/dns_one/util.rb
 - lib/dns_one/version.rb
 - lib/dns_one/zone_search.rb