Skip to content

Commit

Permalink
init mysql source
Browse files Browse the repository at this point in the history
  • Loading branch information
hooopo committed May 8, 2016
0 parents commit 836b3ca
Show file tree
Hide file tree
Showing 17 changed files with 298 additions and 0 deletions.
9 changes: 9 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
/.bundle/
/.yardoc
/Gemfile.lock
/_yardoc/
/coverage/
/doc/
/pkg/
/spec/reports/
/tmp/
4 changes: 4 additions & 0 deletions .travis.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
language: ruby
rvm:
- 2.2.4
before_install: gem install bundler -v 1.11.2
4 changes: 4 additions & 0 deletions Gemfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
source 'https://rubygems.org'

# Specify your gem's dependencies in kiba-plus.gemspec
gemspec
21 changes: 21 additions & 0 deletions LICENSE.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
The MIT License (MIT)

Copyright (c) 2016 Hooopo

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
41 changes: 41 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
# Kiba::Plus

Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/kiba/plus`. To experiment with that code, run `bin/console` for an interactive prompt.

TODO: Delete this and the text above, and describe your gem

## Installation

Add this line to your application's Gemfile:

```ruby
gem 'kiba-plus'
```

And then execute:

$ bundle

Or install it yourself as:

$ gem install kiba-plus

## Usage

TODO: Write usage instructions here

## Development

After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).

## Contributing

Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/kiba-plus.


## License

The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).

10 changes: 10 additions & 0 deletions Rakefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
require "bundler/gem_tasks"
require "rake/testtask"

Rake::TestTask.new(:test) do |t|
t.libs << "test"
t.libs << "lib"
t.test_files = FileList['test/**/*_test.rb']
end

task :default => :spec
14 changes: 14 additions & 0 deletions bin/console
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
#!/usr/bin/env ruby

require "bundler/setup"
require "kiba/plus"

# You can add fixtures and/or initialization code here to make experimenting
# with your gem easier. You can also use a different console, if you like.

# (If you use this, don't forget to add pry to your Gemfile!)
# require "pry"
# Pry.start

require "irb"
IRB.start
8 changes: 8 additions & 0 deletions bin/setup
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
#!/usr/bin/env bash
set -euo pipefail
IFS=$'\n\t'
set -vx

bundle install

# Do any other automated setup that you need to do here
33 changes: 33 additions & 0 deletions kiba-plus.gemspec
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# coding: utf-8
lib = File.expand_path('../lib', __FILE__)
$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
require 'kiba/plus/version'

Gem::Specification.new do |spec|
spec.name = "kiba-plus"
spec.version = Kiba::Plus::VERSION
spec.authors = ["Hooopo"]
spec.email = ["hoooopo@gmail.com"]

spec.summary = %q{kiba plus}
spec.description = %q{kiba plus}
spec.homepage = ""
spec.license = "MIT"

# Prevent pushing this gem to RubyGems.org by setting 'allowed_push_host', or
# delete this section to allow pushing this gem to any host.
if spec.respond_to?(:metadata)
spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'"
else
raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
end

spec.files = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
spec.bindir = "exe"
spec.executables = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
spec.require_paths = ["lib"]
spec.add_runtime_dependency "kiba"
spec.add_development_dependency "bundler", "~> 1.11"
spec.add_development_dependency "rake", "~> 10.0"
spec.add_development_dependency "minitest", "~> 5.0"
end
11 changes: 11 additions & 0 deletions lib/kiba/plus.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
require "kiba/plus/version"
require "kiba/plus/mysql_source"
require "kiba/plus/pg_destination"
require "kiba/plus/csv_destination"
require "kiba/plus/pg_bulk_destination"

module Kiba
module Plus
# Your code goes here...
end
end
25 changes: 25 additions & 0 deletions lib/kiba/plus/csv_destination.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
require 'csv'

module Kiba::Plus
class CsvDestination
def initialize
@csv = CSV.open(output_file, 'w', {col_sep: delimiter})
end

def output_file
'/tmp/plus.csv'
end

def delimiter
', '
end

def write(row)
@csv << row.values
end

def close
@csv.close
end
end
end
50 changes: 50 additions & 0 deletions lib/kiba/plus/mysql_source.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
require 'mysql2'
require 'uri'

module Kiba
module Plus
class MysqlSource
attr_reader :full_load, :client, :connect_url

def initialize(connect_url, full_load = true)
@full_load = full_load
@connect_url = connect_url
@client = Mysql2::Client.new(connect_hash(connect_url))
end

def each
results = client.query(query, as: :hash, symbolize_keys: true, stream: true)
results.each do |row|
yield(row)
end
end

def query
raise 'Not Imp!'
end

def output
[:id, :name]
end

def last_pull_at
end

def full_load?
end

private

def connect_hash(url)
u = URI.parse(url)
{
host: u.host,
username: u.user,
password: u.password,
port: u.port,
database: u.path[1..-1]
}
end
end
end
end
5 changes: 5 additions & 0 deletions lib/kiba/plus/pg_bulk_destination.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
module Kiba::Plus
# TODO
class PgBulkDestination
end
end
43 changes: 43 additions & 0 deletions lib/kiba/plus/pg_destination.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
require 'pg'

module Kiba::Plus
class PgDestination
def initialize(conn_url = nil)
@conn = PG.connect(conn_url || connect_url)
@conn.prepare(prepare_name, prepare_sql)
end

def write(row)
@conn.exec_prepared(prepare_name,
columns.map { |column| row[column]})
rescue PG::Error => ex
puts "ERROR for #{row[:email]}"
puts ex.message
# Maybe, write to db table or file
end

def close
@conn.close
@conn = nil
end

private

def connect_url
raise 'Not Imp!'
end

def prepare_name
self.class.to_s.downcase + "stmt"
end

def prepare_sql
raise 'Not Imp!'
#'insert into users (email, password_digest, created_at, updated_at) values ($1, $2, $3, $4)'
end

def columns
raise 'Not Imp!'
end
end
end
5 changes: 5 additions & 0 deletions lib/kiba/plus/version.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
module Kiba
module Plus
VERSION = "0.1.0"
end
end
11 changes: 11 additions & 0 deletions test/kiba/plus_test.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
require 'test_helper'

class Kiba::PlusTest < Minitest::Test
def test_that_it_has_a_version_number
refute_nil ::Kiba::Plus::VERSION
end

def test_it_does_something_useful
assert false
end
end
4 changes: 4 additions & 0 deletions test/test_helper.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
require 'kiba/plus'

require 'minitest/autorun'

0 comments on commit 836b3ca

Please sign in to comment.