arii 1.7 → 1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a6be30113f3d3eea18c5c4d19f14b9cabb2f1eef
-  data.tar.gz: 3b581b00830cf709707c8622c7d9cf17f2b92a1d
+  metadata.gz: 6ceec14f03110e8d331a9d7ca37d85615809acd1
+  data.tar.gz: bc1b1746618ced80dfb370a3841757c22d70f44b
 SHA512:
-  metadata.gz: 546c5905fd9d05864ad7a04861226bb2d9f00e2abdd343abe8891488a4d743c908c3bc5aafa8120fb087444e4a2b381fb688503b4e1ced61286980674c88cc87
-  data.tar.gz: 6a2cdb0c4d3c634fd76d351e506ddf7bc0e0a4aafbc1a0f8240dd3914e6d94a3b1e881e60702772d0b9f04ee5dade410d48ea0c8803881b934df03a3edbd7e7f
+  metadata.gz: c8f3437c76db16e9da58601dc1a3f7c759e51422ac60e24b133e669121918abbb07efc8f5e06cc4c6f52ae7d1eaadd7734071fcbd1a6fea50f60ebf632a2a441
+  data.tar.gz: 3a924acb65a55a0c0e6d97a17d833063bb0e334b4578cf143e91a7daf54af52a174f562f0064114c4d1f1f02d7a29a3ed42dbbde37d61c5bfe5d6fcf3f9bf8dc

data/arii.gemspec

@@ -25,4 +25,6 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency 'rubyXL'
   spec.add_runtime_dependency 'spreadsheet'
   spec.add_runtime_dependency 'jsonpath', '~> 0'
+  spec.add_runtime_dependency 'tiny_tds', '> 0.6.3.rc1'
+  spec.add_runtime_dependency 'pg'
 end

data/lib/arii/sqldetector.rb

@@ -1,5 +1,7 @@
 #require 'helper'
 require 'mysql2'
+require 'tiny_tds'
+requie 'pg'
 
 module ARII
 
@@ -17,34 +19,100 @@ module ARII
     def detect object
       ARII::Config.log.debug(self.class.name) { "Monitoring #{object[:host]}" }
       begin
-        @client = Mysql2::Client.new(:host => object[:host], :username => object[:username], :password => object[:password], :database => object[:database])
-        @client.query(object[:query]).each(:symbolize_keys => false) do |row|
-          unless object[:cache].nil? then
-            @response = Cashier.verify row[object[:cache]], object, row, object[:seed]
-          else
-            @response = Cashier.verify row["id"], object, row, object[:seed]
+        case object[:server]
+        when 'mysql'
+          @client = Mysql2::Client.new(:host => object[:host], :username => object[:username], :password => object[:password], :database => object[:database])
+          @client.query(object[:query]).each(:symbolize_keys => false) do |row|
+            unless object[:cache].nil? then
+              @response = Cashier.verify row[object[:cache]], object, row, object[:seed]
+            else
+              @response = Cashier.verify row["id"], object, row, object[:seed]
+            end
+
+            # Process ARII cache response
+            @cache = JSON.parse(@response, {:symbolize_names => true})
+            unless @cache[:templates].nil? then
+              @cache[:templates].each do |t|
+                @templates.push t
+              end
+            end
+            # The actual processing
+            #
+            if @cache[:cache][:status] == 100 then
+              ARII::Config.log.info(self.class.name) { "Not on cache, generating payload" }
+              # add row data to payload from selectors (key => key, value => column name)
+              payload = Hash.new
+              object[:selectors].each do |selector|
+                selector.each do |k, v|
+                  payload[k] = row[v]
+                end
+              end
+              # add payload object to payloads list
+              @payloads.push payload
+            end
           end
+        when 'mssql'
+          @client = TinyTds::Client.new username: object[:username], password: object[:password], host: object[:host], database: object[:database], port: object[:port], timeout: 60
+          @results = @client.execute(@agent[:payload][:query])
+          @results.each(:symbolize_keys => false) do |row|
+            unless object[:cache].nil? then
+              @response = Cashier.verify row[object[:cache]], object, row, object[:seed]
+            else
+              @response = Cashier.verify row["id"], object, row, object[:seed]
+            end
 
-          # Process ARII cache response
-          @cache = JSON.parse(@response, {:symbolize_names => true})
-          unless @cache[:templates].nil? then
-            @cache[:templates].each do |t|
-              @templates.push t
+            # Process ARII cache response
+            @cache = JSON.parse(@response, {:symbolize_names => true})
+            unless @cache[:templates].nil? then
+              @cache[:templates].each do |t|
+                @templates.push t
+              end
+            end
+            # The actual processing
+            #
+            if @cache[:cache][:status] == 100 then
+              ARII::Config.log.info(self.class.name) { "Not on cache, generating payload" }
+              # add row data to payload from selectors (key => key, value => column name)
+              payload = Hash.new
+              object[:selectors].each do |selector|
+                selector.each do |k, v|
+                  payload[k] = row[v]
+                end
+              end
+              # add payload object to payloads list
+              @payloads.push payload
             end
           end
-          # The actual processing
-          #
-          if @cache[:cache][:status] == 100 then
-            ARII::Config.log.info(self.class.name) { "Not on cache, generating payload" }
-            # add row data to payload from selectors (key => key, value => column name)
-            payload = Hash.new
-            object[:selectors].each do |selector|
-              selector.each do |k, v|
-                payload[k] = row[v]
+        when 'postgresql'
+          client = PG::Connection.new(:host => object[:host], :user => object[:username], :password => object[:password], :dbname => object[:database])
+          client.exec(object[:query]).each do |row|
+            unless object[:cache].nil? then
+              @response = Cashier.verify row[object[:cache]], object, row, object[:seed]
+            else
+              @response = Cashier.verify row["id"], object, row, object[:seed]
+            end
+
+            # Process ARII cache response
+            @cache = JSON.parse(@response, {:symbolize_names => true})
+            unless @cache[:templates].nil? then
+              @cache[:templates].each do |t|
+                @templates.push t
+              end
+            end
+            # The actual processing
+            #
+            if @cache[:cache][:status] == 100 then
+              ARII::Config.log.info(self.class.name) { "Not on cache, generating payload" }
+              # add row data to payload from selectors (key => key, value => column name)
+              payload = Hash.new
+              object[:selectors].each do |selector|
+                selector.each do |k, v|
+                  payload[k] = row[v]
+                end
               end
+              # add payload object to payloads list
+              @payloads.push payload
             end
-            # add payload object to payloads list
-            @payloads.push payload
           end
         end
       rescue Exception => e

data/lib/arii/version.rb

@@ -1,3 +1,3 @@
 module ARII
-  VERSION = "1.7"
+  VERSION = "1.8"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arii
 version: !ruby/object:Gem::Version
-  version: '1.7'
+  version: '1.8'
 platform: ruby
 authors:
 - Pedro Lopes
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-16 00:00:00.000000000 Z
+date: 2015-05-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -108,6 +108,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: tiny_tds
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: 0.6.3.rc1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: 0.6.3.rc1
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 'ARiiP: integrate everything. Automated real-time integration & interoperability
   platform.'
 email: