Module: Hari::Node::Queries::Relation::Backend::SortedSet

Extended by:
SortedSet
Included in:
SortedSet
Defined in:
lib/hari/node/queries/relation/backend/sorted_set.rb

Instance Method Summary collapse

Instance Method Details

#fetch(node, options = {}) ⇒ Object 



6
7
8
9
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 6

def fetch(node, options = {})
  set = node.sorted_set set_name(options)
  send "fetch_#{options[:result]}", set, options
end

#fetch_count(set, options) ⇒ Object 



35
36
37
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 35

def fetch_count(set, options)
  set.count
end

#fetch_nodes(set, options) ⇒ Object 



30
31
32
33
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 30

def fetch_nodes(set, options)
  nodes_ids = fetch_nodes_ids(set, options)
  nodes_ids.empty? ? [] : Hari.redis.mget(nodes_ids)
end

#fetch_nodes_ids(set, options) ⇒ Object 



25
26
27
28
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 25

def fetch_nodes_ids(set, options)
  index = set.name =~ /in$/ ? 0 : 2
  fetch_relations_ids(set, options).map { |r| r.split(':')[index] }
end

#fetch_relations_ids(set, options) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 11

def fetch_relations_ids(set, options)
  from, limit = options.values_at(:from, :limit)
  limit = limit.try(:to_i)

  if from.present? && from[:direction] == 'up'
    set.range_by_score from[:score], '+inf', desc: true, limit: [0, limit]
  elsif from.present? && from[:direction] == 'down'
    set.range_by_score '-inf', from[:score], desc: true, limit: [0, limit]
  else
    limit -= 1 unless limit <= 0
    set.range from, limit, desc: true
  end
end

#step(start_node, nodes_ids, options = {}) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/hari/node/queries/relation/backend/sorted_set.rb', line 39

def step(start_node, nodes_ids, options = {})
  stream    = start_node.sorted_set("stream:#{SecureRandom.hex(6)}")
  direction = options[:direction] == :in ? 0 : 2
  limit     = options.fetch(:limit, -1).try :to_i

  nodes_ids.each_with_index do |node_id, index|
    prune, stop = true, options.fetch(:step, 5)

    if limit == -1 || stream.count < limit
      prune, stop = false, limit
    end

    start, digg = 0, true

    while digg
      set = Hari(node_id).sorted_set set_name(options)

      if from = options[:from].presence
        args = { desc: true, with_scores: true, limit: [start, stop] }

        if from[:direction] == 'up'
          scored_relations_ids = set.range_by_score(from[:score], '+inf', args)
        elsif from[:direction] == 'down'
          scored_relations_ids = set.range_by_score('-inf', from[:score], args)
        end
      else
        scored_relations_ids = set.range(start, stop, desc: true, with_scores: true)
      end

      break if scored_relations_ids.empty?

      scored_nodes_ids = scored_relations_ids.map { |(r, s)| [s, r.split(':')[direction]] }.flatten
      stream.add *scored_nodes_ids

      last_node_id = scored_nodes_ids.last

      if prune
        stream.trim_by_rank 0, stop

        unless stream.include? last_node_id
          digg = false
          start += stop + 1
        end
      else
        digg = false
      end
    end
  end

  nodes_ids = stream.revrange

  if nodes_ids.any? && options[:result] == :nodes
    Hari.redis.mget nodes_ids
  else
    nodes_ids
  end
end