Vapor 系列教程 - MySQL 数据库

Vapor 默认使用的 SQLite 数据库

创建新项目

vapor new Test

修改相关文件项

Package.swift

import PackageDescription

let package = Package(
    name: "HeyAT",
    dependencies: [
        .package(url: "https://github.com/vapor/vapor.git", from: "3.0.0"),
        .package(url: "https://github.com/vapor/crypto.git", from: "3.3.0"),
        .package(url: "https://github.com/vapor/fluent-mysql.git", from: "3.0.0")
    ],
    targets: [
        .target(name: "App", dependencies: ["FluentMySQL", "Crypto", "Vapor"]),
        .target(name: "Run", dependencies: ["App"]),
        .testTarget(name: "AppTests", dependencies: ["App"])
    ]
)

修改 configure.swift

import FluentMySQL
import Vapor

/// Called before your application initializes.
public func configure(_ config: inout Config, _ env: inout Environment, _ services: inout Services) throws {
    // Register providers first
    try services.register(FluentMySQLProvider())
    
    // Register routes to the router
    let router = EngineRouter.default()
    try routes(router)
    services.register(router, as: Router.self)
    
    // Register middleware
    var middlewares = MiddlewareConfig() // Create _empty_ middleware config
    // middlewares.use(FileMiddleware.self) // Serves files from `Public/` directory
    middlewares.use(ErrorMiddleware.self) // Catches errors and converts to HTTP response
    services.register(middlewares)
    
    // Configure a MySQL database
    var databases =  DatabasesConfig()
    
    //定义mysql数据库配置参数的变量
    let mysqlHost: String
    let mysqlPort: Int
    let mysqlDB: String
    let mysqlUser: String
    let mysqlPass: String
    
    if env == .development || env == .testing {
        //线下环境使用的配置信息
        mysqlHost = "127.0.0.1"
        mysqlPort = 3306
        mysqlDB = "vapor"
        mysqlUser = "root"
        mysqlPass = "qiqi.loveAI001"
    }
    else {
        //生产环境，读取机器配置的配置，如果读取失败，则取默认值
        mysqlHost = Environment.get("MYSQL_HOST") ?? "127.0.0.1"
        mysqlPort = 3306
        mysqlDB = Environment.get("MYSQL_DB") ?? "vapor"
        mysqlUser = Environment.get("MYSQL_USER") ?? "root"
        mysqlPass = Environment.get("MYSQL_PASS") ?? "qiqi.loveAI001"
    }
    
    let mysqlConfig = MySQLDatabaseConfig(
        hostname: mysqlHost,
        port: mysqlPort,
        username: mysqlUser,
        password: mysqlPass,
        database: mysqlDB,
        transport: .unverifiedTLS)
    
    let mysql = MySQLDatabase(config: mysqlConfig)
    databases.add(database: mysql, as: .mysql)
    services.register(databases)
    
    // Migration配置
    Business.defaultDatabase = .mysql
    var migrations = MigrationConfig()
    migrations.add(model: Business.self, database: .mysql)
    migrations.add(model: Instance.self, database: .mysql)
    migrations.add(model: User.self, database: .mysql)
    migrations.add(model: AccessToken.self, database: .mysql)
    
    //开发环境填充测试数据
    if env == .development {
        migrations.add(migration: BusinessSeeder.self, database: .mysql)
//        migrations.add(migration: InstanceSeeder.self, database: .mysql)
    }
    services.register(migrations)
    
    /**
     # 注册migrations命令
     - vapor run migrate: 执行migrate创建表，并执行seed构建测试数据
     - vapor run revert -all: 依照migrate创建表顺序，依次清空表数据后删除该表
     */
    var commandConfig = CommandConfig.default()
    commandConfig.useFluentCommands()
    services.register(commandConfig)
    
}

修改 Todo.swift

import FluentMySQL
import Vapor

/// A single entry of a Todo list.
final class Todo: MySQLModel {
    /// The unique identifier for this `Todo`.
    var id: Int?

    /// A title describing what this `Todo` entails.
    var title: String

    /// Creates a new `Todo`.
    init(id: Int? = nil, title: String) {
        self.id = id
        self.title = title
    }
}

/// Allows `Todo` to be used as a dynamic migration.
extension Todo: Migration { }

/// Allows `Todo` to be encoded to and decoded from HTTP messages.
extension Todo: Content { }

/// Allows `Todo` to be used as a dynamic parameter in route definitions.
extension Todo: Parameter { }