lolitado 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c6f12af33a4803ac99c0ccd8e16ea78739c9b98a46ad4f9ff05b32c5bc5247f8
-  data.tar.gz: bbe3525397830d523c92be5a1cd7dd6813f6a30243dfb2286e936d2634a0f576
+  metadata.gz: a3fd234304855f6114d300e141b4a73a32853571a76ff75b0ca9a27098fa8d75
+  data.tar.gz: a0e6606f7325ee49ab9a023b2960cb3cc2942e3590a093813789a3d9c16ee9dc
 SHA512:
-  metadata.gz: 37fa088b2b3fde544b5cdc5949bc9c6858ae41b9f32755982dd82417b4ceb560d248b43ec0684b72ce85995ab9fdfc5901743509b7b4c57adced2befbb7b44a2
-  data.tar.gz: e33804c5551ec6df52eb9caacdf9d612026fecbf97be56317fc58998c47bd31cbb851161899e8c0486fc1ce788b6347311d01cf9814d773630872d5982967de7
+  metadata.gz: a14021644154f53d50844fb227bc96f65c0e9f9cb2cdce35de9b247b2305596bed93e08a823d467faceb6754b1fb584c03689085513c8d58a90eb8d24a341e9f
+  data.tar.gz: 44059dd091f273aef49f4053e3dd0b2190e86f6b365d6485a57f45e73bfc934c34a736e753193c4ff51afe56441e831fc483cfdc9fcbeea324cf9a1d3ce32b13

data/lib/accessors.rb

@@ -0,0 +1,22 @@
+#
+# Contains the class level methods that are inserted into your api class
+#
+module Accessors
+
+  # just for example "how to create method in accesors"
+  def row name, identifier
+    define_method("delete_#{name}") do |value|
+      db[identifier[:table].to_sym].filter(value).delete
+    end
+  end
+
+  def column name, identifier
+    define_method("avg_#{name}") do |value|
+      db[identifier[:table].to_sym].filter(value).avg(identifier.values.last.to_sym)
+    end
+    define_method("query_#{name}") do |value|
+      db[identifier[:table].to_sym].filter(value).all
+    end
+  end
+
+end

data/lib/db.rb

@@ -0,0 +1,57 @@
+require './accessors'
+
+class DBFactory
+  extend Accessors
+
+  attr_accessor :db
+
+  def initialize db
+    @db = db
+  end
+
+  def query_duration sql, type = true, waiting_time = 10
+    start = Time.now
+    if type
+      result = query(sql, waiting_time)
+    else
+      result = query_empty(sql, waiting_time)
+    end
+    finish = Time.now
+    msecs = (finish - start) * 1000.0
+    hash = {'result' => result, 'duration' => msecs}
+    return hash
+  end
+
+  def query sql, waiting_time = 10
+    result = db[sql].all
+    if result.length == 0
+      if waiting_time != 0
+        sleep 1
+        result = query(sql, waiting_time - 1)
+      end
+    end
+    return result
+  end
+
+  def multiple_query sql
+    splited_sql = sql.split(';')
+    splited_sql.each do |each_sql|
+      query_empty(each_sql)
+    end
+  end
+
+  def query_empty sql, waiting_time = 10
+    result = db[sql].all
+    if result.length != 0
+      if waiting_time != 0
+        sleep 1
+        result = query(sql, waiting_time - 1)
+      end
+    end
+    return result
+  end
+
+  def insert sql
+    db[sql]
+  end
+end

data/lib/pool.rb

@@ -0,0 +1,109 @@
+require 'mysql2'
+require 'sequel'
+require 'net/ssh/gateway'
+
+class Pool
+
+  attr_accessor :db_pool, :ssh_pool, :file_path
+
+  def initialize file_path
+    @db_pool = {}
+    @ssh_pool = {}
+    @file_path = file_path
+  end
+
+  def use(params ={})
+    ssh_name = params.fetch(:ssh, false)
+    db_name = params.fetch(:db, false)
+    if db_pool[db_name].nil? && db_name
+      ssh_key = fetch_ssh_config db_name
+      unless ssh_key.nil?
+        ssh = create_or_use_ssh ssh_key
+        port = forward_port ssh, ssh_key
+        db = connect_database db_name, port
+        db_pool[db_name] = db
+      else
+        db = connect_database_directly db_name
+        db_pool[db_name] = db
+      end
+    elsif ssh_name
+      create_or_use_ssh ssh_name
+    else
+      db_pool[db_name]
+    end
+  end
+
+  def create_or_use_ssh name
+    if ssh_pool[name].nil?
+      ssh = connect_remote_server name
+      ssh_pool[name] = ssh
+    else
+      ssh_pool[name]
+    end
+  end
+
+  def connect_database name, port=false
+    database_file = YAML.load_file(file_path)
+    db_conf = database_file[name]
+    db_conf = db_conf.inject({}){|memo,(k,v)| memo[k.to_sym] = v; memo}
+    db_conf = db_conf.merge(:port => port) if port
+    db = get_ready_for_database db_conf
+  end
+
+  def get_ready_for_database conf
+    begin
+      Sequel.connect(conf)
+    rescue
+      fail "database configuration \n #{conf} \n is not correct, please double check"
+    end
+  end
+
+  def connect_remote_server name
+    database_file = YAML.load_file(file_path)
+    ssh_conf = database_file[name]
+    ssh_conf = ssh_conf.delete_if { |key,value| key == 'database'}
+    ssh = get_ready_for_ssh ssh_conf
+    return ssh
+  end
+
+  def get_ready_for_ssh conf
+    host = conf.delete('host')
+    user = conf.delete('user')
+    options = conf.delete_if {|key,value| key == 'host' && 'user'}
+    options = options.inject({}){|memo,(k,v)| memo[k.to_sym] = v; memo}
+    begin
+      Net::SSH::Gateway.new(host, user, options)
+    rescue
+      fail "ssh configuration \n #{conf} \n is not correct, please double check"
+    end
+  end
+
+  def forward_port ssh, name
+    database_file = YAML.load_file(file_path)
+    ssh_conf = database_file[name]
+    new_conf = ssh_conf.delete('database')
+    host = new_conf.delete('host')
+    remote_port = new_conf.delete('remote_port')
+    local_port = new_conf.delete('local_port')
+    begin
+      ssh.open(host, remote_port, local_port)
+    rescue Errno::EADDRINUSE
+      return local_port
+    rescue
+      fail "fail to forward remote port #{remote_port} to local_port #{local_port}"
+    end
+  end
+
+  def fetch_ssh_config db_name
+    database_file = YAML.load_file(file_path)
+    db_conf  = database_file[db_name]
+    ssh_key = db_conf.delete('ssh')
+  end
+
+  def connect_database_directly db_name
+    database_file = YAML.load_file(file_path)
+    db_conf  = database_file[db_name]
+    db_conf = db_conf.inject({}){|memo,(k,v)| memo[k.to_sym] = v; memo}
+    db = Sequel.connect(db_conf)
+  end
+end

data/lib/request.rb

@@ -0,0 +1,46 @@
+require 'httparty'
+
+class Request
+  include HTTParty
+
+  attr_accessor :new_headers
+  # debug_output $stdout
+
+  def new_response response, msecs
+    new_response = {}
+    new_response.merge({:response => response, :message => response.parsed_response, :status => response.code, :duration => msecs})
+  end
+
+  def request base_uri, method, endpoint, body = false
+    url = base_uri + endpoint
+    case method
+    when "get"
+      start = Time.now
+      response = self.class.get(url, :body => body, :headers => new_headers, :verify => false)
+      finish = Time.now
+    when "post"
+      start = Time.now
+      response = self.class.post(url, :body => body, :headers => new_headers, :verify => false)
+      finish = Time.now
+    when "put"
+      start = Time.now
+      response = self.class.put(url, :body => body, :headers => new_headers, :verify => false)
+      finish = Time.now
+    when "delete"
+      start = Time.now
+      response = self.class.delete(url, :headers => new_headers, :verify => false)
+      finish = Time.now
+    else
+      puts "#{method} is invalid method."
+      exit
+    end
+    msecs = (finish - start) * 1000.0
+    return new_response(response, msecs)
+  end
+
+  def add_headers value
+    @new_headers ||= {}
+    @new_headers = @new_headers.merge(value.to_hash)
+    return @new_headers
+  end
+end

data/lib/version.rb

@@ -0,0 +1,3 @@
+module Lolitado
+  VERSION = "0.0.3"
+end

data/lib/yaml.rb

@@ -0,0 +1,31 @@
+require 'psych'
+
+module Cleric
+  module YAML
+
+    CONFIGURE_PATH = File.expand_path("../data/stage.yml",__FILE__)
+
+    def self.fetch_corresponding_conf_by name
+      all_hash_values = load_yml(CONFIGURE_PATH)
+      conf_value = fetch_value_by all_hash_values, name
+      return conf_value
+    end
+
+    def self.load_yml path
+      Psych.load_file(path)
+    end
+
+    def self.fetch_value_by hash, key
+      if hash == false
+        fail "no configuration in #{ENV['PLATFORM']}.yml"
+      else
+        if hash[key].nil?
+          fail "can not find corresponding configure value for #{key}"
+        else
+          hash[key]
+        end
+      end
+    end
+
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lolitado
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Dolores Zhang
@@ -73,7 +73,13 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/accessors.rb
+- lib/db.rb
 - lib/lolitado.rb
+- lib/pool.rb
+- lib/request.rb
+- lib/version.rb
+- lib/yaml.rb
 homepage: http://github.com/doloreszhang/lolitado
 licenses:
 - MIT