Class: Anemone::Storage::MongoDB

Inherits:

Object

• Object
• Anemone::Storage::MongoDB

show all

Defined in:

lib/anemone/storage/mongodb.rb

Constant Summary

BINARY_FIELDS =

%w(body headers data)

Instance Method Summary (collapse)

• - (Object) [](url)
• - (Object) []=(url, page)
• - (Object) close
• - (Object) delete(url)
• - (Object) each
• - (Boolean) has_key?(url)
• - (MongoDB) initialize(mongo_db, collection_name) constructor

  A new instance of MongoDB.

• - (Object) keys
• - (Object) merge!(hash)
• - (Object) size

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

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

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

73
74
75

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

def close
  @db.connection.close
end

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

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

69
70
71

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

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

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

54
55
56
57

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

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

59
60
61

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

def size
  @collection.count
end