283 lines
11 KiB
Python
283 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
稷下学宫负载均衡演示脚本
|
||
展示八仙论道的API负载分担策略
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
import json
|
||
from datetime import datetime
|
||
|
||
# 添加项目路径
|
||
sys.path.append('/home/ben/liurenchaxin/src')
|
||
|
||
from jixia.engines.jixia_load_balancer import JixiaLoadBalancer, APIResult
|
||
|
||
def print_banner():
|
||
"""打印横幅"""
|
||
print("\n" + "="*80)
|
||
print("🏛️ 稷下学宫八仙论道 - API负载均衡演示系统")
|
||
print("📊 RapidAPI多源数据整合与负载分担策略")
|
||
print("="*80)
|
||
print(f"⏰ 演示时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||
print()
|
||
|
||
def print_immortal_intro():
|
||
"""介绍八仙角色"""
|
||
immortals_info = {
|
||
'吕洞宾': '主力剑仙 - 综合分析与决策 (Alpha Vantage)',
|
||
'何仙姑': '风控专家 - 风险管理与合规 (Yahoo Finance)',
|
||
'张果老': '技术分析师 - 技术指标与图表分析 (Webull)',
|
||
'韩湘子': '基本面研究员 - 财务分析与估值 (Seeking Alpha)',
|
||
'汉钟离': '量化专家 - 数据挖掘与算法交易 (Yahoo Finance)',
|
||
'蓝采和': '情绪分析师 - 市场情绪与舆情监控 (Webull)',
|
||
'曹国舅': '宏观分析师 - 宏观经济与政策分析 (Seeking Alpha)',
|
||
'铁拐李': '逆向投资专家 - 价值发现与逆向思维 (Alpha Vantage)'
|
||
}
|
||
|
||
print("👥 八仙角色与API分配:")
|
||
print("-" * 60)
|
||
for immortal, description in immortals_info.items():
|
||
print(f" {immortal}: {description}")
|
||
print()
|
||
|
||
def demonstrate_single_immortal(load_balancer, immortal_name, symbol):
|
||
"""演示单个仙人的数据获取"""
|
||
print(f"\n🎭 {immortal_name} 单独获取数据演示:")
|
||
print("-" * 40)
|
||
|
||
# 获取股票报价
|
||
result = load_balancer.get_data_for_immortal(immortal_name, 'stock_quote', symbol)
|
||
|
||
if result.success:
|
||
data = result.data
|
||
print(f" ✅ 成功获取 {symbol} 数据:")
|
||
print(f" 💰 价格: ${data.get('price', 'N/A')}")
|
||
print(f" 📈 涨跌: {data.get('change_percent', 'N/A')}")
|
||
volume = data.get('volume', 'N/A')
|
||
if isinstance(volume, (int, float)):
|
||
print(f" 📊 成交量: {volume:,}")
|
||
else:
|
||
print(f" 📊 成交量: {volume}")
|
||
print(f" 🔗 数据源: {result.api_used}")
|
||
print(f" ⏱️ 响应时间: {result.response_time:.2f}秒")
|
||
print(f" 💾 缓存状态: {'是' if result.cached else '否'}")
|
||
else:
|
||
print(f" ❌ 获取失败: {result.error}")
|
||
|
||
def demonstrate_load_distribution(load_balancer):
|
||
"""演示负载分布"""
|
||
print("\n📊 API负载分布统计:")
|
||
print("-" * 40)
|
||
|
||
distribution = load_balancer.get_load_distribution()
|
||
|
||
if not distribution:
|
||
print(" 📝 暂无API调用记录")
|
||
return
|
||
|
||
total_calls = sum(stats['calls'] for stats in distribution.values())
|
||
|
||
for api_name, stats in distribution.items():
|
||
status_icon = "🟢" if stats['healthy'] else "🔴"
|
||
print(f" {status_icon} {api_name}:")
|
||
print(f" 📞 调用次数: {stats['calls']}")
|
||
print(f" 📊 负载占比: {stats['percentage']:.1f}%")
|
||
print(f" ❌ 连续失败: {stats['consecutive_failures']}次")
|
||
print()
|
||
|
||
def demonstrate_api_comparison(load_balancer, symbol):
|
||
"""演示不同API的数据对比"""
|
||
print(f"\n🔍 {symbol} 多API数据对比:")
|
||
print("-" * 50)
|
||
|
||
apis = ['alpha_vantage', 'yahoo_finance_15', 'webull']
|
||
results = {}
|
||
|
||
for api in apis:
|
||
# 临时修改API分配来测试不同数据源
|
||
original_mapping = load_balancer.immortal_api_mapping['stock_quote']['吕洞宾']
|
||
load_balancer.immortal_api_mapping['stock_quote']['吕洞宾'] = api
|
||
|
||
result = load_balancer.get_data_for_immortal('吕洞宾', 'stock_quote', symbol)
|
||
results[api] = result
|
||
|
||
# 恢复原始配置
|
||
load_balancer.immortal_api_mapping['stock_quote']['吕洞宾'] = original_mapping
|
||
|
||
time.sleep(0.5) # 避免过快请求
|
||
|
||
# 显示对比结果
|
||
print(" API数据源对比:")
|
||
for api, result in results.items():
|
||
if result.success:
|
||
data = result.data
|
||
print(f" 📡 {api}:")
|
||
print(f" 💰 ${data.get('price', 'N/A')} ({data.get('change_percent', 'N/A')})")
|
||
print(f" ⏱️ {result.response_time:.2f}s")
|
||
else:
|
||
print(f" 📡 {api}: ❌ {result.error}")
|
||
print()
|
||
|
||
def demonstrate_cache_effectiveness(load_balancer, symbol):
|
||
"""演示缓存效果"""
|
||
print(f"\n💾 缓存效果演示 - {symbol}:")
|
||
print("-" * 40)
|
||
|
||
# 第一次调用（无缓存）
|
||
print(" 🔄 第一次调用（无缓存）:")
|
||
start_time = time.time()
|
||
result1 = load_balancer.get_data_for_immortal('吕洞宾', 'stock_quote', symbol)
|
||
first_call_time = time.time() - start_time
|
||
|
||
if result1.success:
|
||
print(f" ⏱️ 响应时间: {result1.response_time:.2f}秒")
|
||
print(f" 💾 缓存状态: {'命中' if result1.cached else '未命中'}")
|
||
|
||
time.sleep(1)
|
||
|
||
# 第二次调用（有缓存）
|
||
print(" 🔄 第二次调用（有缓存）:")
|
||
start_time = time.time()
|
||
result2 = load_balancer.get_data_for_immortal('吕洞宾', 'stock_quote', symbol)
|
||
second_call_time = time.time() - start_time
|
||
|
||
if result2.success:
|
||
print(f" ⏱️ 响应时间: {result2.response_time:.2f}秒")
|
||
print(f" 💾 缓存状态: {'命中' if result2.cached else '未命中'}")
|
||
|
||
if result2.cached:
|
||
speedup = (first_call_time / second_call_time) if second_call_time > 0 else float('inf')
|
||
print(f" 🚀 性能提升: {speedup:.1f}倍")
|
||
|
||
def demonstrate_failover(load_balancer, symbol):
|
||
"""演示故障转移"""
|
||
print(f"\n🔄 故障转移演示 - {symbol}:")
|
||
print("-" * 40)
|
||
|
||
# 模拟API故障
|
||
print(" ⚠️ 模拟主API故障...")
|
||
|
||
# 临时标记API为不健康
|
||
original_health = load_balancer.health_checker.health_status['alpha_vantage']['healthy']
|
||
load_balancer.health_checker.health_status['alpha_vantage']['healthy'] = False
|
||
load_balancer.health_checker.health_status['alpha_vantage']['consecutive_failures'] = 5
|
||
|
||
# 尝试获取数据（应该自动故障转移）
|
||
result = load_balancer.get_data_for_immortal('吕洞宾', 'stock_quote', symbol)
|
||
|
||
if result.success:
|
||
print(f" ✅ 故障转移成功，使用备用API: {result.api_used}")
|
||
print(f" 💰 获取到价格: ${result.data.get('price', 'N/A')}")
|
||
else:
|
||
print(f" ❌ 故障转移失败: {result.error}")
|
||
|
||
# 恢复API健康状态
|
||
load_balancer.health_checker.health_status['alpha_vantage']['healthy'] = original_health
|
||
load_balancer.health_checker.health_status['alpha_vantage']['consecutive_failures'] = 0
|
||
|
||
print(" 🔧 API健康状态已恢复")
|
||
|
||
def save_demo_results(results, filename='demo_results.json'):
|
||
"""保存演示结果"""
|
||
demo_data = {
|
||
'timestamp': datetime.now().isoformat(),
|
||
'results': {},
|
||
'summary': {
|
||
'total_immortals': len(results),
|
||
'successful_calls': sum(1 for r in results.values() if r.success),
|
||
'failed_calls': sum(1 for r in results.values() if not r.success)
|
||
}
|
||
}
|
||
|
||
for immortal, result in results.items():
|
||
demo_data['results'][immortal] = {
|
||
'success': result.success,
|
||
'api_used': result.api_used,
|
||
'response_time': result.response_time,
|
||
'cached': result.cached,
|
||
'error': result.error,
|
||
'data_summary': {
|
||
'symbol': result.data.get('symbol') if result.success else None,
|
||
'price': result.data.get('price') if result.success else None,
|
||
'change_percent': result.data.get('change_percent') if result.success else None
|
||
}
|
||
}
|
||
|
||
with open(filename, 'w', encoding='utf-8') as f:
|
||
json.dump(demo_data, f, indent=2, ensure_ascii=False)
|
||
|
||
print(f"\n💾 演示结果已保存到: {filename}")
|
||
|
||
def main():
|
||
"""主演示函数"""
|
||
# 检查API密钥
|
||
rapidapi_key = os.getenv('RAPIDAPI_KEY')
|
||
if not rapidapi_key:
|
||
print("❌ 错误: 请设置RAPIDAPI_KEY环境变量")
|
||
print(" 提示: 使用 'doppler run python demo_jixia_load_balancing.py' 运行")
|
||
return
|
||
|
||
print_banner()
|
||
print_immortal_intro()
|
||
|
||
# 创建负载均衡器
|
||
print("🔧 初始化稷下学宫负载均衡器...")
|
||
load_balancer = JixiaLoadBalancer(rapidapi_key)
|
||
print("✅ 负载均衡器初始化完成\n")
|
||
|
||
# 演示股票代码
|
||
demo_symbols = ['AAPL', 'TSLA', 'MSFT']
|
||
|
||
for i, symbol in enumerate(demo_symbols, 1):
|
||
print(f"\n{'='*20} 演示 {i}: {symbol} {'='*20}")
|
||
|
||
# 1. 单个仙人演示
|
||
demonstrate_single_immortal(load_balancer, '吕洞宾', symbol)
|
||
|
||
# 2. 八仙论道演示
|
||
print(f"\n🏛️ 八仙论道完整演示 - {symbol}:")
|
||
debate_results = load_balancer.conduct_immortal_debate(symbol)
|
||
|
||
# 3. 负载分布演示
|
||
demonstrate_load_distribution(load_balancer)
|
||
|
||
# 只在第一个股票上演示高级功能
|
||
if i == 1:
|
||
# 4. API对比演示
|
||
demonstrate_api_comparison(load_balancer, symbol)
|
||
|
||
# 5. 缓存效果演示
|
||
demonstrate_cache_effectiveness(load_balancer, symbol)
|
||
|
||
# 6. 故障转移演示
|
||
demonstrate_failover(load_balancer, symbol)
|
||
|
||
# 保存结果
|
||
save_demo_results(debate_results, f'demo_results_{symbol.lower()}.json')
|
||
|
||
if i < len(demo_symbols):
|
||
print("\n⏳ 等待3秒后继续下一个演示...")
|
||
time.sleep(3)
|
||
|
||
# 最终统计
|
||
print("\n" + "="*80)
|
||
print("📈 演示完成 - 最终负载分布统计:")
|
||
demonstrate_load_distribution(load_balancer)
|
||
|
||
print("\n🎉 稷下学宫API负载均衡演示完成！")
|
||
print("\n💡 关键特性:")
|
||
print(" ✅ 智能负载分担 - 八仙各司其职，分散API压力")
|
||
print(" ✅ 自动故障转移 - API异常时自动切换备用源")
|
||
print(" ✅ 数据标准化 - 统一不同API的数据格式")
|
||
print(" ✅ 智能缓存 - 减少重复调用，提升响应速度")
|
||
print(" ✅ 实时监控 - 跟踪API健康状态和负载分布")
|
||
print("\n📚 查看详细配置: /home/ben/liurenchaxin/src/jixia/config/immortal_api_config.json")
|
||
print("🔧 核心引擎: /home/ben/liurenchaxin/src/jixia/engines/jixia_load_balancer.py")
|
||
print("="*80)
|
||
|
||
if __name__ == "__main__":
|
||
main() |