treport 0.1.0

data/lib/treport/db.rb

@@ -0,0 +1,49 @@
+module Treport
+  class Db
+    AdapterType={
+        Mysql: "mysql2",
+        PostgreSQL: "postgresql",
+        Sqlserver: "sqlserver",
+        Oracle: "oracle"
+    }
+
+    def self.adapter_options
+      [["Mysql", AdapterType[:Mysql]], ["PostgreSQL", AdapterType[:PostgreSQL]], ["Sqlserver", AdapterType[:Sqlserver]], ["Oracle", AdapterType[:Oracle]]]
+    end
+
+    def self.oracle_query(conectstr, sql)
+      require 'ruby-oci8'
+      result = []
+      conn = OCI8.new(conectstr[:username], conectstr[:password], "#{conectstr[:host]}:#{conectstr[:port]}/#{conectstr[:database]}")
+      cursor = conn.parse(sql)
+      cursor.exec
+      while r = cursor.fetch_hash()
+        r.each do |n, v|
+          unless v
+            r[n] = "其他"
+          end
+        end
+        result << r
+      end
+      result
+    end
+
+    def self.mysql_query(conectstr, sql)
+      result = []
+      Mysql2::Client.new(conectstr).query(sql).each do |r|
+        result << r
+      end
+      result
+    end
+
+    def self.sqlserver_query(conectstr, sql)
+      connstr = {}
+      conectstr.each { |k, v| connstr[k.to_sym]=v }
+      result = []
+      TinyTds::Client.new(connstr).execute(sql).each do |row|
+        result << row
+      end
+      result
+    end
+  end
+end

data/lib/treport.rb

@@ -0,0 +1,9 @@
+module Treport
+end
+# load Rails/Railtie
+begin
+  require 'rails'
+rescue LoadError
+  #do nothing
+end
+require 'treport/db'