RethinkDB and NoBrainer Session Store for Ruby on Rails 4

rethinkdb_session_store.rb

# lib/rethinkdb_session_store.rb
require 'rethinkdb'

class RethinkdbSessionStore < ActionDispatch::Session::AbstractStore
  def initialize(app, options = {})
    super
    @r = RethinkDB::RQL.new.table('sessions')
  end

  def get_session(env, session_id)
    session_data = {}
    session_id ||= generate_sid

    @r.get(session_id).run(NoBrainer.connection.raw) do |s|
      session_data = s['data'] unless s.nil?
    end

    [session_id, session_data]
  end

  def set_session(env, session_id, session_data, options)
    result = if @r.get(session_id).run(NoBrainer.connection.raw).nil?
      @r.insert( id: session_id, data: session_data ).run(NoBrainer.connection.raw)
    else
      @r.get(session_id).replace(id: session_id, data: session_data).run(NoBrainer.connection.raw)
    end
    
    session_id
  end

  def destroy_session(env, session_id, options)
    @r.get(session_id).delete.run(NoBrainer.connection.raw)
    generate_sid
  end
end

Based on CustomFileStore