Class: Oria::Server
- Inherits:
-
EventMachine::Connection
- Object
- EventMachine::Connection
- Oria::Server
- Defined in:
- lib/oria/server.rb
Constant Summary collapse
- @@timer =
nil
Class Method Summary collapse
- .debug? ⇒ Boolean
- .log(value, app_key = nil) ⇒ Object
- .logger ⇒ Object
- .running? ⇒ Boolean
- .start(server, port, debug = false, options = {}) ⇒ Object
- .stop ⇒ Object
- .store_hash(app_key) ⇒ Object
Instance Method Summary collapse
Class Method Details
.debug? ⇒ Boolean
9 10 11 |
# File 'lib/oria/server.rb', line 9 def debug? @@debug end |
.log(value, app_key = nil) ⇒ Object
13 14 15 16 17 18 19 |
# File 'lib/oria/server.rb', line 13 def log(value, app_key = nil) if debug? || @@options[:puts] value = "#{Time.now.strftime('%d %b %H:%M:%S')} - #{"#{app_key}: " if app_key}#{value}" logger.debug(value) if debug? puts value if @@options[:puts] end end |
.logger ⇒ Object
21 22 23 24 |
# File 'lib/oria/server.rb', line 21 def logger require 'logger' @@logger ||= Logger.new(log_file, 0, 100 * 1024 * 1024) end |
.running? ⇒ Boolean
26 27 28 29 30 |
# File 'lib/oria/server.rb', line 26 def running? pid && !!Process.getpgid(pid) rescue Errno::ESRCH false end |
.start(server, port, debug = false, options = {}) ⇒ Object
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
# File 'lib/oria/server.rb', line 32 def start(server, port, debug = false, = {}) @@debug = !!debug @@options = log "Server started" @@servers ||= if File.exists?(yaml_store) log "DB loaded from disk" YAML.load_file(yaml_store) else {} end pid = Process.pid EventMachine.run do EventMachine.start_server server, port, Oria::Server File.open(pid_file, 'w') do |file| file.puts pid end log "The server is now ready to accept connections on port #{port}" end end |
.stop ⇒ Object
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
# File 'lib/oria/server.rb', line 52 def stop response = false if File.exists?(pid_file) begin if pid Process.kill(9, pid) response = true end rescue Errno::ESRCH # TODO: We perform error reporting elsewhere. Still, # an actionless rescue doesn't seem like a great idea. ensure File.unlink(pid_file) @@pid = nil end end response end |
.store_hash(app_key) ⇒ Object
71 72 73 74 75 76 77 78 79 80 |
# File 'lib/oria/server.rb', line 71 def store_hash(app_key) @@timer.cancel if @@timer @@timer = EventMachine::Timer.new(10) do log "Writing changes to disk", app_key File.open(yaml_store, 'w') do |store| store.puts YAML.dump(@@servers) end @@timer = nil end end |
Instance Method Details
#post_init ⇒ Object
100 101 102 |
# File 'lib/oria/server.rb', line 100 def post_init Oria::Server.log "Client connected" end |
#receive_data(data) ⇒ Object
104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
# File 'lib/oria/server.rb', line 104 def receive_data(data) Oria::Server.log "Responding to #{data}" data = data.split(' ') method = data.shift data = JSON.parse(data.join(' ')) @app_key = data.delete('app_key') case method when 'GET' response = hash[data['key']] when 'DELETE' if data.empty? hash.clear else response = hash.delete(data['key']) end when 'PUT' if data.key?('key') hash[data['key']] = data['value'] response = data['value'] elsif data.key?('value') response = random_key hash[response] = data['value'] end end if defined?(response) response = JSON.generate({:response => response}) Oria::Server.log "Sending response: #{response}" Oria::Server.store_hash(@app_key) send_data response end rescue JSON::ParserError send_data "Invalid request" end |
#unbind ⇒ Object
138 139 |
# File 'lib/oria/server.rb', line 138 def unbind end |