bakery-ia/services/training/app/services/messaging.py

# ================================================================
# services/training/app/services/messaging.py
# ================================================================
"""
Messaging service for training service
"""

import structlog
from shared.messaging.rabbitmq import RabbitMQClient
from app.core.config import settings

logger = structlog.get_logger()

# Single global instance
training_publisher = RabbitMQClient(settings.RABBITMQ_URL, "training-service")

async def setup_messaging():
    """Initialize messaging for training service"""
    success = await training_publisher.connect()
    if success:
        logger.info("Training service messaging initialized")
    else:
        logger.warning("Training service messaging failed to initialize")

async def cleanup_messaging():
    """Cleanup messaging for training service"""
    await training_publisher.disconnect()
    logger.info("Training service messaging cleaned up")

# Convenience functions for training-specific events
async def publish_training_started(job_data: dict) -> bool:
    """Publish training started event"""
    return await training_publisher.publish_training_event("started", job_data)

async def publish_training_completed(job_data: dict) -> bool:
    """Publish training completed event"""
    return await training_publisher.publish_training_event("completed", job_data)

async def publish_training_failed(job_data: dict) -> bool:
    """Publish training failed event"""
    return await training_publisher.publish_training_event("failed", job_data)

async def publish_model_validated(model_data: dict) -> bool:
    """Publish model validation event"""
    return await training_publisher.publish_training_event("model.validated", model_data)

async def publish_model_saved(model_data: dict) -> bool:
    """Publish model saved event"""
    return await training_publisher.publish_training_event("model.saved", model_data)