rethinkdb_job 0.1.0

data/lib/rethinkdb_job/setup.rb

@@ -0,0 +1,55 @@
+class RethinkDBJob
+  module Setup
+    def rql_db_string(rql_db)
+      rql_db.instance_variable_get("@body")[1].first
+    end
+
+    def database
+      # ugly hack until @rdb.default_db works properly
+      db = rdb.instance_variable_get("@default_opts")[:db]
+      if db
+        rql_db_string(db)
+      else
+        @rdb_config[:db] || "test"
+      end
+    end
+
+    def done_setup?
+      @done_setup
+    end
+
+    def ensure_setup
+      unless done_setup?
+        setup
+        @done_setup = true
+      end
+    end
+
+    def setup
+      begin
+        rdb
+      rescue => e
+        msg = "cannot connect to RethinkDB #{@rdb_config.inspect}"
+        raise e.class, e.message + ", #{msg}", caller
+      end
+
+      begin
+        r.db_create(database).run(rdb)
+      rescue RethinkDB::RqlRuntimeError => err
+        # already exists, cool
+      rescue => e
+        msg = "cannot create db #{database.inspect}"
+        raise e.class, e.message + ", #{msg}", caller
+      end
+
+      begin
+        r.db(database).table_create(table).run(rdb)
+      rescue RethinkDB::RqlRuntimeError => err
+        # already exists, cool
+      rescue => e
+        msg = "cannot create table #{table.inspect} on database #{database.inspect}"
+        raise e.class, e.message + ", #{msg}", caller
+      end
+    end
+  end
+end

data/lib/rethinkdb_job/version.rb

@@ -0,0 +1,3 @@
+module RethinkdbJob
+  VERSION = "0.1.0"
+end

data/lib/rethinkdb_job.rb

@@ -0,0 +1,57 @@
+require 'rethinkdb'
+require 'rethinkdb_job/setup'
+
+class RethinkDBJob
+  attr_reader :table, :r
+
+  include RethinkDBJob::Setup
+
+  def initialize(rdb_config={})
+    @rdb_config = rdb_config
+    @table = @rdb_config.delete(:table) || "jobs"
+    @done_setup = false
+
+    @r = RethinkDB::RQL.new
+  end
+
+  def rdb
+    @rdb ||= @r.connect(@rdb_config)
+  end
+
+  def create
+    ensure_setup
+    result = @r.table(@table).insert({}, :return_vals => true).run(@rdb)
+    if result["inserted"] == 1
+      result["new_val"]["id"]
+    else
+      raise "Unable to create new job record"
+    end
+  end
+
+  def find(job_id)
+    @r.table(@table).get(job_id).run(@rdb)
+  end
+
+  TIMESTAMP_COLUMNS = [:job_start, :job_finish]
+  def set_timestamp(job_id, column, time=Time.now)
+    ensure_setup
+    if TIMESTAMP_COLUMNS.include?(column.to_sym)
+      @r.table(@table).get(job_id).update(column => time).run(@rdb)
+      time
+    else
+      raise "expected column to be one of #{timestamp_columns.inspect} (#{column.inspect})"
+    end
+  end
+
+  def logdir
+    @rdb_config[:logdir]
+  end
+  
+  def logfile(job_id)
+    File.join(logdir, "job_#{job_id}.log")
+  end
+
+  def tail(job_id, lines=10)
+    `tail -#{Integer(lines).to_s} #{logfile(job_id)} 2>&1`
+  end
+end

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification
+name: rethinkdb_job
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Duane Johnson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-08-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rethinkdb
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.13.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.13.0
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: guard-rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Stores job queue data in rethinkdb
+email:
+- duane.johnson@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/rethinkdb_job.rb
+- lib/rethinkdb_job/setup.rb
+- lib/rethinkdb_job/version.rb
+homepage: https://github.com//rethinkdb_job
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: ! '[none]'
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Stores job queue data in rethinkdb
+test_files: []
+has_rdoc: