Skip to content

datarobot_genai.drmcp.core.server_life_cycle

server_life_cycle

BaseServerLifecycle

Base server lifecycle interface with safe default implementations.

This class provides hooks that are called at different stages of the server lifecycle. Subclasses can override any or all of these methods to add custom behavior. All methods have safe no-op defaults, so you only need to implement what you need.

Lifecycle Order
  1. pre_server_start() - Before server initialization
  2. Server starts
  3. post_server_start() - After server is ready
  4. Server runs...
  5. Shutdown signal received
  6. pre_server_shutdown() - Before server cleanup
  7. Server stops
Example 
class MyLifecycle(BaseServerLifecycle):
    async def pre_server_start(self, mcp: FastMCP) -> None:
        # Initialize resources
        self.db = await connect_database()

    async def pre_server_shutdown(self, mcp: FastMCP) -> None:
        # Clean up resources
        await self.db.close()

    # post_server_start not implemented - will use safe default (no-op)
Source code in datarobot_genai/drmcp/core/server_life_cycle.py 
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
class BaseServerLifecycle:
    """
    Base server lifecycle interface with safe default implementations.

    This class provides hooks that are called at different stages of the server lifecycle.
    Subclasses can override any or all of these methods to add custom behavior.
    All methods have safe no-op defaults, so you only need to implement what you need.

    Lifecycle Order:
        1. pre_server_start()  - Before server initialization
        2. Server starts
        3. post_server_start() - After server is ready
        4. Server runs...
        5. Shutdown signal received
        6. pre_server_shutdown() - Before server cleanup
        7. Server stops

    Example:
        ```python
        class MyLifecycle(BaseServerLifecycle):
            async def pre_server_start(self, mcp: FastMCP) -> None:
                # Initialize resources
                self.db = await connect_database()

            async def pre_server_shutdown(self, mcp: FastMCP) -> None:
                # Clean up resources
                await self.db.close()

            # post_server_start not implemented - will use safe default (no-op)
        ```
    """

    async def pre_server_start(self, mcp: FastMCP) -> None:
        """
        Call before the server starts.

        Use this to:
        - Initialize resources
        - Set up connections
        - Validate configuration
        - Prepare server state

        Args:
            mcp: The FastMCP instance that will be started

        Note:
            Override this method in your subclass to add custom initialization.
            The default implementation is a safe no-op.
        """
        pass

    async def post_server_start(self, mcp: FastMCP) -> None:
        """
        Call after the server has started and is ready to handle requests.

        Use this to:
        - Register additional handlers
        - Start background tasks
        - Initialize delayed resources
        - Log startup completion

        Args:
            mcp: The running FastMCP instance

        Note:
            Override this method in your subclass to add post-startup logic.
            The default implementation is a safe no-op.
        """
        pass

    async def pre_server_shutdown(self, mcp: FastMCP) -> None:
        """
        Call before the server shuts down.

        Use this to:
        - Close database connections
        - Save application state
        - Clean up temporary files
        - Stop background tasks
        - Release resources

        Args:
            mcp: The running FastMCP instance

        Note:
            Override this method in your subclass to add cleanup logic.
            The default implementation is a safe no-op.
            This is ALWAYS called, even on Ctrl+C or errors.
        """
        pass

pre_server_start async 

pre_server_start(mcp: FastMCP) -> None

Call before the server starts.

Use this to: - Initialize resources - Set up connections - Validate configuration - Prepare server state

Parameters:

Name Type Description Default
mcp FastMCP

The FastMCP instance that will be started

 required
Note

Override this method in your subclass to add custom initialization. The default implementation is a safe no-op.

Source code in datarobot_genai/drmcp/core/server_life_cycle.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
async def pre_server_start(self, mcp: FastMCP) -> None:
    """
    Call before the server starts.

    Use this to:
    - Initialize resources
    - Set up connections
    - Validate configuration
    - Prepare server state

    Args:
        mcp: The FastMCP instance that will be started

    Note:
        Override this method in your subclass to add custom initialization.
        The default implementation is a safe no-op.
    """
    pass

post_server_start async 

post_server_start(mcp: FastMCP) -> None

Call after the server has started and is ready to handle requests.

Use this to: - Register additional handlers - Start background tasks - Initialize delayed resources - Log startup completion

Parameters:

Name Type Description Default
mcp FastMCP

The running FastMCP instance

 required
Note

Override this method in your subclass to add post-startup logic. The default implementation is a safe no-op.

Source code in datarobot_genai/drmcp/core/server_life_cycle.py 
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
async def post_server_start(self, mcp: FastMCP) -> None:
    """
    Call after the server has started and is ready to handle requests.

    Use this to:
    - Register additional handlers
    - Start background tasks
    - Initialize delayed resources
    - Log startup completion

    Args:
        mcp: The running FastMCP instance

    Note:
        Override this method in your subclass to add post-startup logic.
        The default implementation is a safe no-op.
    """
    pass

pre_server_shutdown async 

pre_server_shutdown(mcp: FastMCP) -> None

Call before the server shuts down.

Use this to: - Close database connections - Save application state - Clean up temporary files - Stop background tasks - Release resources

Parameters:

Name Type Description Default
mcp FastMCP

The running FastMCP instance

 required
Note

Override this method in your subclass to add cleanup logic. The default implementation is a safe no-op. This is ALWAYS called, even on Ctrl+C or errors.

Source code in datarobot_genai/drmcp/core/server_life_cycle.py 
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
async def pre_server_shutdown(self, mcp: FastMCP) -> None:
    """
    Call before the server shuts down.

    Use this to:
    - Close database connections
    - Save application state
    - Clean up temporary files
    - Stop background tasks
    - Release resources

    Args:
        mcp: The running FastMCP instance

    Note:
        Override this method in your subclass to add cleanup logic.
        The default implementation is a safe no-op.
        This is ALWAYS called, even on Ctrl+C or errors.
    """
    pass