Class: Dragonfly::DataStorage::MongoDataStore

Inherits:
Object
  • Object
show all
Defined in:
lib/dragonfly/data_storage/mongo_data_store.rb

Constant Summary collapse

OBJECT_ID =

Mongo gem deprecated ObjectID in favour of ObjectId

defined?(BSON::ObjectId) ? BSON::ObjectId : BSON::ObjectID
INVALID_OBJECT_ID = 
defined?(BSON::InvalidObjectId) ? BSON::InvalidObjectId : BSON::InvalidObjectID

Instance Method Summary collapse

Constructor Details

#initialize(opts = {}) ⇒ MongoDataStore

Returns a new instance of MongoDataStore.



22
23
24
25
26
27
28
29
30
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 22

def initialize(opts={})
  self.host = opts[:host]
  self.port = opts[:port]
  self.database = opts[:database] if opts[:database]
  self.username = opts[:username]
  self.password = opts[:password]
  self.connection = opts[:connection]
  self.db = opts[:db]
end

Instance Method Details

#connection ⇒ Object 



60
61
62
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 60

def connection
  @connection ||= Mongo::Connection.new(host, port)
end

#db ⇒ Object 



64
65
66
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 64

def db
  @db ||= connection.db(database)
end

#destroy(uid) ⇒ Object

Raises:



53
54
55
56
57
58
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 53

def destroy(uid)
  ensure_authenticated!
  grid.delete(bson_id(uid))
rescue Mongo::GridFileNotFound, INVALID_OBJECT_ID => e
  raise DataNotFound, "#{e} - #{uid}"
end

#grid ⇒ Object 



68
69
70
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 68

def grid
  @grid ||= Mongo::Grid.new(db)
end

#retrieve(uid) ⇒ Object

Raises:



40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 40

def retrieve(uid)
  ensure_authenticated!
  grid_io = grid.get(bson_id(uid))
  meta = marshal_decode(grid_io.metadata)
  meta.merge!(:stored_at => grid_io.upload_date)
  [
    grid_io.read,
    meta
  ]
rescue Mongo::GridFileNotFound, INVALID_OBJECT_ID => e
  raise DataNotFound, "#{e} - #{uid}"
end

#store(temp_object, opts = {}) ⇒ Object 



32
33
34
35
36
37
38
 
# File 'lib/dragonfly/data_storage/mongo_data_store.rb', line 32

def store(temp_object, opts={})
  ensure_authenticated!
  temp_object.file do |f|
    mongo_id = grid.put(f, :metadata => marshal_encode(opts[:meta] || {}))
    mongo_id.to_s
  end
end