使用 Swift 3.0 操作 MySQL 資料庫
- 高洛峰原創
- 2016-11-22 10:20:441847瀏覽
開始
在這篇教學中，我們採用 Ubuntu 16.04 系統和 MySQL 5.7。 MySQL 5.7 引進了一系列的新功能。其中一個就是提供了更有效率的儲存 JSON 資料的能力，同時提供了查詢 JSON 資料內部的能力。稍後如果 MySQL 5.7 成為了 Ubuntu 16.04 上預設的 MySQL 版本以後，我們將使用 Ubuntu 16.04 作為我們的作業系統。
如果你還沒安裝 Swift， 你可以使用 apt-get 方式來安裝。請參閱這篇文章的說明安裝。 2016 年 9 月底，蘋果也開始在 Ubuntu16.04 上編譯 Swift 的鏡像。請查看 Swift.org 以取得更多的資訊。
建立資料庫
我們把資料庫命名為 swift_test, 指派的使用者是 swift，密碼是 swiftpass，如果你熟悉 MySQL，你應該知道需要執行 GRANT ALL ON swift_test.* 進行授權。
下面是這部分的命令：
# sudo mysql ... mysql> create user swift; Query OK, 0 rows affected (0.00 sec) mysql> create database swift_test; Query OK, 1 row affected (0.00 sec) mysql> grant all on swift_test.* to 'swift'@'localhost' identified by 'swiftpass'; Query OK, 0 rows affected, 1 warning (0.00 sec) mysql> flush privileges; Query OK, 0 rows affected (0.00 sec) mysql> quit Bye
創建Swift 套件
現在開始正式進行編碼，首先創建一個套件：
# mkdir swift_mysql # swift package init --type executable
編寫Package.swift 檔案：
import PackageDescription
let package = Package(
name: "swift_mysql",
dependencies:[
.Package(url:"https://github.com/vapor/mysql", majorVersion:1)
]
)第二步，我們使用一些輔助的工具程式碼來產生一些隨機的數據，填充到資料庫中。在 Sources 目錄下加入 utils.swift 檔案並在裡面加入以下內容：
import Glibc
class Random {
static let initialize:Void = {
srandom(UInt32(time(nil)))
return ()
}()
}
func randomString(ofLength length:Int) -> String {
Random.initialize
let charactersString = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
let charactersArray:[Character] = Array(charactersString.characters)
var string = ""
for _ in 0..<length {
string.append(charactersArray[Int(random()) % charactersArray.count])
}
return string
}
func randomInt() -> Int {
Random.initialize
return Int(random() % 10000)
}Vapor MySQL
接下來是真正的程式碼，我們的 main.swift 檔案使用了 Vapor MySQL 模組。
連接資料庫
將以下程式碼新增以下程式碼到 Sources/main.swift 中：
import Glibc
import MySQL
var mysql:Database
do {
mysql = try Database(host:"localhost",
user:"swift",
password:"swiftpass",
database:"swift_test")
try mysql.execute("SELECT @@version")
} catch {
print("Unable to connect to MySQL: \(error)")
exit(-1)
}以上程式碼設定資料庫並且處理 mysql。建構器 Database(host:String, user:String, password:String, database:String) 一目了然。
語句 try mysql.execute("SELECT @@version”) 是用來測試保證我們連接正確，並且成功連接到了資料庫。如果 do 程式碼區塊運作無錯誤，接下來就可以開始操作資料庫了！
整數和字串
所有對 MySQL 的呼叫都會透過 execute(_:String) 方法。要注意的是這個方法和一些抽象API 的方法不同，例如 .create(table:String, ...) 或 .insert(table:String, …。execute 取得原始的SQL 語句並傳給MySQL 連接器。
do {
try mysql.execute("DROP TABLE IF EXISTS foo")
try mysql.execute("CREATE TABLE foo (bar INT(4), baz VARCHAR(16))")
for i in 1...10 {
let int = randomInt()
let string = randomString(ofLength:16)
try mysql.execute("INSERT INTO foo VALUES (\(int), '\(string)')")
}
// Query
let results = try mysql.execute("SELECT * FROM foo")
for result in results {
if let bar = result["bar"]?.int,
let baz = result["baz"]?.string {
print("\(bar)\t\(baz)")
}
}
} catch {
print("Error: \(error)")
exit(-1)
}查詢結果也是使用的 execute(_:String) 方法。
Node 类型是 Vapor 中的数据结构，用于转化为不同的类型。你可以从这里获取更多的信息。使用 Node 类型来表达 MySQL 可以方便的转换成对应的 Swift 类型。比如：let bar = result["bar"]?.int 给我们一个整型。
继续
接着我们来看一些更复杂的例子，比如创建一个表，包含了 MySQL 的 DATE， POINT 和 JSON 数据类型。我们的表名叫 samples。
do {
try mysql.execute("DROP TABLE IF EXISTS samples")
try mysql.execute("CREATE TABLE samples (id INT PRIMARY KEY AUTO_INCREMENT, created_at DATETIME, location POINT, reading JSON)")
// ... Date
// ... Point
// ... Sample
// ... Insert
// ... Query
} catch {
print("Error: \(error)")
exit(-1)
}要插入一个日期到数据库中，需要正确的 SQL 语句：
// ... Date let now = Date() let formatter = DateFormatter() formatter.dateFormat = "yyyy-MM-dd HH:mm:ss" // MySQL will accept this format let created_at = formatter.string(from:date)
接下来使用 Swift 元组来创建一个 POINT：
// ... Point let location = (37.20262, -112.98785) // latitude, longitude
最后，我们来处理 MySQL 5.7 中新的 JSON 数据类型，此外我们使用了 Jay 包来快速将一个 Swift 字典 [String:Any] 转换为 JSON 格式的字符串。
// ... Sample
let sample:[String:Any] = [
"heading":90,
"gps":[
"latitude":37.20262,
"longitude":-112.98785
],
"speed":82,
"temperature":200
]提示：你不需要显式在 Package.swift 中声明对 Jay 的依赖，因为在 MySQL 的包中已经包含了这个依赖。接下来我们把 JSON 数据转换为 String，用来拼凑 MySQL 语句。
let sampleData = try Jay(formatting:.minified).dataFromJson(any:sample) // [UInt8] let sampleJSON = String(data:Data(sampleData), encoding:.utf8)
这样我们就有了 date， point 和 JSON 字符串（sample) 了， 现在添加数据到 sample 表中：
// ... Insert let stmt = "INSERT INTO samples (created_at, location, sample) VALUES ('\(created_at)', POINT\(point), '\(sampleJSON)')" try mysql.execute(stmt)
请注意我们在处理 POINT 时候，使用了一些技巧。在对 (point) 展开为字符串 (37.20262, -112.98785) 后，完整的字符串是 POINT(37.20262, -112.98785)，这是 MySQL 所需要的数据，整个语句的字符串如下：
INSERT INTO samples (created_at, location, sample) VALUES ('2016-09-21 22:28:44', POINT(37.202620000000003, -112.98784999999999), '{"gps":{"latitude":37.20262,"longitude":-112.98785},"heading":90,"speed":82,"temperature":200}')获取结果
警告：在写这篇文章的时候(2016-09-22), Vapor MySQL 1.0.0 有一个 bug：在读取 POINT 数据类型时会 crash 掉，所以不得不在下面代码中加入 do 代码块，然后不使用 select 语句。我们在 Vapor MySQL 中记录了这个 issue，等这个 issue 修复以后，我们将更新文章。
在下面的例子中，我们将使用 MySQL 5.7 中引入对 JSON 数据内部的查询特性，使用 SELECT … WHERE 查询 JSON 数据。在这里查询的是 samples 表中 JSON 数据类型 sample中、speed 字段大于 80 的数据。
// ... 查询
let results = try mysql.execute("SELECT created_at,sample FROM samples where JSON_EXTRACT(sample, '$.speed') > 80")
for result in results {
if let sample = result["sample"]?.object,
let speed = sample["speed"]?.int,
let temperature = sample["temperature"]?.int,
let created_at = result["created_at"]?.string {
print("Time:\(created_at)\tSpeed:\(speed)\tTemperature:\(temperature)")
}
}这里做一些说明。JSON_EXTRACT 函数是用来 返回从 JSON 文档中的数据，根据传入的路径参数选择文档中满足条件的数据。在本例中，我们解包了列 sample 中的 speed 值。
为了循环处理结果，我们使用了 for result in results 语句，接着使用 if let 语句验证结果数据。首先使用 let sample = result["sample"]?.object 获取一个字典，对应 MySQL 中的 JSON 文档，这是一句关键的代码！Vapor MySQL 库并没有返回一个 String，而 String 还需进行 JSON 的解析。这个解析工作库已经帮你做了，所以你可以直接使用 sample 字典啦。
剩下的 let 语句给了我们 speed，temperature 和 created_at。注意 created_at 在 MySQL 中是 DATETIME 类型，我们读取它为字符串。为了在 Swift 中转换成 Date 类型，需要使用 .date(from:String) 方法加一个 DateFormatter 来做类型转换。
获取代码
如果你想直接运行代码，请到 github 上下载我们的代码。
在任何地方使用 swift build 进行编译，运行可执行代码，不要忘了你还需要拥有一个数据库，用户名并且授权通过。