Skip to content

Commit

Permalink
Use a buffer rather than reading the Twitter response one byte at a t…
Browse files Browse the repository at this point in the history
…ime. This actually respects the buffer_size option that was already there, but unused.
  • Loading branch information
Dan Fairs committed Jan 31, 2012
1 parent 41fe5dc commit a6d701d
Showing 1 changed file with 20 additions and 12 deletions.
32 changes: 20 additions & 12 deletions tweepy/streaming.py
Expand Up @@ -135,23 +135,31 @@ def _run(self):
if exception:
raise

def _data(self, data):
for d in [dt for dt in data.split('\n') if dt]:
if self.listener.on_data(d) is False:
self.running = False

def _read_loop(self, resp):
while self.running:
while self.running:
if resp.isclosed():
break

# read length
data = ''
buf = ''
while True:
c = resp.read(1)
if c == '\n':
break
data += c
data = data.strip()

# read data and pass into listener
if self.listener.on_data(data) is False:
self.running = False
c = resp.read(self.buffer_size)
idx = c.rfind('\n')
if idx > -1:
# There is an index. Store the tail part for later,
# and process the head part as messages. We use idx + 1
# as we dont' actually want to store the newline.
data = buf + c[:idx]
buf = c[idx + 1:]
self._data(data)
else:
# No newline found, so we add this to our accumulated
# buffer
buf += c

def _start(self, async):
self.running = True
Expand Down

0 comments on commit a6d701d

Please sign in to comment.