Module: Riagent::Configuration

Included in:
Riagent
Defined in:
lib/riagent/configuration.rb

Instance Method Summary collapse

Instance Method Details

#config ⇒ Object 



25
26
27
 
# File 'lib/riagent/configuration.rb', line 25

def config
  @config ||= {}
end

#config=(value) ⇒ Object 



29
30
31
 
# File 'lib/riagent/configuration.rb', line 29

def config=(value)
  @config = value
end

#config_for(environment = :development) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/riagent/configuration.rb', line 33

def config_for(environment=:development)
  if self.config.present?
    env_config = self.config[environment.to_s]
  else
    env_config = {
      'host' => ENV['RIAK_HOST'],
      'http_port' => ENV['RIAK_HTTP_PORT'],
      'pb_port' => ENV['RIAK_PB_PORT']
    }
  end
  env_config
end

#init_clients(environment = :development) ⇒ Object 



46
47
48
 
# File 'lib/riagent/configuration.rb', line 46

def init_clients(environment=:development)
  self.init_riak_json_client(environment)
end

#init_riak_json_client(environment = :development) ⇒ Object 



50
51
52
53
54
 
# File 'lib/riagent/configuration.rb', line 50

def init_riak_json_client(environment=:development)
  env_config = self.config_for(environment)
  client = RiakJson::Client.new(env_config['host'], env_config['http_port'])
  self.riak_json_client = client
end

#load_config_file(config_file_path) ⇒ Object 



56
57
58
59
60
 
# File 'lib/riagent/configuration.rb', line 56

def load_config_file(config_file_path)
  config_file = File.expand_path(config_file_path)
  config_hash = YAML.load(ERB.new(File.read(config_file)).result)
  self.config = config_hash
end

#riak_json_client ⇒ RiakJson::Client

Returns The client for the current thread.

Returns:

  • (RiakJson::Client) —

    The client for the current thread.



63
64
65
 
# File 'lib/riagent/configuration.rb', line 63

def riak_json_client
  Thread.current[:riak_json_client] ||= nil
end

#riak_json_client=(value) ⇒ Object

Sets the client for the current thread.

Parameters:

  • value (RiakJson::Client) —

    the client



69
70
71
 
# File 'lib/riagent/configuration.rb', line 69

def riak_json_client=(value)
  Thread.current[:riak_json_client] = value
end