ruboty-postgres 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 11e4de665e99eed7b510e88ace66499b0ce0e653
+  data.tar.gz: 2623c04f9300a4744d97bf52a3aded42f5f265b5
+SHA512:
+  metadata.gz: 973d9d438a2c7d0b34852b7312568bd79c5ea1ea6481e66f0c5af8009694edf621b95488bc298cdad6094e1254b17b9a3d19e6f28f807d63697d128b6edd2553
+  data.tar.gz: 1c3b792badeb2546cdbf932bf18e2563529e798b5284fca7a1bc26b500690909460e5bf207b04bb13a5a4035ad553d72b9a9b8a47a4af245b96e32f1c146690c

data/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/vendor/bundle/
+

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015 SHIOYA, Hiromu
+
+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.

data/README.md

@@ -0,0 +1,37 @@
+# Ruboty::Postgres
+
+store memory of [ruboty](https://github.com/r7kamura/ruboty) in postgresql
+
+## ENV
+
+- `POSTGRES_HOST`
+    - host of postgres (default: localhost)
+    - optional
+- `POSTGRES_PORT`
+    - port number of postgres (default: 5432)
+    - optional
+- `POSTGRES_USER`
+    - user name of postgres
+    - required
+- `POSTGRES_PASSWORD`
+    - password of postgres
+    - required
+- `POSTGRES_DBNAME`
+    - database name of postgres
+    - required
+- `POSTGRES_NAMESPACE`
+    - relation name of postgres (default: ruboty)
+    - optional
+- `POSTGRES_BOTNAME`
+    - name of your ruboty (default: ruboty)
+    - optional
+    - max length: 240
+- `POSTGRES_SAVE_INTERVAL`
+    - sleep duration between each save (default: 10)
+    - optional
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+
+core of implementation refers to [r7kamura/ruboty-redis](https://github.com/r7kamura/ruboty-redis) with :sushi: .

data/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+
+task default: :test
+
+desc 'Run test_unit based test'
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+  t.test_files = Dir['test/**/test_*.rb']
+  t.verbose = true
+end

data/lib/ruboty/brains/postgres.rb

@@ -0,0 +1,134 @@
+require 'pg'
+
+module Ruboty
+  module Brains
+    class Postgres < Base
+      attr_reader :thread
+
+      env :POSTGRES_HOST,          'host of postgres (default: localhost)',          optional: true
+      env :POSTGRES_PORT,          'port number of postgres (default: 5432)',        optional: true
+      env :POSTGRES_USER,          'user name of postgres',                          optional: false
+      env :POSTGRES_PASSWORD,      'password of postgres',                           optional: false
+      env :POSTGRES_DBNAME,        'database name of postgres',                      optional: false
+      env :POSTGRES_NAMESPACE,     'relation name of postgres (default: ruboty)',    optional: true
+      env :POSTGRES_BOTNAME,       'name of your ruboty (default: ruboty)',          optional: true
+      env :POSTGRES_SAVE_INTERVAL, 'sleep duration between each save (default: 10)', optional: true
+
+      def initialize
+        super
+        @thread = Thread.new { sync }
+        @thread.abort_on_exception = true
+      end
+
+      def data
+        @data ||= load || {}
+      end
+
+      # Override.
+      def validate!
+        super
+        Ruboty.die("#{self.class.usage}") unless host
+      end
+
+      private
+
+      def save
+        str = connection.escape_bytea(Marshal.dump(data))
+        connection.exec_prepared('save_statement', [str, botname])
+      end
+
+      def load
+        result = connection.exec_prepared('load_statement', [botname])
+        if result.count > 0
+          Marshal.load(connection.unescape_bytea(result.first['marshal']))
+        else
+          str = connection.escape_bytea(Marshal.dump({}))
+          sql = "INSERT INTO #{namespace} (botname, marshal) VALUES ($1, $2);"
+          connection.prepare('insert_statement', sql)
+          connection.exec_prepared('insert_statement', [botname, str])
+          {}
+        end
+      end
+
+      def sync
+        loop do
+          wait
+          save
+        end
+      end
+
+      def wait
+        sleep(interval)
+      end
+
+      def connection
+        if @connection.nil?
+          @connection = PG::connect(
+            host:     host,
+            user:     user,
+            password: password,
+            dbname:   dbname,
+            port:     port
+          )
+          migrate(@connection)
+          prepare_save_and_load_statement(@connection)
+        end
+        @connection
+      end
+
+      def host
+        ENV['POSTGRES_HOST'] || 'localhost'
+      end
+
+      def port
+        ENV['POSTGRES_PORT'] || 5432
+      end
+
+      def user
+        ENV['POSTGRES_USER'] || raise('ENV["POSTGRES_USER"] is required.')
+      end
+
+      def password
+        ENV['POSTGRES_PASSWORD'] || raise('ENV["POSTGRES_PASSWORD"] is required.')
+      end
+
+      def dbname
+        ENV['POSTGRES_DBNAME'] || raise('ENV["POSTGRES_DBNAME"] is required.')
+      end
+
+      def namespace
+        ENV['POSTGRES_NAMESPACE'] || 'ruboty'
+      end
+
+      def botname
+        ENV['POSTGRES_BOTNAME'] || 'ruboty'
+      end
+
+      def interval
+        ENV['POSTGRES_SAVE_INTERVAL'].to_i || 10
+      end
+
+      def migrate(conn)
+        unless relation_exists?(conn)
+          sql = "CREATE TABLE #{namespace} (botname VARCHAR(240) PRIMARY KEY, marshal BYTEA);"
+          conn.exec(sql)
+          load
+        end
+      end
+
+      def relation_exists?(conn)
+        sql = "SELECT relname FROM pg_class WHERE relkind = 'r' AND relname = $1;"
+        conn.prepare('exist_statement', sql)
+        result = conn.exec_prepared('exist_statement', [namespace])
+        return false if result.count == 0
+        result.first['relname'] == namespace
+      end
+
+      # only this statment is called repeatedly
+      def prepare_save_and_load_statement(conn)
+        conn.prepare('save_statement', "UPDATE #{namespace} SET marshal = $1 WHERE botname = $2;")
+        conn.prepare('load_statement', "SELECT marshal FROM #{namespace} WHERE botname = $1;")
+      end
+    end
+  end
+end

data/lib/ruboty/postgres.rb

@@ -0,0 +1,2 @@
+require 'ruboty/postgres/version'
+require 'ruboty/brains/postgres'

data/lib/ruboty/postgres/version.rb

@@ -0,0 +1,5 @@
+module Ruboty
+  module Postgres
+    VERSION = '0.1.0'
+  end
+end

data/ruboty-postgres.gemspec

@@ -0,0 +1,29 @@
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ruboty/postgres/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ruboty-postgres'
+  spec.version       = Ruboty::Postgres::VERSION
+  spec.authors       = ['SHIOYA, Hiromu']
+  spec.email         = ['kwappa.856@gmail.com']
+
+  spec.summary       = 'store memory of ruboty in postgresql'
+  spec.description   = spec.summary
+  spec.homepage      = 'https://github.com/kwappa/ruboty-postgres'
+  spec.license       = 'MIT'
+
+
+  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_dependency 'ruboty'
+  spec.add_dependency 'pg'
+
+  spec.add_development_dependency 'bundler', '~> 1.10'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'test-unit'
+end

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification
+name: ruboty-postgres
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- SHIOYA, Hiromu
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-12-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ruboty
+  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'
+- !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'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: store memory of ruboty in postgresql
+email:
+- kwappa.856@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/ruboty/brains/postgres.rb
+- lib/ruboty/postgres.rb
+- lib/ruboty/postgres/version.rb
+- ruboty-postgres.gemspec
+homepage: https://github.com/kwappa/ruboty-postgres
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: store memory of ruboty in postgresql
+test_files: []