138 lines
5.0 KiB
Python
138 lines
5.0 KiB
Python
from cli.core import (
|
|
DEFAULT_CORE_SERVICES,
|
|
CliContextFactory,
|
|
CliExecutionService,
|
|
)
|
|
from cli.exit_codes import DEPENDENCY_ERROR, SUCCESS
|
|
from cli.runtime.db import DATABASE_RUNTIME, DatabaseRuntimeService
|
|
from cli.runtime.ops import OPERATIONS_RUNTIME, OperationsRuntimeService
|
|
|
|
from .presenter import OpsCommandPresenter
|
|
|
|
|
|
class OpsCommandController:
|
|
"""
|
|
运维命令控制器。
|
|
|
|
该控制器负责组织 `ops` 命令组的上下文准备、runtime 调用、
|
|
payload 注入，以及基于输出格式选择 presenter 或直接返回 JSON。
|
|
|
|
:param context_factory: CLI 上下文工厂
|
|
:param execution_service: CLI 执行服务
|
|
:param presenter: 运维命令文本渲染器
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
context_factory: CliContextFactory | None = None,
|
|
execution_service: CliExecutionService | None = None,
|
|
presenter: OpsCommandPresenter | None = None,
|
|
database_runtime: DatabaseRuntimeService | None = None,
|
|
operations_runtime: OperationsRuntimeService | None = None,
|
|
) -> None:
|
|
"""
|
|
初始化运维命令控制器。
|
|
|
|
:param context_factory: CLI 上下文工厂
|
|
:param execution_service: CLI 执行服务
|
|
:param presenter: 运维命令文本渲染器
|
|
:param database_runtime: 数据库运行时服务
|
|
:param operations_runtime: 运维运行时服务
|
|
:return: None
|
|
"""
|
|
self.context_factory = context_factory or DEFAULT_CORE_SERVICES.context_factory
|
|
self.execution_service = execution_service or DEFAULT_CORE_SERVICES.execution_service
|
|
self.presenter = presenter or OpsCommandPresenter()
|
|
self.database_runtime = database_runtime or DATABASE_RUNTIME
|
|
self.operations_runtime = operations_runtime or OPERATIONS_RUNTIME
|
|
|
|
def ping_db(self, env: str, output: str) -> None:
|
|
"""
|
|
检查数据库连接。
|
|
|
|
:param env: 当前命令运行环境
|
|
:param output: 输出格式
|
|
:return: None
|
|
"""
|
|
ctx = self.context_factory.build_readonly(env, output)
|
|
self.execution_service.complete_payload(
|
|
ctx, self.execution_service.run_async(self.database_runtime.ping_database())
|
|
)
|
|
|
|
def ping_redis(self, env: str, output: str) -> None:
|
|
"""
|
|
检查 Redis 连接。
|
|
|
|
:param env: 当前命令运行环境
|
|
:param output: 输出格式
|
|
:return: None
|
|
"""
|
|
ctx = self.context_factory.build_readonly(env, output)
|
|
self.execution_service.complete_payload(
|
|
ctx,
|
|
self.execution_service.run_async(self.operations_runtime.ping_redis()),
|
|
)
|
|
|
|
def health(self, env: str, output: str) -> None:
|
|
"""
|
|
输出基础健康检查结果。
|
|
|
|
:param env: 当前命令运行环境
|
|
:param output: 输出格式
|
|
:return: None
|
|
"""
|
|
ctx = self.context_factory.build_readonly(env, output)
|
|
db_status = self.execution_service.run_async(self.database_runtime.ping_database())
|
|
redis_status = self.execution_service.run_async(self.operations_runtime.ping_redis())
|
|
payload = {
|
|
'env': ctx.env,
|
|
'database': db_status,
|
|
'redis': redis_status,
|
|
'ok': db_status.get('ok', False) and redis_status.get('ok', False),
|
|
}
|
|
exit_code = SUCCESS if payload['ok'] else DEPENDENCY_ERROR
|
|
self.execution_service.complete_payload_with_text(
|
|
ctx,
|
|
payload,
|
|
text_builder=self.presenter.build_health_text,
|
|
default_exit_code=exit_code,
|
|
)
|
|
|
|
def deps(self, env: str, output: str, *, include_dev: bool) -> None:
|
|
"""
|
|
检查当前 CLI 和后端运行依赖版本。
|
|
|
|
:param env: 当前命令运行环境
|
|
:param output: 输出格式
|
|
:param include_dev: 是否附带输出开发依赖
|
|
:return: None
|
|
"""
|
|
ctx = self.context_factory.build_readonly(env, output)
|
|
payload = self.operations_runtime.get_dependency_versions(include_dev=include_dev)
|
|
exit_code = SUCCESS if payload.get('ok', False) else DEPENDENCY_ERROR
|
|
self.execution_service.complete_payload_with_text(
|
|
ctx,
|
|
payload,
|
|
text_builder=self.presenter.build_dependencies_text,
|
|
default_exit_code=exit_code,
|
|
)
|
|
|
|
def server_info(self, env: str, output: str) -> None:
|
|
"""
|
|
输出服务器运行时信息。
|
|
|
|
:param env: 当前命令运行环境
|
|
:param output: 输出格式
|
|
:return: None
|
|
"""
|
|
ctx = self.context_factory.build_readonly(env, output)
|
|
payload = self.execution_service.run_async(self.operations_runtime.get_server_info())
|
|
self.execution_service.complete_payload_with_text(
|
|
ctx,
|
|
payload,
|
|
text_builder=self.presenter.build_server_info_text,
|
|
default_exit_code=SUCCESS,
|
|
text_condition=lambda data: data.get('ok', False),
|
|
)
|