nginx-lens 0.3.3__tar.gz → 0.3.4__tar.gz

{nginx_lens-0.3.3/nginx_lens.egg-info → nginx_lens-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nginx-lens
-Version: 0.3.3
+Version: 0.3.4
 Summary: CLI-инструмент для анализа, визуализации и диагностики конфигураций Nginx
 Author: Daniil Astrouski
 Author-email: shelovesuastra@gmail.com

{nginx_lens-0.3.3 → nginx_lens-0.3.4}/commands/health.py

@@ -14,14 +14,17 @@ def health(
     retries: int = typer.Option(1, help="Количество попыток"),
     mode: str = typer.Option("tcp", help="Режим проверки: tcp или http", case_sensitive=False),
     resolve: bool = typer.Option(False, "--resolve", "-r", help="Показать резолвленные IP-адреса"),
+    max_workers: int = typer.Option(10, "--max-workers", "-w", help="Максимальное количество потоков для параллельной обработки"),
 ):
     """
     Проверяет доступность upstream-серверов, определённых в nginx.conf. Выводит таблицу.
+    Использует параллельную обработку для ускорения проверки множества upstream серверов.
 
     Пример:
         nginx-lens health /etc/nginx/nginx.conf
         nginx-lens health /etc/nginx/nginx.conf --timeout 5 --retries 3 --mode http
         nginx-lens health /etc/nginx/nginx.conf --resolve
+        nginx-lens health /etc/nginx/nginx.conf --max-workers 20
     """
     exit_code = 0
     
@@ -35,12 +38,12 @@ def health(
         sys.exit(1)
 
     upstreams = tree.get_upstreams()
-    results = check_upstreams(upstreams, timeout=timeout, retries=retries, mode=mode.lower())
+    results = check_upstreams(upstreams, timeout=timeout, retries=retries, mode=mode.lower(), max_workers=max_workers)
     
     # Если нужно показать резолвленные IP-адреса
     resolved_info = {}
     if resolve:
-        resolved_info = resolve_upstreams(upstreams)
+        resolved_info = resolve_upstreams(upstreams, max_workers=max_workers)
 
     table = Table(show_header=True, header_style="bold blue")
     table.add_column("Address")

{nginx_lens-0.3.3 → nginx_lens-0.3.4}/commands/resolve.py

@@ -10,12 +10,15 @@ console = Console()
 
 def resolve(
     config_path: str = typer.Argument(..., help="Путь к nginx.conf"),
+    max_workers: int = typer.Option(10, "--max-workers", "-w", help="Максимальное количество потоков для параллельной обработки"),
 ):
     """
     Резолвит DNS имена upstream-серверов в IP-адреса.
+    Использует параллельную обработку для ускорения резолвинга множества upstream серверов.
 
     Пример:
         nginx-lens resolve /etc/nginx/nginx.conf
+        nginx-lens resolve /etc/nginx/nginx.conf --max-workers 20
     """
     exit_code = 0
     
@@ -33,7 +36,7 @@ def resolve(
         console.print("[yellow]Не найдено ни одного upstream в конфигурации.[/yellow]")
         sys.exit(0)  # Нет upstream - это не ошибка, просто нет чего проверять
     
-    results = resolve_upstreams(upstreams)
+    results = resolve_upstreams(upstreams, max_workers=max_workers)
 
     table = Table(show_header=True, header_style="bold blue")
     table.add_column("Upstream Name")

{nginx_lens-0.3.3 → nginx_lens-0.3.4/nginx_lens.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nginx-lens
-Version: 0.3.3
+Version: 0.3.4
 Summary: CLI-инструмент для анализа, визуализации и диагностики конфигураций Nginx
 Author: Daniil Astrouski
 Author-email: shelovesuastra@gmail.com

{nginx_lens-0.3.3 → nginx_lens-0.3.4}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="nginx-lens",
-    version="0.3.3",
+    version="0.3.4",
     description="CLI-инструмент для анализа, визуализации и диагностики конфигураций Nginx",
     author="Daniil Astrouski",
     author_email="shelovesuastra@gmail.com",

{nginx_lens-0.3.3 → nginx_lens-0.3.4}/upstream_checker/checker.py

@@ -3,7 +3,8 @@
 import socket
 import time
 import http.client
-from typing import Dict, List
+from typing import Dict, List, Tuple
+from concurrent.futures import ThreadPoolExecutor, as_completed
 try:
     import dns.resolver
     import dns.exception
@@ -185,11 +186,16 @@ def _resolve_with_socket(host: str, port: str) -> List[str]:
 
 
 def resolve_upstreams(
-    upstreams: Dict[str, List[str]]
+    upstreams: Dict[str, List[str]],
+    max_workers: int = 10
 ) -> Dict[str, List[dict]]:
     """
     Резолвит DNS имена upstream-серверов в IP-адреса.
     
+    Args:
+        upstreams: Словарь upstream серверов
+        max_workers: Максимальное количество потоков для параллельной обработки
+    
     Возвращает:
     {
         "backend": [
@@ -200,27 +206,71 @@ def resolve_upstreams(
         ]
     }
     """
-    results = {}
+    # Собираем все задачи для параллельной обработки
+    tasks = []
+    task_to_key = {}
+    
     for name, servers in upstreams.items():
-        results[name] = []
-        for srv in servers:
-            resolved = resolve_address(srv)
-            results[name].append({
-                "address": srv,
-                "resolved": resolved
-            })
+        for idx, srv in enumerate(servers):
+            key = (name, idx, srv)
+            tasks.append((key, srv))
+            task_to_key[key] = (name, idx)
+    
+    results = {}
+    for name in upstreams.keys():
+        results[name] = [None] * len(upstreams[name])
+    
+    # Если нет задач, возвращаем пустой результат
+    if not tasks:
+        return results
+    
+    # Параллельная обработка резолвинга
+    with ThreadPoolExecutor(max_workers=max_workers) as executor:
+        future_to_key = {executor.submit(resolve_address, srv): key for key, srv in tasks}
+        
+        for future in as_completed(future_to_key):
+            key = future_to_key[future]
+            name, idx = task_to_key[key]
+            try:
+                resolved = future.result()
+                results[name][idx] = {
+                    "address": key[2],
+                    "resolved": resolved
+                }
+            except Exception:
+                results[name][idx] = {
+                    "address": key[2],
+                    "resolved": []
+                }
+    
     return results
 
 
+def _check_single_upstream(srv: str, timeout: float, retries: int, mode: str) -> Tuple[str, bool]:
+    """Вспомогательная функция для проверки одного upstream сервера."""
+    if mode.lower() == "http":
+        healthy = check_http(srv, timeout, retries)
+    else:
+        healthy = check_tcp(srv, timeout, retries)
+    return (srv, healthy)
+
+
 def check_upstreams(
     upstreams: Dict[str, List[str]],
     timeout: float = 2.0,
     retries: int = 1,
-    mode: str = "tcp"
+    mode: str = "tcp",
+    max_workers: int = 10
 ) -> Dict[str, List[dict]]:
     """
     Проверяет доступность upstream-серверов.
-    mode: "tcp" (по умолчанию) или "http"
+    
+    Args:
+        upstreams: Словарь upstream серверов
+        timeout: Таймаут проверки (сек)
+        retries: Количество попыток
+        mode: "tcp" (по умолчанию) или "http"
+        max_workers: Максимальное количество потоков для параллельной обработки
     
     Возвращает:
     {
@@ -230,13 +280,38 @@ def check_upstreams(
         ]
     }
     """
-    results = {}
+    # Собираем все задачи для параллельной обработки
+    tasks = []
+    task_to_key = {}
+    
     for name, servers in upstreams.items():
-        results[name] = []
-        for srv in servers:
-            if mode.lower() == "http":
-                healthy = check_http(srv, timeout, retries)
-            else:
-                healthy = check_tcp(srv, timeout, retries)
-            results[name].append({"address": srv, "healthy": healthy})
+        for idx, srv in enumerate(servers):
+            key = (name, idx, srv)
+            tasks.append((key, srv))
+            task_to_key[key] = (name, idx)
+    
+    results = {}
+    for name in upstreams.keys():
+        results[name] = [None] * len(upstreams[name])
+    
+    # Если нет задач, возвращаем пустой результат
+    if not tasks:
+        return results
+    
+    # Параллельная обработка проверок
+    with ThreadPoolExecutor(max_workers=max_workers) as executor:
+        future_to_key = {
+            executor.submit(_check_single_upstream, srv, timeout, retries, mode): key 
+            for key, srv in tasks
+        }
+        
+        for future in as_completed(future_to_key):
+            key = future_to_key[future]
+            name, idx = task_to_key[key]
+            try:
+                srv, healthy = future.result()
+                results[name][idx] = {"address": srv, "healthy": healthy}
+            except Exception:
+                results[name][idx] = {"address": key[2], "healthy": False}
+    
     return results