Class: Bayesball::Persistence::Sequel

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/bayesball/persistence/sequel.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(dataset) ⇒ Sequel

Returns a new instance of Sequel.



7
8
9
 
# File 'lib/bayesball/persistence/sequel.rb', line 7

def initialize(dataset)
  @dataset = dataset
end

Instance Attribute Details

#dataset ⇒ Object (readonly)

Returns the value of attribute dataset.



6
7
8
 
# File 'lib/bayesball/persistence/sequel.rb', line 6

def dataset
  @dataset
end

Instance Method Details

#[](key) ⇒ Object 



15
16
17
 
# File 'lib/bayesball/persistence/sequel.rb', line 15

def [](key)
  dataset.where(bucket: key).to_hash(:word, :count)
end

#[]=(key, value) ⇒ Object 



19
20
21
22
23
24
 
# File 'lib/bayesball/persistence/sequel.rb', line 19

def []=(key, value)
  dataset.where(bucket: key).delete
  value.each do |word, count|
    dataset.insert word: word, count: count, bucket: key
  end
end

#each ⇒ Object 



26
27
28
29
30
31
32
33
34
35
 
# File 'lib/bayesball/persistence/sequel.rb', line 26

def each
  if block_given?
    dataset.to_a.group_by{|i| i[:bucket] }.each do |bucket, group|
      counts = Hash[group.map{|g| [g[:word], g[:count]]}]
      yield [bucket, counts]
    end
  else
    self.to_enum
  end
end

#empty? ⇒ Boolean

Returns:

  • (Boolean) 


11
12
13
 
# File 'lib/bayesball/persistence/sequel.rb', line 11

def empty?
  @collection.count == 0
end