Class: EtcdDiscovery::Registrar

Inherits:
Object
  • Object
show all
Defined in:
lib/etcd-discovery/registrar.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(service, host) ⇒ Registrar

Returns a new instance of Registrar.



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/etcd-discovery/registrar.rb', line 24

def initialize(service, host)
  @logger = Logger.new($stdout)

  if host.is_a? Hash
    @host = Host.new host.merge('service_name' => service, 'uuid' => SecureRandom.uuid)
  elsif host.is_a? EtcdDiscovery::Host
    @host = host
  else
    raise TypeError, "host should be a Hash or a Etcd::Host, is a #{host.class}"
  end

  @service = EtcdDiscovery::Service.new service_params
  @state = :new
  @user     = @host.attributes['user']
  @password = @host.attributes['password']
end

Instance Attribute Details

#host ⇒ Object (readonly)

Returns the value of attribute host.



19
20
21
 
# File 'lib/etcd-discovery/registrar.rb', line 19

def host
  @host
end

#password ⇒ Object (readonly)

Returns the value of attribute password.



22
23
24
 
# File 'lib/etcd-discovery/registrar.rb', line 22

def password
  @password
end

#service ⇒ Object (readonly)

Returns the value of attribute service.



20
21
22
 
# File 'lib/etcd-discovery/registrar.rb', line 20

def service
  @service
end

#state ⇒ Object (readonly)

Returns the value of attribute state.



16
17
18
 
# File 'lib/etcd-discovery/registrar.rb', line 16

def state
  @state
end

#thread ⇒ Object (readonly)

Returns the value of attribute thread.



17
18
19
 
# File 'lib/etcd-discovery/registrar.rb', line 17

def thread
  @thread
end

#user ⇒ Object (readonly)

Returns the value of attribute user.



21
22
23
 
# File 'lib/etcd-discovery/registrar.rb', line 21

def user
  @user
end

#watcher ⇒ Object (readonly)

Returns the value of attribute watcher.



18
19
20
 
# File 'lib/etcd-discovery/registrar.rb', line 18

def watcher
  @watcher
end

Instance Method Details

#client ⇒ Object 



95
96
97
 
# File 'lib/etcd-discovery/registrar.rb', line 95

def client
  config.client
end

#config ⇒ Object 



99
100
101
 
# File 'lib/etcd-discovery/registrar.rb', line 99

def config
  EtcdDiscovery.config
end

#host_key ⇒ Object 



103
104
105
 
# File 'lib/etcd-discovery/registrar.rb', line 103

def host_key
  "/services/#{@service.attributes['name']}/#{@host.attributes['uuid']}"
end

#register ⇒ Object 



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
 
# File 'lib/etcd-discovery/registrar.rb', line 41

def register
  if @state == :started
    @logger.warn "#{@service} is already registered"
    return
  end

  @state = :started

  service_value = @service.to_json

  # Do not start credentials synchro if the service is not public or has no credentials
  if @service.attributes['public'] && (@service.attributes['user'].present? || @service.attributes['password'].present?)
    @watcher = Thread.new {
      @logger.warn "Watcher #{@service.attributes['name']} started"
      index = 0
      while @state == :started
        resp = client.watch service_key, {index: index}
        value = JSON.parse resp.node.value
        @user = value['user']
        @password = value['password']
        @host.set_credentials user, password
        @service.set_credentials user, password
        index = resp.etcd_index
      end
    }
  end

  client.set(service_key, value: service_value)
  @thread = Thread.new {
    @logger.warn "Register '#{@service}' started"
    while @state == :started
      value = @host.to_json
      begin
        client.set(host_key, value: value, ttl: config.register_ttl)
      rescue => e
        @logger.warn "Fail to set #{service_key}: #{e}, #{e.message}, #{e.class}"
      end
      sleep config.register_renew
    end
    @logger.warn "Register '#{@service}' stopped"
  }


  return self
end

#service_key ⇒ Object 



107
108
109
 
# File 'lib/etcd-discovery/registrar.rb', line 107

def service_key
  "/services_infos/#{@service.attributes['name']}"
end

#service_params ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/etcd-discovery/registrar.rb', line 111

def service_params
  params = {
    'name' =>     host.attributes['service_name'],
    'critical' => host.attributes['critical'],
    'user' =>     host.attributes['user'],
    'password' => host.attributes['password'],
    'public' =>   host.attributes['public']
  }
  params['hostname'] = host.attributes['name']if params['public']
  params['ports'] = host.attributes['ports']  if params['public']
  return params
end

#stop ⇒ Object

Raises:



87
88
89
90
91
92
93
 
# File 'lib/etcd-discovery/registrar.rb', line 87

def stop
  raise InvalidStateError.new(@state, :started) if @state != :started
  @logger.debug "Set state to :stopped"
  @state = :stopped
  @logger.debug "Delete #{key}"
  client.delete(key)
end