Golang操作sqlite3数据库教程
本文介绍sqlite数据库,如何使用sqlite3包操作轻量级关系型数据库。
sqlite概述
sqlite是嵌入式关系型数据库引擎,官方描述为自包含的、无服务的、零配置并支持事务的关系型数据库引擎。我们在产品开发过程中经常使用,可以快速进行产品验证和部署,快速实现产品演示。
Go有sql包,它提供了sql(或类sql)数据库的通用接口。sql包必须与数据库驱动程序一起使用。
创建SQLITE数据库
使用sqlite3命令行工具创建数据库,查询数据。
sudo apt install sqlite3
安装好工具,创建数据库:
sqlite3 test.dbSQLite version 3.37.2 2022-01-06 13:25:41Enter ".help" for usage hints.sqlite>
test.db是sqlite3命令的参数,即为数据库名称,也是磁盘上的文件名。如果文件存在就打开,不存在则创建:
sqlite> .tablessqlite> .exit$ lstest.db
.tables名称列出test.db数据库中所有表,当前没有表;.exit命令终止活动session命令行工具。ls命令显示当前目录下的文件,可以看到我们创建的test.db文件。
打印版本
下面示例打印sqlite3的版本,主要通过SELECT SQLITE_VERSION()
语句返回版本号:
package mainimport ("database/sql""fmt""log"_ "github.com/mattn/go-sqlite3")func main() {db, err := sql.Open("sqlite3", ":memory:")if err != nil {log.Fatal(err)}defer db.Close()var version stringerr = db.QueryRow("SELECT SQLITE_VERSION()").Scan(&version)if err != nil {log.Fatal(err)}fmt.Println(version)}
首先导入包,这里导入前缀为下划线,则init函数被执行。然后注册驱动;
db, err := sql.Open("sqlite3", ":memory:")
Open函数指定驱动名称和数据源名称,本示例连接内存数据库。接着调用db.Close函数,确保关闭数据库并阻止启动新的查询。
err = db.QueryRow("SELECT SQLITE_VERSION()").Scan(&version)
QueryRow执行查询,至少返回一行。Scan函数从返回行拷贝列至version变量。
$ go run main.go3.39.4
Exec执行DML语句
Exec函数执行没有返回记录的查询语句,即DML语句:
package mainimport ("database/sql""fmt""log"_ "github.com/mattn/go-sqlite3")func main() {db, err := sql.Open("sqlite3", "test.db")if err != nil {log.Fatal(err)}defer db.Close()sts := `DROP TABLE IF EXISTS cars;CREATE TABLE cars(id INTEGER PRIMARY KEY, name TEXT, price INT);INSERT INTO cars(name, price) VALUES('Audi',52642);INSERT INTO cars(name, price) VALUES('Mercedes',57127);INSERT INTO cars(name, price) VALUES('Skoda',9000);INSERT INTO cars(name, price) VALUES('Volvo',29000);INSERT INTO cars(name, price) VALUES('Bentley',350000);INSERT INTO cars(name, price) VALUES('Citroen',21000);INSERT INTO cars(name, price) VALUES('Hummer',41400);INSERT INTO cars(name, price) VALUES('Volkswagen',21600);`_, err = db.Exec(sts)if err != nil {log.Fatal(err)}fmt.Println("table cars created")}
首先创建数据库,生成新的数据库文件:
db, err := sql.Open("sqlite3", "test.db")
接着创建表并插入数据。通过_, err = db.Exec(sts)
语句执行DML。
查询记录
查询语句返回符合条件记录,典型是SELECT,可选参数为占位符:
package mainimport ("database/sql""fmt""log"_ "github.com/mattn/go-sqlite3")func main() {db, err := sql.Open("sqlite3", "test.db")if err != nil {log.Fatal(err)}defer db.Close()rows, err := db.Query("SELECT * FROM cars")if err != nil {log.Fatal(err)}defer rows.Close()for rows.Next() {var id intvar name stringvar price interr = rows.Scan(&id, &name, &price)if err != nil {log.Fatal(err)}fmt.Printf("%d %s %d\n", id, name, price)}}
执行查询语句:
rows, err := db.Query("SELECT * FROM cars")
Next为Scan方法准备下一个结果行。成功返回True,如果没有结果或发生错误返回false。
for rows.Next() {var id intvar name stringvar price interr = rows.Scan(&id, &name, &price)if err != nil {log.Fatal(err)}fmt.Printf("%d %s %d\n", id, name, price)}
Scan方法获取字段值,最后打印记录的列值。
带参数的prepared 语句
使用prepared 语句,可以包括占位符代替直接在语句中写值,保证数据库操作的性能和安全。
package mainimport ( "database/sql" "fmt" "log" _ "github.com/mattn/go-sqlite3")func main() { db, err := sql.Open("sqlite3", "test.db") if err != nil { log.Fatal(err) } defer db.Close() stm, err := db.Prepare("SELECT * FROM cars WHERE id = ?") if err != nil { log.Fatal(err) } defer stm.Close() var id int var name string var price int cid := 3 err = stm.QueryRow(cid).Scan(&id, &name, &price) if err != nil { log.Fatal(err) } fmt.Printf("%d %s %d\n", id, name, price)}
使用prepare函数执行prepare语句查询特定行:
stm, err := db.Prepare("SELECT * FROM cars WHERE id = ?")
传入参数给QueryRow函数,以替换占位符:
err = stm.QueryRow(cid).Scan(&id, &name, &price)
也可以一步实现上面两步功能:
row := db.QueryRow("SELECT * FROM cars WHERE id = ?", cid)
返回影响行数
RowsAffected 返回执行DML语句影响的行数:
package mainimport ( "database/sql" "fmt" "log" _ "github.com/mattn/go-sqlite3")func main() { db, err := sql.Open("sqlite3", "test.db") if err != nil { log.Fatal(err) } defer db.Close() res, err := db.Exec("DELETE FROM cars WHERE id IN (1, 2, 3)") if err != nil { log.Fatal(err) } n, err := res.RowsAffected() if err != nil { log.Fatal(err) } fmt.Printf("The statement has affected %d rows\n", n)}
上面示例使用delete语句删除三行记录,然后打印结果进行验证。运行结果如下:
$ go run main.go The statement has affected 3 rows