Class: Anemone::Storage::MongoDB

Inherits:
Object
  • Object
show all
Defined in:
lib/anemone/storage/mongodb.rb

Constant Summary

BINARY_FIELDS =
%w(body headers data)

Instance Method Summary (collapse)

Constructor Details

- (MongoDB) initialize(mongo_db, collection_name)

Returns a new instance of MongoDB



14
15
16
17
18
19
# File 'lib/anemone/storage/mongodb.rb', line 14

def initialize(mongo_db, collection_name)
  @db = mongo_db
  @collection = @db[collection_name]
  @collection.remove
  @collection.create_index 'url'
end

Instance Method Details

- (Object) [](url)



21
22
23
24
25
# File 'lib/anemone/storage/mongodb.rb', line 21

def [](url)
  if value = @collection.find_one('url' => url.to_s)
    load_page(value)
  end
end

- (Object) []=(url, page)



27
28
29
30
31
32
33
34
35
36
37
# File 'lib/anemone/storage/mongodb.rb', line 27

def []=(url, page)
  hash = page.to_hash
  BINARY_FIELDS.each do |field|
    hash[field] = BSON::Binary.new(hash[field]) unless hash[field].nil?
  end
  @collection.update(
    {'url' => page.url.to_s},
    hash,
    :upsert => true
  )
end

- (Object) close



73
74
75
# File 'lib/anemone/storage/mongodb.rb', line 73

def close
  @db.connection.close
end

- (Object) delete(url)



39
40
41
42
43
# File 'lib/anemone/storage/mongodb.rb', line 39

def delete(url)
  page = self[url]
  @collection.remove('url' => url.to_s)
  page
end

- (Object) each



45
46
47
48
49
50
51
52
# File 'lib/anemone/storage/mongodb.rb', line 45

def each
  @collection.find do |cursor|
    cursor.each do |doc|
      page = load_page(doc)
      yield page.url.to_s, page 
    end
  end
end

- (Boolean) has_key?(url)

Returns:

  • (Boolean)


69
70
71
# File 'lib/anemone/storage/mongodb.rb', line 69

def has_key?(url)
  !!@collection.find_one('url' => url.to_s)
end

- (Object) keys



63
64
65
66
67
# File 'lib/anemone/storage/mongodb.rb', line 63

def keys
  keys = []
  self.each { |k, v| keys << k.to_s }
  keys
end

- (Object) merge!(hash)



54
55
56
57
# File 'lib/anemone/storage/mongodb.rb', line 54

def merge!(hash)
  hash.each { |key, value| self[key] = value }
  self
end

- (Object) size



59
60
61
# File 'lib/anemone/storage/mongodb.rb', line 59

def size
  @collection.count
end