index.py 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. import asyncio
  2. import sys
  3. import uvicorn
  4. import signal
  5. from api import *
  6. from sockets.socket_server import *
  7. import uvicorn.loops.auto
  8. import uvicorn.protocols.http.auto
  9. import uvicorn.protocols.websockets.auto
  10. import uvicorn.lifespan.on
  11. from multiprocessing import Process, freeze_support
  12. from service.init_load_source import init_load_source
  13. def handle_shutdown(signum, frame):
  14. """关闭系统应用服务"""
  15. # 终止事件循环
  16. loop = asyncio.get_event_loop()
  17. loop.call_soon_threadsafe(loop.stop)
  18. sys.exit(0)
  19. async def run_server():
  20. # 启动uvicorn服务器
  21. isDebug = True if IS_DEBUG == "true" else False
  22. config = uvicorn.Config(
  23. APP_RUN,
  24. host=APP_HOST,
  25. port=int(PORT),
  26. reload=isDebug,
  27. workers=int(APP_WORKS),
  28. loop="auto",
  29. )
  30. server = uvicorn.Server(config)
  31. await server.serve()
  32. async def main():
  33. # await init_load_source().load_source()
  34. # await run_server()
  35. await asyncio.gather(run_server(), init_load_source().load_source())
  36. if __name__ == "__main__":
  37. if sys.platform == "win32":
  38. asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
  39. signal.signal(signal.SIGINT, handle_shutdown)
  40. signal.signal(signal.SIGTERM, handle_shutdown)
  41. print("python server is running at port:", PORT)
  42. print("python server is running at port:", APP_RUN)
  43. loop = asyncio.new_event_loop()
  44. try:
  45. loop.run_until_complete(main())
  46. finally:
  47. loop.close()
  48. # isDebug = True if IS_DEBUG == "true" else False
  49. # uvicorn.run(
  50. # app=APP_RUN,
  51. # host=APP_HOST,
  52. # port=int(PORT),
  53. # reload=isDebug,
  54. # loop="auto",
  55. # )