111 lines
3.0 KiB
Python
111 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import time
|
|
import datetime
|
|
import traceback
|
|
import threading
|
|
|
|
import q7
|
|
|
|
class _LocalLogNode:
|
|
def __init__(self, msg):
|
|
self.msg = msg
|
|
self.nextNode = None
|
|
|
|
class _UdpLog:
|
|
_instance = None
|
|
|
|
@classmethod
|
|
def instance(cls):
|
|
if not cls._instance:
|
|
cls._instance = _UdpLog()
|
|
return cls._instance
|
|
|
|
def __init__(self):
|
|
self._logDir = ''
|
|
self._topNode = None
|
|
self._botNode = None
|
|
self._workNode = None
|
|
self._terminated = False
|
|
self._cond = threading.Condition()
|
|
self._msgLock = threading.Lock()
|
|
self._workerThread = threading.Thread(target = self._saveToFile, args = ())
|
|
|
|
def init(self):
|
|
self._workerThread.start()
|
|
|
|
def unInit(self):
|
|
self._terminated = True
|
|
self._cond.acquire()
|
|
self._cond.notifyAll()
|
|
self._cond.release()
|
|
self._workerThread.join()
|
|
|
|
def setLogDirAndCreate(self, logdir):
|
|
if not os.path.exists(logdir):
|
|
os.makedirs(logdir)
|
|
self._logDir = logdir
|
|
|
|
def info(self, msg):
|
|
self._addLog(str(datetime.datetime.now()) + '[INFO]' + str(msg))
|
|
|
|
def debug(self, msg):
|
|
self._addLog(str(datetime.datetime.now()) + '[DEBUG]' + str(msg))
|
|
|
|
def warning(self, msg):
|
|
self._addLog(str(datetime.datetime.now()) + '[WARNING]' + str(msg))
|
|
|
|
def error(self, msg):
|
|
self._addLog(str(datetime.datetime.now()) + '[ERROR]' + str(msg))
|
|
|
|
def _addLog(self, msg):
|
|
node = _LocalLogNode(msg)
|
|
self._msgLock.acquire()
|
|
try:
|
|
if self._botNode == None:
|
|
self._topNode = node
|
|
self._botNode = node
|
|
else:
|
|
self._botNode.nextNode = node
|
|
self._botNode = node
|
|
except:
|
|
traceback.print_exc()
|
|
finally:
|
|
self._msgLock.release()
|
|
|
|
|
|
def _saveToFile(self):
|
|
try:
|
|
while not self._terminated:
|
|
self._internalSave()
|
|
self._cond.acquire()
|
|
self._cond.wait(10)
|
|
self._cond.release()
|
|
self._internalSave()
|
|
except:
|
|
traceback.print_exec()
|
|
|
|
def _internalSave(self):
|
|
if not self._workNode and self._topNode:
|
|
self._msgLock.acquire()
|
|
try:
|
|
self._workNode = self._topNode
|
|
self._topNode = None
|
|
self._botNode = None
|
|
finally:
|
|
self._msgLock.release()
|
|
if self._workNode:
|
|
filename = datetime.datetime.now().strftime('log_' + str(os.getpid()) + '_%Y%m%d.log')
|
|
try:
|
|
with open(os.path.join(self._logDir, filename), 'a') as f:
|
|
while self._workNode:
|
|
nextNode = self._workNode.nextNode
|
|
f.write(self._workNode.msg + '\r\n')
|
|
self._workNode = nextNode
|
|
except Exception as e:
|
|
print(str(e), flush=True)
|
|
|
|
def instance():
|
|
return _UdpLog.instance()
|