1
0
Fork 1
mirror of https://github.com/thatmattlove/hyperglass.git synced 2026-01-30 05:49:22 +00:00
thatmattlove-hyperglass/hyperglass/execution/enhanced.py
Wilhelm Schonfeldt 0398966062
feat: Add structured traceroute support with comprehensive IP enrichment
MAJOR NEW ARCHITECTURE - STRUCTURED TRACEROUTE:
- Complete rewrite of traceroute data processing with structured output
- Dedicated TracerouteResult and TracerouteHop data models
- Platform-specific parsers with unified output format
- Rich metadata including ASN, organization, country, and prefix information
- AS path visualization with organization names in React Flow charts

SUPPORTED PLATFORMS:
- TraceroutePluginMikrotik: Handles MikroTik's complex multi-table format
  * Progressive statistics parsing with deduplication
  * Timeout hop handling and continuation line processing
  * Loss percentage and RTT statistics extraction
- TraceroutePluginHuawei: Unix-style traceroute format parser
  * Standard hop_number ip_address rtt format support
  * Timeout hop detection with * notation
  * Automatic cleanup of excessive trailing timeouts

COMPREHENSIVE IP ENRICHMENT SYSTEM:
- Offline enrichment using BGP.tools bulk data (1.3M+ CIDR entries)
- PeeringDB integration for IXP detection and ASN organization data
- Ultra-fast pickle cache system with combined data files
- Integer-based bitwise IP matching for maximum performance
- Bulk ASN organization lookup capabilities
- Private/reserved IP handling with AS0 fallbacks
- Country code mapping from ASN database
- Graceful fallbacks for missing enrichment data

FRONTEND ENHANCEMENTS:
- New traceroute table components with consistent formatting
- Enhanced AS path visualization with organization names
- Improved copy-to-clipboard functionality with structured data
- Unified table styling across BGP and traceroute results
- Better error handling and loading states

CONCURRENT PROCESSING INFRASTRUCTURE:
- Thread executor implementation for blocking I/O operations
- Query deduplication system to prevent resource conflicts
- Non-blocking Redis cache operations using asyncio executors
- Event coordination for waiting requests
- Background cleanup for completed operations
- Prevents website hangs during long-running queries

PLUGIN ARCHITECTURE IMPROVEMENTS:
- Platform-aware plugin system with proper execution restrictions
- Enhanced MikroTik garbage output cleaning
- IP enrichment plugins for both BGP routes and traceroute
- Conditional plugin execution based on platform detection
- Proper async/sync plugin method handling

CRITICAL BUG FIXES:
- Fixed double AS prefix bug (ASAS123456 → AS123456)
- Resolved TracerouteHop avg_rtt field/property conflicts
- Corrected Huawei traceroute source field validation
- Fixed plugin platform restriction enforcement
- Eliminated blocking I/O causing UI freezes
- Proper timeout and empty response caching prevention
- Enhanced private IP range detection and handling

PERFORMANCE OPTIMIZATIONS:
- Pickle cache system reduces startup time from seconds to milliseconds
- Bulk processing for ASN organization lookups
- Simplified IXP detection using single PeeringDB API call
- Efficient CIDR network sorting and integer-based lookups
- Reduced external API calls by 90%+
- Optimized memory usage for large datasets

API & ROUTING ENHANCEMENTS:
- Enhanced API routes with proper error handling
- Improved middleware for concurrent request processing
- Better state management and event handling
- Enhanced task processing with thread pool execution

This represents a complete transformation of hyperglass traceroute capabilities,
moving from basic text output to rich, structured data with comprehensive
network intelligence and concurrent processing support.
2025-09-28 13:48:04 +02:00

50 lines
1.7 KiB
Python

"""Example integration of BGP.tools enrichment into the main execution flow."""
# Standard Library
import typing as t
# Project
from hyperglass.log import log
from hyperglass.execution.main import execute as original_execute
from hyperglass.execution.enrichment import execute_with_enrichment
from hyperglass.models.data import OutputDataModel
if t.TYPE_CHECKING:
from hyperglass.models.api import Query
async def execute_enhanced(query: "Query") -> t.Union[OutputDataModel, str]:
"""Enhanced execute function with BGP.tools enrichment.
This can be used to replace the original execute function in hyperglass.execution.main
to add automatic BGP.tools enrichment to all query results.
Usage:
# In hyperglass/api/routes.py, replace:
# from hyperglass.execution.main import execute
# with:
# from hyperglass.execution.enhanced import execute_enhanced as execute
"""
return await execute_with_enrichment(query, original_execute)
# Optional: Patch the original execute function
def monkey_patch_execute():
"""Monkey patch the original execute function with enhanced version.
This can be called during application startup to automatically enable
BGP.tools enrichment without changing imports throughout the codebase.
Usage:
# In hyperglass application startup code:
from hyperglass.execution.enhanced import monkey_patch_execute
monkey_patch_execute()
"""
import hyperglass.execution.main
import hyperglass.api.routes
# Replace the execute function in both modules
hyperglass.execution.main.execute = execute_enhanced
hyperglass.api.routes.execute = execute_enhanced
log.info("BGP.tools enrichment enabled via monkey patching")