8000 Cleanup sender · moriyoshi/fluent-logger-python@c64fce5 · GitHub
[go: up one dir, main page]

Skip to content

Commit c64fce5

Browse files
committed
Cleanup sender
1 parent 6f5dbf0 commit c64fce5

File tree

1 file changed

+17
-27
lines changed

1 file changed

+17
-27
lines changed

fluent/sender.py

Lines changed: 17 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,32 +1,28 @@
1-
import os
2-
import sys, urllib
31
import msgpack
42
import socket
53
import threading
6-
import json
74
import time
85

9-
global_sender = None
6+
_global_sender = None
107

118
def setup(tag, **kwargs):
12-
host = ('host' in kwargs) and kwargs['host'] or 'localhost'
13-
port = ('port' in kwargs) and kwargs['port'] or 24224
9+
host = kwargs.get('host', 'localhost')
10+
port = kwargs.get('port', 24224)
1411

15-
global global_sender
16-
global_sender = FluentSender(tag, host=host, port=port)
12+
global _global_sender
13+
_global_sender = FluentSender(tag, host=host, port=port)
1714

1815
def get_global_sender():
19-
global global_sender
20-
return global_sender
16+
return _global_sender
2117

2218
class FluentSender(object):
2319
def __init__(self,
24-
tag,
25-
host='127.0.0.1',
26-
port=24224,
27-
bufmax=1*1024*1024,
28-
timeout=3.0,
29-
verbose=False):
20+
tag,
21+
host='localhost',
22+
port=24224,
23+
bufmax=1*1024*1024,
24+
timeout=3.0,
25+
verbose=False):
3026

3127
self.tag = tag
3228
self.host = host
@@ -64,9 +60,9 @@ def _close(self):
6460
self.socket = None
6561

6662
def _make_packet(self, label, data):
67-
tag = "%s.%s" % (self.tag, label)
68-
cur_time = int(time.mktime(time.localtime()))
69-
packet = [ tag, cur_time, data ]
63+
tag = '.'.join(self.tag, label)
64+
cur_time = int(time.time())
65+
packet = (tag, cur_time, data)
7066
if self.verbose:
7167
print packet
7268
return self.packer.pack(packet)
@@ -89,17 +85,11 @@ def _send_internal(self, bytes):
8985
self._reconnect()
9086

9187
# send message
92-
total = len(bytes)
93-
nsent = 0
94-
while nsent < total:
95-
n = self.socket.send(bytes[nsent:])
96-
if n <= 0:
97-
raise RuntimeError("socket connection broken")
98-
nsent += n
88+
self.socket.sendall(bytes)
9989

10090
# send finished
10191
self.pendings = None
102-
except:
92+
except Exception:
10393
# close socket
10494
self._close()
10595
# clear buffer if it exceeds max bufer size

0 commit comments

Comments
 (0)
0