Contents

require 'mongo'
require_relative 'channel_tasks'

class StoreTasks
  def initialize(channel=nil, dispatcher=nil)
    puts "init store tasks"
    @@mongo_db ||= Mongo::MongoClient.new("localhost", 27017)
    @@db ||= @@mongo_db.db("development")
    
    @channel = channel
    @dispatcher = dispatcher
  end
  
  def db
    @@db
  end
  
  def save(collection, data)
    puts "Insert: #{data.inspect} on #{collection.inspect}"
    # Try to create
    # TODO: Seems mongo is dumb and doesn't let you upsert with custom id's
    begin
      @@db[collection].insert(data)
      id = {'_id' => data.delete('_id')}
    rescue Mongo::OperationFailure => error
      # Really mongo client?
      if error.message[/^11000[:]/]
        # Update because the id already exists
        id = {'_id' => data.delete('_id')}
        @@db[collection].update(id, data)
      else
        raise
      end
    end
    
    id = id['_id']

    puts "DATA: #{data.merge('_id' => id).symbolize_keys.inspect}"
    ChannelTasks.send_message_to_channel("#{collection}##{id}", ['update', nil, id, data.merge('_id' => id).symbolize_keys], @channel)
  end
  
  def find(collection, scope, query=nil)
    results = @@db[collection].find(scope).to_a.map {|item| item.symbolize_keys }
    puts "FIND: #{collection.inspect} - #{scope} - #{results.inspect}"
    
    return results
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
volt-0.4.7	app/volt/tasks/store_tasks.rb
volt-0.4.5	app/volt/tasks/store_tasks.rb
volt-0.4.4	app/volt/tasks/store_tasks.rb
volt-0.4.3	app/volt/tasks/store_tasks.rb
volt-0.4.2	app/volt/tasks/store_tasks.rb
volt-0.4.1	app/volt/tasks/store_tasks.rb