30 lines
		
	
	
		
			904 B
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			30 lines
		
	
	
		
			904 B
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| require 'lib/string_helper'
 | |
| 
 | |
| module Connectors
 | |
|   module Helpers
 | |
|     module UpdateCommand
 | |
| 
 | |
|       # when included, this module adds method #update and alias for it.
 | |
|       # Alias name depends on base class name.
 | |
|       # We need this alias to forward methods from MongoConnector to resources connectors.
 | |
| 
 | |
|       def self.included(base)
 | |
|         resource_name = StringHelper.underscore_class(base)
 | |
|         method_name = "#{resource_name}_update".to_sym
 | |
|         alias_method method_name, :update
 | |
|       end
 | |
| 
 | |
|       def update(record)
 | |
|         record.validate!
 | |
|         collection.update({"_id" => record.id}, record.to_mongo_hash)
 | |
|       rescue Mongo::OperationFailure => e
 | |
|         if e.message =~ /^11000/
 | |
|           resource_name = StringHelper.underscore_class(record.class)
 | |
|           raise InvalidRecord.new("Duplicate key error: #{resource_name} with id '#{record.id}'")
 | |
|         end
 | |
|       end
 | |
| 
 | |
|     end
 | |
|   end
 | |
| end
 | 
