105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
import daemon
|
||
import time
|
||
import logging
|
||
import logging.handlers
|
||
import multiprocessing
|
||
from multiprocessing import Event
|
||
from src.usb import Usb
|
||
from src.mqtt import Mqtt
|
||
from src.payload import USBdata
|
||
from src.config import Config
|
||
import signal
|
||
from daemon.pidfile import PIDLockFile
|
||
|
||
|
||
|
||
class App():
|
||
def __init__(self):
|
||
# 加载配置文件
|
||
self.cfg = Config()
|
||
|
||
def run(self):
|
||
|
||
signal.signal(signal.SIGINT, self.exit)
|
||
signal.signal(signal.SIGTERM, self.exit)
|
||
# signal.signal(signal.SIGKILL, self.exit)
|
||
signal.signal(signal.SIGHUP, self.exit)
|
||
|
||
# 初始化日志
|
||
log = logging.getLogger('gateway')
|
||
log.setLevel(logging.DEBUG) # 实际运行一般用info
|
||
|
||
# 创建一个StreamHandler对象，用于将日志输出到终端上
|
||
console = logging.StreamHandler()
|
||
console.setLevel(logging.DEBUG) # 终端上只显示DEBUG及以上级别的日志
|
||
|
||
# 创建一个RotatingFileHandler对象，用于将日志写入到指定路径的文件中
|
||
fh = logging.handlers.RotatingFileHandler(self.cfg.Sys.logpath, maxBytes=1000000,
|
||
backupCount=1) # 滚动式日志文件位置、大小、数量部署时按需修改
|
||
fh.setLevel(logging.DEBUG)
|
||
|
||
# 为两个Handler对象设置日志记录格式
|
||
formatter = logging.Formatter(u'%(asctime)s [%(levelname)s] %(message)s') # 日志格式
|
||
console.setFormatter(formatter)
|
||
fh.setFormatter(formatter)
|
||
|
||
# 将console和fh两个Handler对象添加到Logger对象log中
|
||
log.addHandler(console)
|
||
log.addHandler(fh)
|
||
|
||
# 公共队列
|
||
q1 = multiprocessing.Queue()
|
||
q2 = multiprocessing.Queue(maxsize=50)
|
||
# 公共事件
|
||
self.event = Event()
|
||
|
||
# 处理线程，包括参数计算和MQTT发送
|
||
mqtt = Mqtt(self.cfg.Mqtt, log)
|
||
err = mqtt.Connect()
|
||
# TODO: mqtt连不上怎么办？
|
||
# 暂时考虑，连不上等待10秒继续连接
|
||
while err != 0:
|
||
log.error("正在重连...")
|
||
err = mqtt.Connect()
|
||
time.sleep(10)
|
||
USBdata_thread = multiprocessing.Process(target=USBdata, args=(q1, q2, log))
|
||
mqtt_thread = multiprocessing.Process(target=mqtt.Start, args=(q2, self.event, self.cfg.Mqtt))
|
||
|
||
|
||
# USB线程
|
||
usb = Usb(self.cfg.Usb, log)
|
||
err = usb.Connect()
|
||
if err != 0:
|
||
exit(err)
|
||
|
||
USBGetInfo = multiprocessing.Process(target=usb.WriteDataWork, args=(q1, self.event))
|
||
USBKeepAliveWork = multiprocessing.Process(target=usb.KeepAliveWork)
|
||
USBGetInfo.start()
|
||
USBKeepAliveWork.start()
|
||
mqtt_thread.start()
|
||
USBdata_thread.start()
|
||
|
||
USBGetInfo.join()
|
||
USBKeepAliveWork.join()
|
||
mqtt_thread.join()
|
||
USBdata_thread.join()
|
||
|
||
log.info("program exit!")
|
||
usb.Disconnect()
|
||
mqtt.Disconnect()
|
||
|
||
def exit(self, signum, frame):
|
||
self.event.set()
|
||
|
||
|
||
app = App()
|
||
|
||
pidfile = PIDLockFile(app.cfg.Sys.pidpath)
|
||
|
||
# 判断pid文件是否存在或锁定
|
||
if pidfile.is_locked():
|
||
pidfile.break_lock()
|
||
|
||
# with daemon.DaemonContext(pidfile=pidfile):
|
||
app.run()
|