kafka_syrup 0.9.0

Sign up to get free protection for your applications and to get access to all the features.
@@ -0,0 +1,59 @@
1
+ module KafkaSyrup
2
+ class TopicProducer
3
+ include Utils
4
+
5
+ attr_accessor :topic, :brokers, :partitions, :partitioner, :opts
6
+
7
+ def initialize(topic, *args)
8
+ self.topic = topic
9
+ self.opts = args.last || {}
10
+
11
+ self.brokers = {}
12
+ self.partitions = {}
13
+
14
+ refresh_metadata
15
+
16
+ self.partitioner ||= ->(msg) do
17
+ msg.hash % partitions.count
18
+ end
19
+ end
20
+
21
+ def refresh_metadata
22
+ brokers.each{ |id, broker| broker && broker.socket && broker.socket.close }
23
+
24
+ self.brokers.clear
25
+ self.partitions.clear
26
+
27
+ meta = KafkaSyrup.get_metadata(topic)
28
+
29
+ meta.brokers.each do |broker|
30
+ broker.extend KafkaSyrup::Broker::Communications
31
+ brokers.store broker.node, broker
32
+ end
33
+
34
+ meta.topics.first.partitions.each do |partition|
35
+ partitions.store partition.id, partition
36
+ end
37
+ end
38
+
39
+ def send_message(msg)
40
+ id = partitioner.call(msg)
41
+
42
+ request = KafkaSyrup::Protocol::ProduceRequest.new
43
+ request.add_topic(topic).add_partition(id).add_message(msg)
44
+
45
+ broker_for_id(id).send_request(request)
46
+
47
+ rescue KafkaResponseErrors::NotLeaderForPartition, SocketReadError
48
+ refresh_metadata
49
+
50
+ retry
51
+ end
52
+
53
+ private
54
+
55
+ def broker_for_id(id)
56
+ brokers[partitions[id].leader]
57
+ end
58
+ end
59
+ end
@@ -0,0 +1,15 @@
1
+ module KafkaSyrup
2
+ module Utils
3
+ def load_args(*args)
4
+ if args.last.is_a?(Hash)
5
+ args.last.each do |attr, val|
6
+ self.send("#{attr}=", val) if self.respond_to?("#{attr}=")
7
+ end
8
+ end
9
+ end
10
+
11
+ def log
12
+ KafkaSyrup.config.logger
13
+ end
14
+ end
15
+ end
@@ -0,0 +1,3 @@
1
+ module KafkaSyrup
2
+ VERSION = "0.9.0"
3
+ end
metadata ADDED
@@ -0,0 +1,188 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: kafka_syrup
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.9.0
5
+ platform: ruby
6
+ authors:
7
+ - Delbert Mitten
8
+ autorequire:
9
+ bindir: exe
10
+ cert_chain: []
11
+ date: 2015-06-30 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: zk
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - "~>"
18
+ - !ruby/object:Gem::Version
19
+ version: '1.9'
20
+ type: :runtime
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - "~>"
25
+ - !ruby/object:Gem::Version
26
+ version: '1.9'
27
+ - !ruby/object:Gem::Dependency
28
+ name: multi_json
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - "~>"
32
+ - !ruby/object:Gem::Version
33
+ version: '1.8'
34
+ type: :runtime
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - "~>"
39
+ - !ruby/object:Gem::Version
40
+ version: '1.8'
41
+ - !ruby/object:Gem::Dependency
42
+ name: bundler
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - "~>"
46
+ - !ruby/object:Gem::Version
47
+ version: '1.9'
48
+ type: :development
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - "~>"
53
+ - !ruby/object:Gem::Version
54
+ version: '1.9'
55
+ - !ruby/object:Gem::Dependency
56
+ name: rake
57
+ requirement: !ruby/object:Gem::Requirement
58
+ requirements:
59
+ - - "~>"
60
+ - !ruby/object:Gem::Version
61
+ version: '10.0'
62
+ type: :development
63
+ prerelease: false
64
+ version_requirements: !ruby/object:Gem::Requirement
65
+ requirements:
66
+ - - "~>"
67
+ - !ruby/object:Gem::Version
68
+ version: '10.0'
69
+ - !ruby/object:Gem::Dependency
70
+ name: minitest
71
+ requirement: !ruby/object:Gem::Requirement
72
+ requirements:
73
+ - - "~>"
74
+ - !ruby/object:Gem::Version
75
+ version: '5.6'
76
+ type: :development
77
+ prerelease: false
78
+ version_requirements: !ruby/object:Gem::Requirement
79
+ requirements:
80
+ - - "~>"
81
+ - !ruby/object:Gem::Version
82
+ version: '5.6'
83
+ - !ruby/object:Gem::Dependency
84
+ name: minitest-reporters
85
+ requirement: !ruby/object:Gem::Requirement
86
+ requirements:
87
+ - - "~>"
88
+ - !ruby/object:Gem::Version
89
+ version: '1.0'
90
+ type: :development
91
+ prerelease: false
92
+ version_requirements: !ruby/object:Gem::Requirement
93
+ requirements:
94
+ - - "~>"
95
+ - !ruby/object:Gem::Version
96
+ version: '1.0'
97
+ - !ruby/object:Gem::Dependency
98
+ name: mocha
99
+ requirement: !ruby/object:Gem::Requirement
100
+ requirements:
101
+ - - "~>"
102
+ - !ruby/object:Gem::Version
103
+ version: '1.1'
104
+ type: :development
105
+ prerelease: false
106
+ version_requirements: !ruby/object:Gem::Requirement
107
+ requirements:
108
+ - - "~>"
109
+ - !ruby/object:Gem::Version
110
+ version: '1.1'
111
+ - !ruby/object:Gem::Dependency
112
+ name: pry
113
+ requirement: !ruby/object:Gem::Requirement
114
+ requirements:
115
+ - - "~>"
116
+ - !ruby/object:Gem::Version
117
+ version: '0.10'
118
+ type: :development
119
+ prerelease: false
120
+ version_requirements: !ruby/object:Gem::Requirement
121
+ requirements:
122
+ - - "~>"
123
+ - !ruby/object:Gem::Version
124
+ version: '0.10'
125
+ description: A high level Kafka client that supports producer, low level consumers,
126
+ and high level consumers.
127
+ email:
128
+ - drmitten@gmail.com
129
+ executables: []
130
+ extensions: []
131
+ extra_rdoc_files: []
132
+ files:
133
+ - ".gitignore"
134
+ - ".travis.yml"
135
+ - Gemfile
136
+ - LICENSE
137
+ - README.md
138
+ - Rakefile
139
+ - kafka_syrup.gemspec
140
+ - lib/kafka_syrup.rb
141
+ - lib/kafka_syrup/broker.rb
142
+ - lib/kafka_syrup/encoding.rb
143
+ - lib/kafka_syrup/errors.rb
144
+ - lib/kafka_syrup/partition_consumer.rb
145
+ - lib/kafka_syrup/protocol.rb
146
+ - lib/kafka_syrup/protocol/base.rb
147
+ - lib/kafka_syrup/protocol/errors.rb
148
+ - lib/kafka_syrup/protocol/fetch_request.rb
149
+ - lib/kafka_syrup/protocol/fetch_response.rb
150
+ - lib/kafka_syrup/protocol/message.rb
151
+ - lib/kafka_syrup/protocol/message_set.rb
152
+ - lib/kafka_syrup/protocol/metadata_request.rb
153
+ - lib/kafka_syrup/protocol/metadata_response.rb
154
+ - lib/kafka_syrup/protocol/offset_request.rb
155
+ - lib/kafka_syrup/protocol/offset_response.rb
156
+ - lib/kafka_syrup/protocol/produce_request.rb
157
+ - lib/kafka_syrup/protocol/produce_response.rb
158
+ - lib/kafka_syrup/protocol/request.rb
159
+ - lib/kafka_syrup/protocol/response.rb
160
+ - lib/kafka_syrup/topic_consumer.rb
161
+ - lib/kafka_syrup/topic_producer.rb
162
+ - lib/kafka_syrup/utils.rb
163
+ - lib/kafka_syrup/version.rb
164
+ homepage: https://github.com/drmitten/kafka_syrup
165
+ licenses:
166
+ - MIT
167
+ metadata: {}
168
+ post_install_message:
169
+ rdoc_options: []
170
+ require_paths:
171
+ - lib
172
+ required_ruby_version: !ruby/object:Gem::Requirement
173
+ requirements:
174
+ - - ">="
175
+ - !ruby/object:Gem::Version
176
+ version: '0'
177
+ required_rubygems_version: !ruby/object:Gem::Requirement
178
+ requirements:
179
+ - - ">="
180
+ - !ruby/object:Gem::Version
181
+ version: '0'
182
+ requirements: []
183
+ rubyforge_project:
184
+ rubygems_version: 2.4.5
185
+ signing_key:
186
+ specification_version: 4
187
+ summary: A high level Kafka client.
188
+ test_files: []