238 lines
7.8 KiB
Python
238 lines
7.8 KiB
Python
"""
|
||
Predefined Scheduled Tasks
|
||
All registered tasks for scheduled execution
|
||
"""
|
||
import logging
|
||
from datetime import datetime, timedelta
|
||
from typing import Dict, Any, List, Optional
|
||
from sqlalchemy import select
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy.dialects.postgresql import insert
|
||
|
||
from app.jobs.registry import task_registry
|
||
from app.models.db.celestial_body import CelestialBody
|
||
from app.models.db.position import Position
|
||
from app.services.horizons import HorizonsService
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@task_registry.register(
|
||
name="sync_solar_system_positions",
|
||
description="同步太阳系天体位置数据，从NASA Horizons API获取指定天体的位置数据并保存到数据库",
|
||
category="data_sync",
|
||
parameters=[
|
||
{
|
||
"name": "body_ids",
|
||
"type": "array",
|
||
"description": "要同步的天体ID列表，例如['10', '199', '299']。如果不指定，则同步所有活跃的太阳系天体",
|
||
"required": False,
|
||
"default": None
|
||
},
|
||
{
|
||
"name": "days",
|
||
"type": "integer",
|
||
"description": "同步天数，从今天开始向未来延伸的天数",
|
||
"required": False,
|
||
"default": 7
|
||
},
|
||
{
|
||
"name": "source",
|
||
"type": "string",
|
||
"description": "数据源标记，用于标识数据来源",
|
||
"required": False,
|
||
"default": "nasa_horizons_cron"
|
||
}
|
||
]
|
||
)
|
||
async def sync_solar_system_positions(
|
||
db: AsyncSession,
|
||
logger: logging.Logger,
|
||
params: Dict[str, Any]
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
Sync solar system body positions from NASA Horizons
|
||
|
||
Args:
|
||
db: Database session
|
||
logger: Logger instance
|
||
params: Task parameters
|
||
- body_ids: List of body IDs to sync (optional, defaults to all active)
|
||
- days: Number of days to sync (default: 7)
|
||
- source: Source tag for the data (default: "nasa_horizons_cron")
|
||
|
||
Returns:
|
||
Summary of sync operation
|
||
"""
|
||
body_ids = params.get("body_ids")
|
||
days = params.get("days", 7)
|
||
source = params.get("source", "nasa_horizons_cron")
|
||
|
||
logger.info(f"Starting solar system position sync: days={days}, source={source}")
|
||
|
||
# Get list of bodies to sync
|
||
if body_ids:
|
||
# Use specified body IDs
|
||
result = await db.execute(
|
||
select(CelestialBody).where(
|
||
CelestialBody.id.in_(body_ids),
|
||
CelestialBody.is_active == True
|
||
)
|
||
)
|
||
bodies = result.scalars().all()
|
||
logger.info(f"Syncing {len(bodies)} specified bodies")
|
||
else:
|
||
# Get all active solar system bodies
|
||
# Typically solar system bodies include planets, dwarf planets, and major satellites
|
||
result = await db.execute(
|
||
select(CelestialBody).where(
|
||
CelestialBody.is_active == True,
|
||
CelestialBody.system_id == 1,
|
||
CelestialBody.type.in_(['planet', 'dwarf_planet', 'satellite'])
|
||
)
|
||
)
|
||
bodies = result.scalars().all()
|
||
logger.info(f"Syncing all {len(bodies)} active solar system bodies")
|
||
|
||
if not bodies:
|
||
logger.warning("No bodies found to sync")
|
||
return {
|
||
"success": True,
|
||
"bodies_synced": 0,
|
||
"total_positions": 0,
|
||
"message": "No bodies found"
|
||
}
|
||
|
||
# Initialize services
|
||
horizons = HorizonsService()
|
||
|
||
# Sync positions for each body
|
||
total_positions = 0
|
||
synced_bodies = []
|
||
failed_bodies = []
|
||
|
||
start_time = datetime.utcnow()
|
||
end_time = start_time + timedelta(days=days)
|
||
|
||
for body in bodies:
|
||
# Use savepoint for this body's operations
|
||
async with db.begin_nested(): # Creates a SAVEPOINT
|
||
try:
|
||
logger.debug(f"Fetching positions for {body.name} ({body.id})")
|
||
|
||
# Fetch positions from NASA Horizons
|
||
positions = await horizons.get_body_positions(
|
||
body_id=body.id,
|
||
start_time=start_time,
|
||
end_time=end_time,
|
||
step="1d" # Daily positions
|
||
)
|
||
|
||
# Save positions to database (upsert logic)
|
||
count = 0
|
||
for pos in positions:
|
||
# Use PostgreSQL's INSERT ... ON CONFLICT to handle duplicates
|
||
stmt = insert(Position).values(
|
||
body_id=body.id,
|
||
time=pos.time,
|
||
x=pos.x,
|
||
y=pos.y,
|
||
z=pos.z,
|
||
vx=getattr(pos, 'vx', None),
|
||
vy=getattr(pos, 'vy', None),
|
||
vz=getattr(pos, 'vz', None),
|
||
source=source
|
||
)
|
||
|
||
# On conflict (body_id, time), update the existing record
|
||
stmt = stmt.on_conflict_do_update(
|
||
index_elements=['body_id', 'time'],
|
||
set_={
|
||
'x': pos.x,
|
||
'y': pos.y,
|
||
'z': pos.z,
|
||
'vx': getattr(pos, 'vx', None),
|
||
'vy': getattr(pos, 'vy', None),
|
||
'vz': getattr(pos, 'vz', None),
|
||
'source': source
|
||
}
|
||
)
|
||
|
||
await db.execute(stmt)
|
||
count += 1
|
||
|
||
# Savepoint will auto-commit if no exception
|
||
total_positions += count
|
||
synced_bodies.append(body.name)
|
||
logger.debug(f"Saved {count} positions for {body.name}")
|
||
|
||
except Exception as e:
|
||
# Savepoint will auto-rollback on exception
|
||
logger.error(f"Failed to sync {body.name}: {str(e)}")
|
||
failed_bodies.append({"body": body.name, "error": str(e)})
|
||
# Continue to next body
|
||
|
||
# Summary
|
||
result = {
|
||
"success": len(failed_bodies) == 0,
|
||
"bodies_synced": len(synced_bodies),
|
||
"total_positions": total_positions,
|
||
"synced_bodies": synced_bodies,
|
||
"failed_bodies": failed_bodies,
|
||
"time_range": f"{start_time.date()} to {end_time.date()}",
|
||
"source": source
|
||
}
|
||
|
||
logger.info(f"Sync completed: {len(synced_bodies)} bodies, {total_positions} positions")
|
||
return result
|
||
|
||
|
||
@task_registry.register(
|
||
name="sync_celestial_events",
|
||
description="同步天体事件数据（预留功能，暂未实现）",
|
||
category="data_sync",
|
||
parameters=[
|
||
{
|
||
"name": "event_types",
|
||
"type": "array",
|
||
"description": "事件类型列表，如['eclipse', 'conjunction', 'opposition']",
|
||
"required": False,
|
||
"default": None
|
||
},
|
||
{
|
||
"name": "days_ahead",
|
||
"type": "integer",
|
||
"description": "向未来查询的天数",
|
||
"required": False,
|
||
"default": 30
|
||
}
|
||
]
|
||
)
|
||
async def sync_celestial_events(
|
||
db: AsyncSession,
|
||
logger: logging.Logger,
|
||
params: Dict[str, Any]
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
Sync celestial events (PLACEHOLDER - NOT IMPLEMENTED YET)
|
||
|
||
This is a reserved task for future implementation.
|
||
It will sync astronomical events like eclipses, conjunctions, oppositions, etc.
|
||
|
||
Args:
|
||
db: Database session
|
||
logger: Logger instance
|
||
params: Task parameters
|
||
- event_types: Types of events to sync
|
||
- days_ahead: Number of days ahead to query
|
||
|
||
Returns:
|
||
Summary of sync operation
|
||
"""
|
||
logger.warning("sync_celestial_events is not implemented yet")
|
||
return {
|
||
"success": False,
|
||
"message": "This task is reserved for future implementation",
|
||
"events_synced": 0
|
||
}
|