def readn(sock, sz, tmout=2.0):
"""Read 'sz' bytes fro2 socket.
Raise XSPSessionEOF if we get 0 bytes before sz.
"""
log = get_logger("nl_xsp_recv.readn")
log.info("start", sz=sz)
t0 = time.time()
buf = ""
while len(buf) < sz:
b = ""
try:
b = sock.recv(sz - len(buf))
except socket.error, (errno, errmsg):
stopnow = True
if errno == 11:
dt = time.time() - t0
if dt < tmout:
stopnow = False
if stopnow:
log.warn("socket.error", msg=errmsg, errno=errno)
break
else:
time.sleep(0.1)
#print("xsp hdr len={len:d}".format(len=len(xsp_hdr)))
if len(b) == 0:
dt = time.time() - t0
if dt > tmout:
log.warn("readn.eof")
break
else:
time.sleep(0.1)
else:
buf += b
def config_logger(namespace=NETLOGGER_NAMESPACE, level = None, filename = None):
tmpLog = nllog.get_logger(namespace)
tmpLog.propagate = False
nllog.PROJECT_NAMESPACE = namespace
if filename:
add_filehandler(tmpLog, filename)
else:
handler = logging.StreamHandler()
handler.setFormatter(LogFormatter("%(message)s"))
tmpLog.addHandler(handler)
if level == "WARN":
tmpLog.setLevel(logging.WARN)
elif level == "TRACE":
tmpLog.setLevel(nllog.TRACE)
elif level == "DEBUG":
tmpLog.setLevel(logging.DEBUG)
if not filename:
enable_pretty_logging()
elif level == "CONSOLE":
tmpLog.setLevel(25)
else:
tmpLog.setLevel(logging.INFO)
return tmpLog
def config_logger(namespace=NETLOGGER_NAMESPACE, level=None, filename=None):
tmpLog = nllog.get_logger(namespace)
tmpLog.propagate = False
nllog.PROJECT_NAMESPACE = namespace
if filename:
add_filehandler(tmpLog, filename)
else:
handler = logging.StreamHandler()
handler.setFormatter(LogFormatter("%(message)s"))
tmpLog.addHandler(handler)
if level == "WARN":
tmpLog.setLevel(logging.WARN)
elif level == "TRACE":
tmpLog.setLevel(nllog.TRACE)
elif level == "DEBUG":
tmpLog.setLevel(logging.DEBUG)
if not filename:
enable_pretty_logging()
elif level == "CONSOLE":
tmpLog.setLevel(25)
else:
tmpLog.setLevel(logging.INFO)
return tmpLog
def get_logger(namespace=NETLOGGER_NAMESPACE, logfile=None, level=None):
"""Return logger object"""
# Test if netlogger is initialized
if nllog.PROJECT_NAMESPACE != NETLOGGER_NAMESPACE:
config_logger()
if logfile:
add_filehandler(logfile)
if level:
set_level(level)
return nllog.get_logger(namespace)
def get_logger(namespace=NETLOGGER_NAMESPACE):
"""Return logger object"""
# Test if netlloger is initialized
if nllog.PROJECT_NAMESPACE != NETLOGGER_NAMESPACE:
config_logger()
return nllog.get_logger(namespace)
