如何在Vapor运行时检测Worker类型，仅Web Worker执行数据库迁移

在Vapor中，你可以通过几种内置方式判断当前Worker是Web类型还是队列（含定时队列）类型，从而实现仅Web Worker执行数据库迁移的需求：

方法1：检查HTTP服务器配置 #

Web Worker的核心标识是会启用HTTP服务，而队列Worker不会加载HTTP服务器配置。你可以直接通过app.http.server.configuration的端口判断：

public func configure(_ app: Application) throws {
    app.middleware = .init()

    // 仅当是Web Worker时执行迁移
    if app.http.server.configuration.port != 0 {
        try app.autoMigrate().wait()
        // 这里添加Web Worker专属的其他配置
    } else {
        // 队列Worker的专属逻辑
    }
}

默认情况下，Vapor的Web Worker会将端口设为8080（或环境变量指定的端口），而队列Worker的HTTP端口会保持默认的0（即不启动HTTP服务）。

方法2：通过环境变量标记 #

你可以在启动不同Worker时设置自定义环境变量，明确标记Worker类型：

public func configure(_ app: Application) throws {
    app.middleware = .init()

    let workerType = app.environment.get("WORKER_TYPE") ?? "web"
    if workerType == "web" {
        try app.autoMigrate().wait()
    } else {
        // 队列Worker逻辑
    }
}

启动命令示例：

• Web Worker：WORKER_TYPE=web swift run App serve
• 队列Worker：WORKER_TYPE=queue swift run App queues
• 定时队列Worker：WORKER_TYPE=scheduled-queue swift run App queues --scheduled

方法3：检查队列组件状态 #

队列Worker会初始化queues组件并配置队列名称，你可以通过这一点判断：

public func configure(_ app: Application) throws {
    app.middleware = .init()

    // 如果配置了队列名称，说明是队列Worker
    if !app.queues.configuration.queueNames.isEmpty {
        // 队列Worker逻辑
    } else {
        try app.autoMigrate().wait()
    }
}

推荐使用方法1，无需额外配置环境变量，完全利用Vapor的内置行为即可准确区分Worker类型，避免多Worker同时执行迁移导致的冲突。

内容的提问来源于stack exchange，提问作者seeppp