Golang操作sqlite3数据库教程

  • A+
所属分类:编程茶楼

Golang操作sqlite3数据库教程

本文介绍sqlite数据库,如何使用sqlite3包操作轻量级关系型数据库

sqlite概述

sqlite是嵌入式关系型数据库引擎,官方描述为自包含的、无服务的、零配置并支持事务的关系型数据库引擎。我们在产品开发过程中经常使用,可以快速进行产品验证和部署,快速实现产品演示。

Go有sql包,它提供了sql(或类sql)数据库的通用接口。sql包必须与数据库驱动程序一起使用。

创建SQLITE数据库

使用sqlite3命令行工具创建数据库,查询数据。

sudo apt install sqlite3
  • 1

安装好工具,创建数据库:

sqlite3 test.db
SQLite version 3.37.2 2022-01-06 13:25:41
Enter ".help" for usage hints.
sqlite> 

  • 1
  • 2
  • 3
  • 4
  • 5

test.db是sqlite3命令的参数,即为数据库名称,也是磁盘上的文件名。如果文件存在就打开,不存在则创建:

sqlite> .tables
sqlite> .exit

$ ls
test.db

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

.tables名称列出test.db数据库中所有表,当前没有表;.exit命令终止活动session命令行工具。ls命令显示当前目录下的文件,可以看到我们创建的test.db文件。

打印版本

下面示例打印sqlite3的版本,主要通过SELECT SQLITE_VERSION()语句返回版本号:

package main

import (
	"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 string
	err = db.QueryRow("SELECT SQLITE_VERSION()").Scan(&version)

	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(version)
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30

首先导入包,这里导入前缀为下划线,则init函数被执行。然后注册驱动;

db, err := sql.Open("sqlite3", ":memory:")
  • 1

Open函数指定驱动名称和数据源名称,本示例连接内存数据库。接着调用db.Close函数,确保关闭数据库并阻止启动新的查询。

err = db.QueryRow("SELECT SQLITE_VERSION()").Scan(&version)
  • 1

QueryRow执行查询,至少返回一行。Scan函数从返回行拷贝列至version变量。

$ go run main.go
3.39.4

  • 1
  • 2
  • 3

Exec执行DML语句

Exec函数执行没有返回记录的查询语句,即DML语句:

package main

import (
	"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")
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41

首先创建数据库,生成新的数据库文件:

db, err := sql.Open("sqlite3", "test.db")
  • 1

接着创建表并插入数据。通过_, err = db.Exec(sts)语句执行DML。

查询记录

查询语句返回符合条件记录,典型是SELECT,可选参数为占位符:

package main

import (
	"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 int
		var name string
		var price int

		err = rows.Scan(&id, &name, &price)

		if err != nil {
			log.Fatal(err)
		}

		fmt.Printf("%d %s %d\n", id, name, price)
	}
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44

执行查询语句:

rows, err := db.Query("SELECT * FROM cars")
  • 1

Next为Scan方法准备下一个结果行。成功返回True,如果没有结果或发生错误返回false。

	for rows.Next() {

		var id int
		var name string
		var price int

		err = rows.Scan(&id, &name, &price)

		if err != nil {
			log.Fatal(err)
		}

		fmt.Printf("%d %s %d\n", id, name, price)
	}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15

Scan方法获取字段值,最后打印记录的列值。

带参数的prepared 语句

使用prepared 语句,可以包括占位符代替直接在语句中写值,保证数据库操作的性能和安全。

package main

import (
    "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)
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42

使用prepare函数执行prepare语句查询特定行:

stm, err := db.Prepare("SELECT * FROM cars WHERE id = ?")
  • 1

传入参数给QueryRow函数,以替换占位符:

err = stm.QueryRow(cid).Scan(&id, &name, &price)
  • 1

也可以一步实现上面两步功能:

row := db.QueryRow("SELECT * FROM cars WHERE id = ?", cid)
  • 1

返回影响行数

RowsAffected 返回执行DML语句影响的行数:

package main

import (
    "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)
}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34

上面示例使用delete语句删除三行记录,然后打印结果进行验证。运行结果如下:

$ go run main.go 
The statement has affected 3 rows
  • 1
  • 2
  • 我的微信
  • 这是我的微信扫一扫
  • weinxin
  • 我的微信公众号
  • 我的微信公众号扫一扫
  • weinxin