Go语言操作SQLite数据库|极客教程

Go语言操作SQLite数据库

SQLite是一种轻量级的嵌入式关系型数据库，不需要独立的服务器进程，将数据库引擎与应用程序打包在一起。Go语言是一种简洁、高效的编程语言，有着很好的并发性能和丰富的标准库。在Go中操作SQLite数据库可以通过第三方库实现，本文将介绍如何使用Go语言操作SQLite数据库。

安装SQLite

在开始使用SQLite之前，需要先安装SQLite数据库。可以访问SQLite的官方网站下载最新的SQLite安装包进行安装。安装完成后，可以通过命令行或者专门的SQLite工具来操作数据库。

Go语言SQLite库

Go语言提供了多个第三方库用于操作SQLite数据库，其中比较常用的是github.com/mattn/go-sqlite3库。该库提供了与SQLite数据库进行交互的API，并且易于使用。

可以通过以下命令安装go-sqlite3库：

go get github.com/mattn/go-sqlite3

初始化数据库连接

在Go中操作SQLite数据库首先需要建立数据库连接。可以通过以下代码来初始化一个数据库连接：

package main

import (
    "database/sql"
    _ "github.com/mattn/go-sqlite3"
)

func main() {
    db, err := sql.Open("sqlite3", "test.db")
    if err != nil {
        panic(err)
    }
    defer db.Close()
}

在上面的代码中，使用sql.Open函数连接SQLite数据库，并指定了数据库文件名为test.db。连接成功之后，使用defer db.Close()来关闭数据库连接，避免资源泄漏。

创建表

一旦建立了数据库连接，就可以操作数据库表。可以通过以下代码来创建一个名为users的表：

func createTable(db *sql.DB) {
    sqlStmt := `
    CREATE TABLE users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT,
        age INTEGER
    );
    `
    _, err := db.Exec(sqlStmt)
    if err != nil {
        panic(err)
    }
}

在上面的代码中，使用CREATE TABLE语句创建了一个名为users的表，表中包含id、name和age三个字段。id字段为主键且自增，name字段为文本类型，age字段为整数类型。

插入数据

在表创建完成之后，可以通过以下代码向users表中插入数据：

func insertData(db *sql.DB, name string, age int) {
    sqlStmt := `INSERT INTO users (name, age) VALUES (?, ?)`
    _, err := db.Exec(sqlStmt, name, age)
    if err != nil {
        panic(err)
    }
}

在上面的代码中，使用INSERT INTO语句向users表中插入数据，通过?占位符来确保SQL注入漏洞的安全性。然后通过db.Exec方法执行SQL语句。

查询数据

可以通过以下代码查询数据：

func queryData(db *sql.DB) {
    rows, err := db.Query("SELECT * FROM users")
    if err != nil {
        panic(err)
    }
    defer rows.Close()

    for rows.Next() {
        var id int
        var name string
        var age int
        rows.Scan(&id, &name, &age)
        fmt.Println(id, name, age)
    }

    if err = rows.Err(); err != nil {
        panic(err)
    }
}

在上面的代码中，使用SELECT语句查询users表中的所有数据，通过rows.Scan方法将查询结果赋值给对应的变量，并打印出来。

更新数据

可以通过以下代码更新数据：

func updateData(db *sql.DB, id int, newName string, newAge int) {
    sqlStmt := "UPDATE users SET name = ?, age = ? WHERE id = ?"
    _, err := db.Exec(sqlStmt, newName, newAge, id)
    if err != nil {
        panic(err)
    }
}

在上面的代码中，使用UPDATE语句更新users表中的数据，通过?占位符传递要更新的字段值。

删除数据

可以通过以下代码删除数据：

func deleteData(db *sql.DB, id int) {
    sqlStmt := "DELETE FROM users WHERE id = ?"
    _, err := db.Exec(sqlStmt, id)
    if err != nil {
        panic(err)
    }
}