Vapor には Command というモジュールがあり、これを利用すると CLI のコマンドを作ることができる (ややこしいことに、これを使いたい場合は vapor/console を依存関係に追加する必要がある) 。

だが、その中でデータベースにアクセスしようとして困った。

やろうとしたことは、

コマンドを実行してユーザーを作りたい

というもので、当然データベースにデータを INSERT する必要がある。

CLI を作る場合、Command プロトコルに準拠した型を用意し、その run メソッドに処理を実装するのだが、その際に利用できる CommandContext というコンテナは DatabaseConnectable を満たさないのだ。

argument type 'Container' does not conform to expected type 'DatabaseConnectable'

で、質問して教えてもらった方法が、ここにあるように、Conitainer には withNewConnection という拡張メソッドがあり、それを使うとクロージャに Database.Connection が渡るのでデータベースにアクセスすることができる。

例えばやりたかったことは↓で実現可能。


...

    func run(using context: CommandContext) throws -> Future<Void> {
        let email = try context.argument("email")
        let password = try context.requireOption("password")
        let passwordHash = try BCrypt.hash(password)
        let user = User(email: email, password: passwordHash)
        return context.container.withNewConnection(to: .psql) { db in
            return user.save(on: db).transform(to: ())
        }
    }

...

ちなみに newConnection というメソッドもあるが、こちらは自分でコネクションを閉じる必要があるため、普通は withNewConnection を使ったほうが良いと思う。