go使用sqlite3数据库实现CURD操作

🕗 发布于 2024-11-30 06:24 数据库 golang sqlite

工具类项目慢慢喜欢使用sqlite3数据库。工具类项目不存在并发问题，部署项目也不需要额外去安装数据库服务端，非常省事。

1、安装sqlite3驱动

go get github.com/mattn/go-sqlite3

2、go连接sqlite3初始化方法

var db *sql.DB

func init() {
// 打开或创建一个 SQLite 数据库文件
var err error
db, err = sql.Open("sqlite3", "./db.db")
if err != nil {
log.Fatal(err)
}
// 测试连接
if err = db.Ping(); err != nil {
log.Fatal(err)
}
}

这里涉及到一个init()函数。在 Go 语言中，init() 函数是一个特殊的函数，它在程序开始执行时自动调用，这里就可以拿来连接数据库初始化操作。每个包都可以包含任意数量的 init() 函数，并且这些函数会在包被导入时自动执行。init() 函数通常用于初始化操作，例如设置变量、打开文件或数据库连接等。

init() 函数的特点：

自动调用：init() 函数不需要显式调用，它们在包被导入时自动执行。
执行顺序：
- 如果一个包中有多个 init() 函数，它们会按照在源代码中的顺序依次执行。
- 如果一个包导入了其他包，那么被导入包的 init() 函数会在导入它的包的 init() 函数之前执行。
- 无参数和返回值：init() 函数没有参数也没有返回值。
- 不能被其他函数调用：init() 函数只能由运行时系统调用，不能被其他函数调用。

3、测试Users结构体

type Users struct {
ID    int
Name  string
Email string
}

结构体字段的首字母是否大写决定了该字段是否对包外可见（即是否具有公共访问权限）

知识点：

首字母大写：如果一个标识符（包括结构体字段、函数名、变量名等）的首字母是大写的，那么它就是可以被其他包访问的，即它是“导出的”或“公开的”。
首字母小写：如果一个标识符的首字母是小写的，那么它只能在其定义的包内部访问，对于其他包是不可见的，即它是“未导出的”或“私有的”。

4、users表CURD操作

// CreateUser 创建新用户
func (u *Users) CreateUser() (int64, error) {
stmt, err := db.Prepare("INSERT INTO users (name, email) VALUES (?, ?)")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.Name, u.Email)
if err != nil {
return 0, err
}
return res.LastInsertId()
}

// GetUserByID 根据 ID 获取用户
func (u *Users) GetUserByID(id int) error {
row := db.QueryRow("SELECT id, name, email FROM users WHERE id = ?", id)
return row.Scan(&u.ID, &u.Name, &u.Email)
}

// GetAllUsers 获取所有用户
func GetAllUsers() ([]*Users, error) {
rows, err := db.Query("SELECT id, name, email FROM users")
if err != nil {
return nil, err
}
defer rows.Close()

var users []*Users
for rows.Next() {
user := &Users{}
if err := rows.Scan(&user.ID, &user.Name, &user.Email); err != nil {
return nil, err
}
users = append(users, user)
}
if err := rows.Err(); err != nil {
return nil, err
}
return users, nil
}

// UpdateUser 更新用户信息
func (u *Users) UpdateUser() (int64, error) {
stmt, err := db.Prepare("UPDATE users SET name = ?, email = ? WHERE id = ?")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.Name, u.Email, u.ID)
if err != nil {
return 0, err
}
return res.RowsAffected()
}

// DeleteUser 删除用户
func (u *Users) DeleteUser() (int64, error) {
stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.ID)
if err != nil {
return 0, err
}
return res.RowsAffected()
}

这里的CURD操作，都是方法。方法是一种特殊类型的函数，它有一个接收者（receiver）。接收者可以是任何类型的值或指针。这里的接收者就是(u *Users)，用的是指针。

5、go使用sqlite3数据库实现CURD操作

// main.go
package main

import (
"database/sql"
"fmt"
"log"

_ "github.com/mattn/go-sqlite3" // 导入 sqlite3 驱动
)

type Users struct {
ID    int
Name  string
Email string
}

var db *sql.DB

func init() {
// 打开或创建一个 SQLite 数据库文件
var err error
db, err = sql.Open("sqlite3", "./db.db")
if err != nil {
log.Fatal(err)
}
// 测试连接
if err = db.Ping(); err != nil {
log.Fatal(err)
}
// 创建用户表
createTableSQL := `
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
email TEXT NOT NULL UNIQUE
);
`
// 执行 SQL 语句
_, err = db.Exec(createTableSQL)
if err != nil {
log.Fatal(err)
}
}

// CreateUser 创建新用户
func (u *Users) CreateUser() (int64, error) {
stmt, err := db.Prepare("INSERT INTO users (name, email) VALUES (?, ?)")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.Name, u.Email)
if err != nil {
return 0, err
}
return res.LastInsertId()
}

// GetUserByID 根据 ID 获取用户
func (u *Users) GetUserByID(id int) error {
row := db.QueryRow("SELECT id, name, email FROM users WHERE id = ?", id)
return row.Scan(&u.ID, &u.Name, &u.Email)
}

// GetAllUsers 获取所有用户
func GetAllUsers() ([]*Users, error) {
rows, err := db.Query("SELECT id, name, email FROM users")
if err != nil {
return nil, err
}
defer rows.Close()

var users []*Users
for rows.Next() {
user := &Users{}
if err := rows.Scan(&user.ID, &user.Name, &user.Email); err != nil {
return nil, err
}
users = append(users, user)
}
if err := rows.Err(); err != nil {
return nil, err
}
return users, nil
}

// UpdateUser 更新用户信息
func (u *Users) UpdateUser() (int64, error) {
stmt, err := db.Prepare("UPDATE users SET name = ?, email = ? WHERE id = ?")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.Name, u.Email, u.ID)
if err != nil {
return 0, err
}
return res.RowsAffected()
}

// DeleteUser 删除用户
func (u *Users) DeleteUser() (int64, error) {
stmt, err := db.Prepare("DELETE FROM users WHERE id = ?")
if err != nil {
return 0, err
}
res, err := stmt.Exec(u.ID)
if err != nil {
return 0, err
}
return res.RowsAffected()
}

func main() {
fmt.Println("main函数开始...")

// 创建用户
user := &Users{Name: "buddha", Email: "3539949705@qq.com"}
id, err := user.CreateUser()
if err != nil {
log.Fatalf("Failed to create user: %v", err)
}
fmt.Printf("Created user with ID: %d\n", id)

// 获取用户
user = &Users{}
if err := user.GetUserByID(int(id)); err != nil {
log.Fatalf("Failed to get user: %v", err)
}
fmt.Printf("User: ID: %d, Name: %s, Email: %s\n", user.ID, user.Name, user.Email)

// 更新用户
user.Name = "buddha2080"
user.Email = "3539949704@qq.com"
affectedRows, err := user.UpdateUser()
if err != nil {
log.Fatalf("Failed to update user: %v", err)
}
fmt.Printf("Updated %d rows\n", affectedRows)

// 获取所有用户
users, err := GetAllUsers()
if err != nil {
log.Fatalf("Failed to get all users: %v", err)
}
for _, u := range users {
fmt.Printf("User: id: %d, name: %s, email: %s\n", u.ID, u.Name, u.Email)
}

// 删除用户
affectedRows, err = user.DeleteUser()
if err != nil {
log.Fatalf("Failed to delete user: %v", err)
}
fmt.Printf("Deleted %d rows\n", affectedRows)

fmt.Println("main函数结束...")
}

在这里插入图片描述

原文地址：https://blog.csdn.net/rulaixiong/article/details/144132123

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：TsingtaoAI具身智能高校实训方案通过华为昇腾技术认证
下一篇：【视频】OpenCV：读写视频文件VideoCapture和VideoWriter

.NET(C#) 如何配置用户首选项及保存用户设置
.NET(C#) 如何配置用户首选项及保存用户设置
阅读更多2024-12-14
【最新】北大数字普惠金融指数数据集-省市县（2011-2023年）
郭峰,王靖一,王芳,孔涛,张勋,程志云.测度中国数字普惠金融发展:指数编制与空间特征[J].经济学(季刊),2020,19(04):1401-1418.时间跨度：省级和城市级指数时间跨度为2011-2
阅读更多2024-12-14
GESP202412 四级【Recamán】题解（AC）
a11ak−1−kkakak−1−kak−1k小杨想知道 Recamán 数列的前n项从小到大排序后的结果。手动计算非常困难，小杨希望你能帮他解决这个问题。
阅读更多2024-12-14
IDEA遇到EasyConnect中的网络资源无法访问的问题
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。原文链接：https://blog.csdn.net/wanshanyu_/article/de
阅读更多2024-12-14
双目摄像头标定方法
此时已经完成标定，左下角为反投影误差，右边为外参可视化。将双目左右目拍的图像上传（左右目最好不少于20张）此时回到主页面，即可看到成功导出。把这些误差大的删除即可。
阅读更多2024-12-14
Servlet、omcat服务器架构与工作原理
Servlet是运行在服务器端的Java程序，它的主要职责之一是接收并处理来自客户端（如浏览器）的HTTP请求。当客户端发送一个请求到服务器时，Servlet可以解析请求中的信息，例如请求的URL路径
阅读更多2024-12-14
Vue生命周期钩子函数：深入解析与实践
作为高级Vue前端开发人员，对Vue组件的生命周期钩子函数有着深刻的理解是至关重要的。生命周期钩子函数是指在Vue组件的创建、更新、销毁等过程中，Vue自动调用的一系列方法。通过这些钩子函数，我们可以
阅读更多2024-12-14
安卓开发--使用android studio发布APP
app发布
阅读更多2024-12-14
数据结构与算法学习笔记----拓扑排序
@ author: 明月清了个风。
阅读更多2024-12-14
python 将数据保存到现有的Excel文件的新工作表
out_file = ‘query.xlsx’df1 = pd.DataFrame(out_data)若直接写入：df1.to_excel(out_file, index=False, sheet_n
阅读更多2024-12-14