Class: Corpshort::Backends::Redis

Inherits:
Base
  • Object
show all
Defined in:
lib/corpshort/backends/redis.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(redis: ::Redis.method(:current), prefix: "corpshort:") ⇒ Redis

Returns a new instance of Redis.



10
11
12
13
 
# File 'lib/corpshort/backends/redis.rb', line 10

def initialize(redis: ::Redis.method(:current), prefix: "corpshort:")
  @redis = redis
  @prefix = prefix
end

Instance Attribute Details

#prefix ⇒ Object (readonly)

Returns the value of attribute prefix.



15
16
17
 
# File 'lib/corpshort/backends/redis.rb', line 15

def prefix
  @prefix
end

Instance Method Details



50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/corpshort/backends/redis.rb', line 50

def delete_link(link)
  key = link_key(link)
  redis.watch(key) do
    url = redis.hget(key, 'url')
    link_url_key = url_key(url)

    redis.multi do |m|
      m.zrem(links_key, link)
      m.zrem(link_url_key, link)
      m.del(key)
    end
  end
end

#get_link(name) ⇒ Object 



41
42
43
44
45
46
47
48
 
# File 'lib/corpshort/backends/redis.rb', line 41

def get_link(name)
  data = redis.hgetall(link_key(name))
  if data && !data.empty?
    Link.new(data, backend: self)
  else
    nil
  end
end

#list_links(token: nil, limit: 30) ⇒ Object 



87
88
89
90
91
92
93
94
95
 
# File 'lib/corpshort/backends/redis.rb', line 87

def list_links(token: nil, limit: 30)
  names = if token
            redis.zrevrangebyscore(links_key, "(#{token}", '-inf', limit: [0, limit], with_scores: true)
          else
            redis.zrevrangebyscore(links_key, '+inf', '-inf', limit: [0, limit], with_scores: true)
          end

  [names.map(&:first), names[-1]&.last]
end

#list_links_by_url(url) ⇒ Object 



83
84
85
 
# File 'lib/corpshort/backends/redis.rb', line 83

def list_links_by_url(url)
  redis.zrevrangebyscore(url_key(url), '+inf', '-inf')
end

#put_link(link, create_only: false) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/corpshort/backends/redis.rb', line 17

def put_link(link, create_only: false)
  key = link_key(link)

  redis.watch(key) do 
    old_url = redis.hget(key, 'url')

    if create_only && old_url
      redis.unwatch(key)
      raise ConflictError, "#{link.name} already exists"
    end

    redis.multi do |m|
      m.del(key)
      m.mapped_hmset(key, link.as_json)
      m.zadd(links_key, link.updated_at.to_i, link.name)

      if old_url && link.url != old_url
        m.zrem(url_key(old_url), link.name)
      end
      m.zadd(url_key(link.url), link.updated_at.to_i, link.name)
    end
  end
end

#redis ⇒ Object 



97
98
99
 
# File 'lib/corpshort/backends/redis.rb', line 97

def redis
  Thread.current[redis_thread_key] ||= @redis.call
end

#rename_link(link, new_name) ⇒ Object 



64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/corpshort/backends/redis.rb', line 64

def rename_link(link, new_name)
  link_key = link_key(link)
  new_key = link_key(new_name)

  redis.watch(link_key) do
    url = redis.hget(link_key, 'url')
    link_url_key = url_key(url)

    redis.multi do |m|
      m.renamenx(link_key, new_key)
      m.hset(new_key, 'name', new_name)
      m.zrem(links_key, link.name)
      m.zadd(links_key, link.updated_at.to_i, new_name)
      m.zrem(link_url_key, link.name)
      m.zadd(link_url_key, link.updated_at.to_i, new_name)
    end
  end
end