sensu-em 2.4.0-x86-mingw32

Sign up to get free protection for your applications and to get access to all the features.
Files changed (178) hide show
  1. checksums.yaml +7 -0
  2. data/.gitignore +21 -0
  3. data/.travis.yml +12 -0
  4. data/.yardopts +7 -0
  5. data/CHANGELOG.md +33 -0
  6. data/GNU +281 -0
  7. data/Gemfile +2 -0
  8. data/LICENSE +60 -0
  9. data/README.md +109 -0
  10. data/Rakefile +20 -0
  11. data/docs/DocumentationGuidesIndex.md +27 -0
  12. data/docs/GettingStarted.md +521 -0
  13. data/docs/old/ChangeLog +211 -0
  14. data/docs/old/DEFERRABLES +246 -0
  15. data/docs/old/EPOLL +141 -0
  16. data/docs/old/INSTALL +13 -0
  17. data/docs/old/KEYBOARD +42 -0
  18. data/docs/old/LEGAL +25 -0
  19. data/docs/old/LIGHTWEIGHT_CONCURRENCY +130 -0
  20. data/docs/old/PURE_RUBY +75 -0
  21. data/docs/old/RELEASE_NOTES +94 -0
  22. data/docs/old/SMTP +4 -0
  23. data/docs/old/SPAWNED_PROCESSES +148 -0
  24. data/docs/old/TODO +8 -0
  25. data/eventmachine.gemspec +37 -0
  26. data/examples/guides/getting_started/01_eventmachine_echo_server.rb +18 -0
  27. data/examples/guides/getting_started/02_eventmachine_echo_server_that_recognizes_exit_command.rb +22 -0
  28. data/examples/guides/getting_started/03_simple_chat_server.rb +149 -0
  29. data/examples/guides/getting_started/04_simple_chat_server_step_one.rb +27 -0
  30. data/examples/guides/getting_started/05_simple_chat_server_step_two.rb +43 -0
  31. data/examples/guides/getting_started/06_simple_chat_server_step_three.rb +98 -0
  32. data/examples/guides/getting_started/07_simple_chat_server_step_four.rb +121 -0
  33. data/examples/guides/getting_started/08_simple_chat_server_step_five.rb +141 -0
  34. data/examples/old/ex_channel.rb +43 -0
  35. data/examples/old/ex_queue.rb +2 -0
  36. data/examples/old/ex_tick_loop_array.rb +15 -0
  37. data/examples/old/ex_tick_loop_counter.rb +32 -0
  38. data/examples/old/helper.rb +2 -0
  39. data/ext/binder.cpp +124 -0
  40. data/ext/binder.h +46 -0
  41. data/ext/cmain.cpp +887 -0
  42. data/ext/ed.cpp +1992 -0
  43. data/ext/ed.h +424 -0
  44. data/ext/em.cpp +2352 -0
  45. data/ext/em.h +253 -0
  46. data/ext/eventmachine.h +128 -0
  47. data/ext/extconf.rb +179 -0
  48. data/ext/fastfilereader/extconf.rb +103 -0
  49. data/ext/fastfilereader/mapper.cpp +214 -0
  50. data/ext/fastfilereader/mapper.h +59 -0
  51. data/ext/fastfilereader/rubymain.cpp +127 -0
  52. data/ext/kb.cpp +79 -0
  53. data/ext/page.cpp +107 -0
  54. data/ext/page.h +51 -0
  55. data/ext/pipe.cpp +347 -0
  56. data/ext/project.h +161 -0
  57. data/ext/rubymain.cpp +1318 -0
  58. data/ext/ssl.cpp +476 -0
  59. data/ext/ssl.h +95 -0
  60. data/java/.classpath +6 -0
  61. data/java/.gitignore +1 -0
  62. data/java/.project +17 -0
  63. data/java/src/com/rubyeventmachine/DatagramPacket.java +13 -0
  64. data/java/src/com/rubyeventmachine/EmReactor.java +531 -0
  65. data/java/src/com/rubyeventmachine/EmReactorException.java +40 -0
  66. data/java/src/com/rubyeventmachine/EventCallback.java +7 -0
  67. data/java/src/com/rubyeventmachine/EventCode.java +26 -0
  68. data/java/src/com/rubyeventmachine/EventableChannel.java +130 -0
  69. data/java/src/com/rubyeventmachine/EventableDatagramChannel.java +179 -0
  70. data/java/src/com/rubyeventmachine/EventableSocketChannel.java +405 -0
  71. data/java/src/com/rubyeventmachine/SslBox.java +311 -0
  72. data/lib/em/buftok.rb +110 -0
  73. data/lib/em/callback.rb +58 -0
  74. data/lib/em/channel.rb +64 -0
  75. data/lib/em/completion.rb +304 -0
  76. data/lib/em/connection.rb +716 -0
  77. data/lib/em/deferrable.rb +210 -0
  78. data/lib/em/deferrable/pool.rb +2 -0
  79. data/lib/em/file_watch.rb +73 -0
  80. data/lib/em/future.rb +61 -0
  81. data/lib/em/iterator.rb +231 -0
  82. data/lib/em/messages.rb +66 -0
  83. data/lib/em/pool.rb +151 -0
  84. data/lib/em/process_watch.rb +45 -0
  85. data/lib/em/processes.rb +123 -0
  86. data/lib/em/protocols.rb +37 -0
  87. data/lib/em/protocols/header_and_content.rb +138 -0
  88. data/lib/em/protocols/httpclient.rb +279 -0
  89. data/lib/em/protocols/httpclient2.rb +600 -0
  90. data/lib/em/protocols/line_and_text.rb +125 -0
  91. data/lib/em/protocols/line_protocol.rb +29 -0
  92. data/lib/em/protocols/linetext2.rb +161 -0
  93. data/lib/em/protocols/memcache.rb +331 -0
  94. data/lib/em/protocols/object_protocol.rb +46 -0
  95. data/lib/em/protocols/postgres3.rb +246 -0
  96. data/lib/em/protocols/saslauth.rb +175 -0
  97. data/lib/em/protocols/smtpclient.rb +365 -0
  98. data/lib/em/protocols/smtpserver.rb +643 -0
  99. data/lib/em/protocols/socks4.rb +66 -0
  100. data/lib/em/protocols/stomp.rb +205 -0
  101. data/lib/em/protocols/tcptest.rb +54 -0
  102. data/lib/em/pure_ruby.rb +1017 -0
  103. data/lib/em/queue.rb +71 -0
  104. data/lib/em/resolver.rb +209 -0
  105. data/lib/em/spawnable.rb +84 -0
  106. data/lib/em/streamer.rb +118 -0
  107. data/lib/em/threaded_resource.rb +90 -0
  108. data/lib/em/tick_loop.rb +85 -0
  109. data/lib/em/timers.rb +61 -0
  110. data/lib/em/version.rb +3 -0
  111. data/lib/eventmachine.rb +1553 -0
  112. data/lib/fastfilereaderext.rb +2 -0
  113. data/lib/jeventmachine.rb +321 -0
  114. data/lib/rubyeventmachine.rb +2 -0
  115. data/rakelib/cpp.rake_example +77 -0
  116. data/rakelib/package.rake +98 -0
  117. data/rakelib/test.rake +8 -0
  118. data/tests/client.crt +31 -0
  119. data/tests/client.key +51 -0
  120. data/tests/em_test_helper.rb +64 -0
  121. data/tests/server.crt +36 -0
  122. data/tests/server.key +51 -0
  123. data/tests/test_attach.rb +150 -0
  124. data/tests/test_basic.rb +294 -0
  125. data/tests/test_channel.rb +62 -0
  126. data/tests/test_completion.rb +177 -0
  127. data/tests/test_connection_count.rb +53 -0
  128. data/tests/test_defer.rb +18 -0
  129. data/tests/test_deferrable.rb +35 -0
  130. data/tests/test_epoll.rb +145 -0
  131. data/tests/test_error_handler.rb +38 -0
  132. data/tests/test_exc.rb +28 -0
  133. data/tests/test_file_watch.rb +65 -0
  134. data/tests/test_futures.rb +170 -0
  135. data/tests/test_get_sock_opt.rb +37 -0
  136. data/tests/test_handler_check.rb +35 -0
  137. data/tests/test_hc.rb +155 -0
  138. data/tests/test_httpclient.rb +190 -0
  139. data/tests/test_httpclient2.rb +133 -0
  140. data/tests/test_idle_connection.rb +25 -0
  141. data/tests/test_inactivity_timeout.rb +54 -0
  142. data/tests/test_iterator.rb +97 -0
  143. data/tests/test_kb.rb +34 -0
  144. data/tests/test_line_protocol.rb +33 -0
  145. data/tests/test_ltp.rb +138 -0
  146. data/tests/test_ltp2.rb +288 -0
  147. data/tests/test_next_tick.rb +104 -0
  148. data/tests/test_object_protocol.rb +36 -0
  149. data/tests/test_pause.rb +102 -0
  150. data/tests/test_pending_connect_timeout.rb +52 -0
  151. data/tests/test_pool.rb +194 -0
  152. data/tests/test_process_watch.rb +48 -0
  153. data/tests/test_processes.rb +128 -0
  154. data/tests/test_proxy_connection.rb +180 -0
  155. data/tests/test_pure.rb +88 -0
  156. data/tests/test_queue.rb +50 -0
  157. data/tests/test_resolver.rb +55 -0
  158. data/tests/test_running.rb +14 -0
  159. data/tests/test_sasl.rb +47 -0
  160. data/tests/test_send_file.rb +217 -0
  161. data/tests/test_servers.rb +33 -0
  162. data/tests/test_set_sock_opt.rb +37 -0
  163. data/tests/test_shutdown_hooks.rb +23 -0
  164. data/tests/test_smtpclient.rb +55 -0
  165. data/tests/test_smtpserver.rb +57 -0
  166. data/tests/test_spawn.rb +293 -0
  167. data/tests/test_ssl_args.rb +78 -0
  168. data/tests/test_ssl_echo_data.rb +60 -0
  169. data/tests/test_ssl_methods.rb +56 -0
  170. data/tests/test_ssl_verify.rb +82 -0
  171. data/tests/test_stomp.rb +37 -0
  172. data/tests/test_system.rb +42 -0
  173. data/tests/test_threaded_resource.rb +53 -0
  174. data/tests/test_tick_loop.rb +59 -0
  175. data/tests/test_timers.rb +123 -0
  176. data/tests/test_ud.rb +8 -0
  177. data/tests/test_unbind_reason.rb +48 -0
  178. metadata +300 -0
@@ -0,0 +1,125 @@
1
+ #--
2
+ #
3
+ # Author:: Francis Cianfrocca (gmail: blackhedd)
4
+ # Homepage:: http://rubyeventmachine.com
5
+ # Date:: 15 November 2006
6
+ #
7
+ # See EventMachine and EventMachine::Connection for documentation and
8
+ # usage examples.
9
+ #
10
+ #----------------------------------------------------------------------------
11
+ #
12
+ # Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
13
+ # Gmail: blackhedd
14
+ #
15
+ # This program is free software; you can redistribute it and/or modify
16
+ # it under the terms of either: 1) the GNU General Public License
17
+ # as published by the Free Software Foundation; either version 2 of the
18
+ # License, or (at your option) any later version; or 2) Ruby's License.
19
+ #
20
+ # See the file COPYING for complete licensing information.
21
+ #
22
+ #---------------------------------------------------------------------------
23
+ #
24
+ #
25
+ #
26
+
27
+ module EventMachine
28
+ module Protocols
29
+ # A protocol that handles line-oriented data with interspersed binary text.
30
+ #
31
+ # This version is optimized for performance. See EventMachine::Protocols::LineText2
32
+ # for a version which is optimized for correctness with regard to binary text blocks
33
+ # that can switch back to line mode.
34
+ class LineAndTextProtocol < Connection
35
+ MaxLineLength = 16*1024
36
+ MaxBinaryLength = 32*1024*1024
37
+
38
+ def initialize *args
39
+ super
40
+ lbp_init_line_state
41
+ end
42
+ def receive_data data
43
+ if @lbp_mode == :lines
44
+ begin
45
+ @lpb_buffer.extract(data).each do |line|
46
+ receive_line(line.chomp) if respond_to?(:receive_line)
47
+ end
48
+ rescue Exception
49
+ receive_error('overlength line') if respond_to?(:receive_error)
50
+ close_connection
51
+ return
52
+ end
53
+ else
54
+ if @lbp_binary_limit > 0
55
+ wanted = @lbp_binary_limit - @lbp_binary_bytes_received
56
+ chunk = nil
57
+ if data.length > wanted
58
+ chunk = data.slice!(0...wanted)
59
+ else
60
+ chunk = data
61
+ data = ""
62
+ end
63
+ @lbp_binary_buffer[@lbp_binary_bytes_received...(@lbp_binary_bytes_received+chunk.length)] = chunk
64
+ @lbp_binary_bytes_received += chunk.length
65
+ if @lbp_binary_bytes_received == @lbp_binary_limit
66
+ receive_binary_data(@lbp_binary_buffer) if respond_to?(:receive_binary_data)
67
+ lbp_init_line_state
68
+ end
69
+ receive_data(data) if data.length > 0
70
+ else
71
+ receive_binary_data(data) if respond_to?(:receive_binary_data)
72
+ data = ""
73
+ end
74
+ end
75
+ end
76
+
77
+ def unbind
78
+ if @lbp_mode == :binary and @lbp_binary_limit > 0
79
+ if respond_to?(:receive_binary_data)
80
+ receive_binary_data( @lbp_binary_buffer[0...@lbp_binary_bytes_received] )
81
+ end
82
+ end
83
+ end
84
+
85
+ # Set up to read the supplied number of binary bytes.
86
+ # This recycles all the data currently waiting in the line buffer, if any.
87
+ # If the limit is nil, then ALL subsequent data will be treated as binary
88
+ # data and passed to the upstream protocol handler as we receive it.
89
+ # If a limit is given, we'll hold the incoming binary data and not
90
+ # pass it upstream until we've seen it all, or until there is an unbind
91
+ # (in which case we'll pass up a partial).
92
+ # Specifying nil for the limit (the default) means there is no limit.
93
+ # Specifiyng zero for the limit will cause an immediate transition back to line mode.
94
+ #
95
+ def set_binary_mode size = nil
96
+ if @lbp_mode == :lines
97
+ if size == 0
98
+ receive_binary_data("") if respond_to?(:receive_binary_data)
99
+ # Do no more work here. Stay in line mode and keep consuming data.
100
+ else
101
+ @lbp_binary_limit = size.to_i # (nil will be stored as zero)
102
+ if @lbp_binary_limit > 0
103
+ raise "Overlength" if @lbp_binary_limit > MaxBinaryLength # arbitrary sanity check
104
+ @lbp_binary_buffer = "\0" * @lbp_binary_limit
105
+ @lbp_binary_bytes_received = 0
106
+ end
107
+
108
+ @lbp_mode = :binary
109
+ receive_data @lpb_buffer.flush
110
+ end
111
+ else
112
+ raise "invalid operation"
113
+ end
114
+ end
115
+
116
+ #--
117
+ # For internal use, establish protocol baseline for handling lines.
118
+ def lbp_init_line_state
119
+ @lpb_buffer = BufferedTokenizer.new("\n", MaxLineLength)
120
+ @lbp_mode = :lines
121
+ end
122
+ private :lbp_init_line_state
123
+ end
124
+ end
125
+ end
@@ -0,0 +1,29 @@
1
+ module EventMachine
2
+ module Protocols
3
+ # LineProtocol will parse out newline terminated strings from a receive_data stream
4
+ #
5
+ # module Server
6
+ # include EM::P::LineProtocol
7
+ #
8
+ # def receive_line(line)
9
+ # send_data("you said: #{line}")
10
+ # end
11
+ # end
12
+ #
13
+ module LineProtocol
14
+ # @private
15
+ def receive_data data
16
+ (@buf ||= '') << data
17
+
18
+ while @buf.slice!(/(.*?)\r?\n/)
19
+ receive_line($1)
20
+ end
21
+ end
22
+
23
+ # Invoked with lines received over the network
24
+ def receive_line(line)
25
+ # stub
26
+ end
27
+ end
28
+ end
29
+ end
@@ -0,0 +1,161 @@
1
+ #--
2
+ #
3
+ # Author:: Francis Cianfrocca (gmail: blackhedd)
4
+ # Homepage:: http://rubyeventmachine.com
5
+ # Date:: 15 November 2006
6
+ #
7
+ # See EventMachine and EventMachine::Connection for documentation and
8
+ # usage examples.
9
+ #
10
+ #----------------------------------------------------------------------------
11
+ #
12
+ # Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
13
+ # Gmail: blackhedd
14
+ #
15
+ # This program is free software; you can redistribute it and/or modify
16
+ # it under the terms of either: 1) the GNU General Public License
17
+ # as published by the Free Software Foundation; either version 2 of the
18
+ # License, or (at your option) any later version; or 2) Ruby's License.
19
+ #
20
+ # See the file COPYING for complete licensing information.
21
+ #
22
+ #---------------------------------------------------------------------------
23
+ #
24
+ #
25
+
26
+ module EventMachine
27
+ module Protocols
28
+ # In the grand, time-honored tradition of re-inventing the wheel, we offer
29
+ # here YET ANOTHER protocol that handles line-oriented data with interspersed
30
+ # binary text. This one trades away some of the performance optimizations of
31
+ # EventMachine::Protocols::LineAndTextProtocol in order to get better correctness
32
+ # with regard to binary text blocks that can switch back to line mode. It also
33
+ # permits the line-delimiter to change in midstream.
34
+ # This was originally written to support Stomp.
35
+ module LineText2
36
+ # TODO! We're not enforcing the limits on header lengths and text-lengths.
37
+ # When we get around to that, call #receive_error if the user defined it, otherwise
38
+ # throw exceptions.
39
+
40
+ MaxLineLength = 16*1024
41
+ MaxBinaryLength = 32*1024*1024
42
+
43
+ #--
44
+ # Will be called recursively until there's no data to read.
45
+ # That way the user-defined handlers we call can modify the
46
+ # handling characteristics on a per-token basis.
47
+ #
48
+ def receive_data data
49
+ return unless (data and data.length > 0)
50
+
51
+ # Do this stuff in lieu of a constructor.
52
+ @lt2_mode ||= :lines
53
+ @lt2_delimiter ||= "\n"
54
+ @lt2_linebuffer ||= []
55
+
56
+ if @lt2_mode == :lines
57
+ if ix = data.index( @lt2_delimiter )
58
+ @lt2_linebuffer << data[0...ix]
59
+ ln = @lt2_linebuffer.join
60
+ @lt2_linebuffer.clear
61
+ if @lt2_delimiter == "\n"
62
+ ln.chomp!
63
+ end
64
+ receive_line ln
65
+ receive_data data[(ix+@lt2_delimiter.length)..-1]
66
+ else
67
+ @lt2_linebuffer << data
68
+ end
69
+ elsif @lt2_mode == :text
70
+ if @lt2_textsize
71
+ needed = @lt2_textsize - @lt2_textpos
72
+ will_take = if data.length > needed
73
+ needed
74
+ else
75
+ data.length
76
+ end
77
+
78
+ @lt2_textbuffer << data[0...will_take]
79
+ tail = data[will_take..-1]
80
+
81
+ @lt2_textpos += will_take
82
+ if @lt2_textpos >= @lt2_textsize
83
+ # Reset line mode (the default behavior) BEFORE calling the
84
+ # receive_binary_data. This makes it possible for user code
85
+ # to call set_text_mode, enabling chains of text blocks
86
+ # (which can possibly be of different sizes).
87
+ set_line_mode
88
+ receive_binary_data @lt2_textbuffer.join
89
+ receive_end_of_binary_data
90
+ end
91
+
92
+ receive_data tail
93
+ else
94
+ receive_binary_data data
95
+ end
96
+ end
97
+ end
98
+
99
+
100
+ def set_delimiter delim
101
+ @lt2_delimiter = delim.to_s
102
+ end
103
+
104
+ # Called internally but also exposed to user code, for the case in which
105
+ # processing of binary data creates a need to transition back to line mode.
106
+ # We support an optional parameter to "throw back" some data, which might
107
+ # be an umprocessed chunk of the transmitted binary data, or something else
108
+ # entirely.
109
+ def set_line_mode data=""
110
+ @lt2_mode = :lines
111
+ (@lt2_linebuffer ||= []).clear
112
+ receive_data data.to_s
113
+ end
114
+
115
+ def set_text_mode size=nil
116
+ if size == 0
117
+ set_line_mode
118
+ else
119
+ @lt2_mode = :text
120
+ (@lt2_textbuffer ||= []).clear
121
+ @lt2_textsize = size # which can be nil, signifying no limit
122
+ @lt2_textpos = 0
123
+ end
124
+ end
125
+
126
+ # Alias for #set_text_mode, added for back-compatibility with LineAndTextProtocol.
127
+ def set_binary_mode size=nil
128
+ set_text_mode size
129
+ end
130
+
131
+ # In case of a dropped connection, we'll send a partial buffer to user code
132
+ # when in sized text mode. User overrides of #receive_binary_data need to
133
+ # be aware that they may get a short buffer.
134
+ def unbind
135
+ @lt2_mode ||= nil
136
+ if @lt2_mode == :text and @lt2_textpos > 0
137
+ receive_binary_data @lt2_textbuffer.join
138
+ end
139
+ end
140
+
141
+ # Stub. Should be subclassed by user code.
142
+ def receive_line ln
143
+ # no-op
144
+ end
145
+
146
+ # Stub. Should be subclassed by user code.
147
+ def receive_binary_data data
148
+ # no-op
149
+ end
150
+
151
+ # Stub. Should be subclassed by user code.
152
+ # This is called when transitioning internally from text mode
153
+ # back to line mode. Useful when client code doesn't want
154
+ # to keep track of how much data it's received.
155
+ def receive_end_of_binary_data
156
+ # no-op
157
+ end
158
+ end
159
+ end
160
+ end
161
+
@@ -0,0 +1,331 @@
1
+ module EventMachine
2
+ module Protocols
3
+ # Implements the Memcache protocol (http://code.sixapart.com/svn/memcached/trunk/server/doc/protocol.txt).
4
+ # Requires memcached >= 1.2.4 w/ noreply support
5
+ #
6
+ # == Usage example
7
+ #
8
+ # EM.run{
9
+ # cache = EM::P::Memcache.connect 'localhost', 11211
10
+ #
11
+ # cache.set :a, 'hello'
12
+ # cache.set :b, 'hi'
13
+ # cache.set :c, 'how are you?'
14
+ # cache.set :d, ''
15
+ #
16
+ # cache.get(:a){ |v| p v }
17
+ # cache.get_hash(:a, :b, :c, :d){ |v| p v }
18
+ # cache.get(:a,:b,:c,:d){ |a,b,c,d| p [a,b,c,d] }
19
+ #
20
+ # cache.get(:a,:z,:b,:y,:d){ |a,z,b,y,d| p [a,z,b,y,d] }
21
+ #
22
+ # cache.get(:missing){ |m| p [:missing=, m] }
23
+ # cache.set(:missing, 'abc'){ p :stored }
24
+ # cache.get(:missing){ |m| p [:missing=, m] }
25
+ # cache.del(:missing){ p :deleted }
26
+ # cache.get(:missing){ |m| p [:missing=, m] }
27
+ # }
28
+ #
29
+ module Memcache
30
+ include EM::Deferrable
31
+
32
+ ##
33
+ # constants
34
+
35
+ unless defined? Cempty
36
+ # @private
37
+ Cstored = 'STORED'.freeze
38
+ # @private
39
+ Cend = 'END'.freeze
40
+ # @private
41
+ Cdeleted = 'DELETED'.freeze
42
+ # @private
43
+ Cunknown = 'NOT_FOUND'.freeze
44
+ # @private
45
+ Cerror = 'ERROR'.freeze
46
+
47
+ # @private
48
+ Cempty = ''.freeze
49
+ # @private
50
+ Cdelimiter = "\r\n".freeze
51
+ end
52
+
53
+ ##
54
+ # commands
55
+
56
+ # Get the value associated with one or multiple keys
57
+ #
58
+ # cache.get(:a){ |v| p v }
59
+ # cache.get(:a,:b,:c,:d){ |a,b,c,d| p [a,b,c,d] }
60
+ #
61
+ def get *keys
62
+ raise ArgumentError unless block_given?
63
+
64
+ callback{
65
+ keys = keys.map{|k| k.to_s.gsub(/\s/,'_') }
66
+ send_data "get #{keys.join(' ')}\r\n"
67
+ @get_cbs << [keys, proc{ |values|
68
+ yield *keys.map{ |k| values[k] }
69
+ }]
70
+ }
71
+ end
72
+
73
+ # Set the value for a given key
74
+ #
75
+ # cache.set :a, 'hello'
76
+ # cache.set(:missing, 'abc'){ puts "stored the value!" }
77
+ #
78
+ def set key, val, exptime = 0, &cb
79
+ callback{
80
+ val = val.to_s
81
+ send_cmd :set, key, 0, exptime, val.respond_to?(:bytesize) ? val.bytesize : val.size, !block_given?
82
+ send_data val
83
+ send_data Cdelimiter
84
+ @set_cbs << cb if cb
85
+ }
86
+ end
87
+
88
+ # Gets multiple values as a hash
89
+ #
90
+ # cache.get_hash(:a, :b, :c, :d){ |h| puts h[:a] }
91
+ #
92
+ def get_hash *keys
93
+ raise ArgumentError unless block_given?
94
+
95
+ get *keys do |*values|
96
+ yield keys.inject({}){ |hash, k| hash.update k => values[keys.index(k)] }
97
+ end
98
+ end
99
+
100
+ # Delete the value associated with a key
101
+ #
102
+ # cache.del :a
103
+ # cache.del(:b){ puts "deleted the value!" }
104
+ #
105
+ def delete key, expires = 0, &cb
106
+ callback{
107
+ send_data "delete #{key} #{expires}#{cb ? '' : ' noreply'}\r\n"
108
+ @del_cbs << cb if cb
109
+ }
110
+ end
111
+ alias del delete
112
+
113
+ # Connect to a memcached server (must support NOREPLY, memcached >= 1.2.4)
114
+ def self.connect host = 'localhost', port = 11211
115
+ EM.connect host, port, self, host, port
116
+ end
117
+
118
+ def send_cmd cmd, key, flags = 0, exptime = 0, bytes = 0, noreply = false
119
+ send_data "#{cmd} #{key} #{flags} #{exptime} #{bytes}#{noreply ? ' noreply' : ''}\r\n"
120
+ end
121
+ private :send_cmd
122
+
123
+ ##
124
+ # errors
125
+
126
+ # @private
127
+ class ParserError < StandardError
128
+ end
129
+
130
+ ##
131
+ # em hooks
132
+
133
+ # @private
134
+ def initialize host, port = 11211
135
+ @host, @port = host, port
136
+ end
137
+
138
+ # @private
139
+ def connection_completed
140
+ @get_cbs = []
141
+ @set_cbs = []
142
+ @del_cbs = []
143
+
144
+ @values = {}
145
+
146
+ @reconnecting = false
147
+ @connected = true
148
+ succeed
149
+ # set_delimiter "\r\n"
150
+ # set_line_mode
151
+ end
152
+
153
+ #--
154
+ # 19Feb09 Switched to a custom parser, LineText2 is recursive and can cause
155
+ # stack overflows when there is too much data.
156
+ # include EM::P::LineText2
157
+ # @private
158
+ def receive_data data
159
+ (@buffer||='') << data
160
+
161
+ while index = @buffer.index(Cdelimiter)
162
+ begin
163
+ line = @buffer.slice!(0,index+2)
164
+ process_cmd line
165
+ rescue ParserError
166
+ @buffer[0...0] = line
167
+ break
168
+ end
169
+ end
170
+ end
171
+
172
+ #--
173
+ # def receive_line line
174
+ # @private
175
+ def process_cmd line
176
+ case line.strip
177
+ when /^VALUE\s+(.+?)\s+(\d+)\s+(\d+)/ # VALUE <key> <flags> <bytes>
178
+ bytes = Integer($3)
179
+ # set_binary_mode bytes+2
180
+ # @cur_key = $1
181
+ if @buffer.size >= bytes + 2
182
+ @values[$1] = @buffer.slice!(0,bytes)
183
+ @buffer.slice!(0,2) # \r\n
184
+ else
185
+ raise ParserError
186
+ end
187
+
188
+ when Cend # END
189
+ if entry = @get_cbs.shift
190
+ keys, cb = entry
191
+ cb.call(@values)
192
+ end
193
+ @values = {}
194
+
195
+ when Cstored # STORED
196
+ if cb = @set_cbs.shift
197
+ cb.call(true)
198
+ end
199
+
200
+ when Cdeleted # DELETED
201
+ if cb = @del_cbs.shift
202
+ cb.call(true)
203
+ end
204
+
205
+ when Cunknown # NOT_FOUND
206
+ if cb = @del_cbs.shift
207
+ cb.call(false)
208
+ end
209
+
210
+ else
211
+ p [:MEMCACHE_UNKNOWN, line]
212
+ end
213
+ end
214
+
215
+ #--
216
+ # def receive_binary_data data
217
+ # @values[@cur_key] = data[0..-3]
218
+ # end
219
+
220
+ # @private
221
+ def unbind
222
+ if @connected or @reconnecting
223
+ EM.add_timer(1){ reconnect @host, @port }
224
+ @connected = false
225
+ @reconnecting = true
226
+ @deferred_status = nil
227
+ else
228
+ raise 'Unable to connect to memcached server'
229
+ end
230
+ end
231
+ end
232
+ end
233
+ end
234
+
235
+ if __FILE__ == $0
236
+ # ruby -I ext:lib -r eventmachine -rubygems lib/protocols/memcache.rb
237
+ require 'em/spec'
238
+
239
+ # @private
240
+ class TestConnection
241
+ include EM::P::Memcache
242
+ def send_data data
243
+ sent_data << data
244
+ end
245
+ def sent_data
246
+ @sent_data ||= ''
247
+ end
248
+
249
+ def initialize
250
+ connection_completed
251
+ end
252
+ end
253
+
254
+ EM.describe EM::Protocols::Memcache do
255
+
256
+ before{
257
+ @c = TestConnection.new
258
+ }
259
+
260
+ should 'send get requests' do
261
+ @c.get('a'){}
262
+ @c.sent_data.should == "get a\r\n"
263
+ done
264
+ end
265
+
266
+ should 'send set requests' do
267
+ @c.set('a', 1){}
268
+ @c.sent_data.should == "set a 0 0 1\r\n1\r\n"
269
+ done
270
+ end
271
+
272
+ should 'use noreply on set without block' do
273
+ @c.set('a', 1)
274
+ @c.sent_data.should == "set a 0 0 1 noreply\r\n1\r\n"
275
+ done
276
+ end
277
+
278
+ should 'send delete requests' do
279
+ @c.del('a')
280
+ @c.sent_data.should == "delete a 0 noreply\r\n"
281
+ done
282
+ end
283
+
284
+ should 'work when get returns no values' do
285
+ @c.get('a'){ |a|
286
+ a.should.be.nil
287
+ done
288
+ }
289
+
290
+ @c.receive_data "END\r\n"
291
+ end
292
+
293
+ should 'invoke block on set' do
294
+ @c.set('a', 1){
295
+ done
296
+ }
297
+
298
+ @c.receive_data "STORED\r\n"
299
+ end
300
+
301
+ should 'invoke block on delete' do
302
+ @c.delete('a'){ |found|
303
+ found.should.be.false
304
+ }
305
+ @c.delete('b'){ |found|
306
+ found.should.be.true
307
+ done
308
+ }
309
+
310
+ @c.receive_data "NOT_FOUND\r\n"
311
+ @c.receive_data "DELETED\r\n"
312
+ end
313
+
314
+ should 'parse split responses' do
315
+ @c.get('a'){ |a|
316
+ a.should == 'abc'
317
+ done
318
+ }
319
+
320
+ @c.receive_data "VAL"
321
+ @c.receive_data "UE a 0 "
322
+ @c.receive_data "3\r\n"
323
+ @c.receive_data "ab"
324
+ @c.receive_data "c"
325
+ @c.receive_data "\r\n"
326
+ @c.receive_data "EN"
327
+ @c.receive_data "D\r\n"
328
+ end
329
+
330
+ end
331
+ end