CockroachDB MCP Server

Instructions

Returns:
    Commit status.

Implementation Reference

• src/cockroachdb_mcp/server.py:391-401 (registration)

  MCP tool registration using @mcp.tool() decorator. Thin wrapper that delegates to connection_manager.commit_transaction() and handles exceptions.

  @mcp.tool()
  async def commit_transaction() -> dict[str, Any]:
      """Commit the current transaction.
  
      Returns:
          Commit status.
      """
      try:
          return await connection_manager.commit_transaction()
      except Exception as e:
          return {"status": "error", "error": str(e)}

• src/cockroachdb_mcp/connection.py:267-284 (handler)

  Core implementation of commit_transaction method in ConnectionManager class. Ensures connection, validates transaction state, commits the transaction, resets autocommit, and updates state.

  async def commit_transaction(self) -> dict[str, Any]:
      """Commit the current transaction.
  
      Returns:
          Commit status.
      """
      conn = await self.ensure_connected()
  
      if not self._state.in_transaction:
          return {"status": "error", "error": "No transaction in progress"}
  
      try:
          await conn.commit()
          await conn.set_autocommit(True)
          self._state.in_transaction = False
          return {"status": "committed", "message": "Transaction committed"}
      except Exception as e:
          return {"status": "error", "error": str(e)}