feat: use uvloop or winloop to speed up asyncio

WorldObservationLog · May 22, 2024 · d5c077a · d5c077a
1 parent cc65d47
commit d5c077a
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1,773 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -21,7 +21,7 @@ jobs:
           poetry run python -m pip install nuitka
       - name: Build
         run: |
-          poetry run python -m nuitka main.py --assume-yes-for-downloads --standalone --follow-imports --include-data-dir=assets=assets --include-data-files=config.example.toml=config.toml --include-data-files=agent.js=agent.js --include-module=mitmproxy_windows
+          poetry run python -m nuitka main.py --assume-yes-for-downloads --standalone --follow-imports --include-data-dir=assets=assets --include-data-files=config.example.toml=config.toml --include-data-files=agent.js=agent.js --include-module=mitmproxy_windows --include-module=winloop
       - name: Rename
         run: |
           ren main.dist AppleMusicDecrypt

diff --git a/main.py b/main.py
@@ -1,8 +1,15 @@
 import asyncio
+import sys
 
 from src.cmd import NewInteractiveShell
 
 if __name__ == '__main__':
+    if sys.platform in ('win32', 'cygwin', 'cli'):
+        import winloop
+        winloop.install()
+    else:
+        import uvloop
+        uvloop.install()
     loop = asyncio.get_event_loop()
     cmd = NewInteractiveShell(loop)
     try: