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

程序员buddha2024-11-30 12:32

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

1、安装sqlite3驱动
go 复制代码 
go get github.com/mattn/go-sqlite3
2、go连接sqlite3初始化方法
go 复制代码 
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结构体
go 复制代码 
type Users struct {
	ID    int
	Name  string
	Email string
}

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

知识点：

  • 首字母大写：如果一个标识符（包括结构体字段、函数名、变量名等）的首字母是大写的，那么它就是可以被其他包访问的，即它是"导出的"或"公开的"。
  • 首字母小写：如果一个标识符的首字母是小写的，那么它只能在其定义的包内部访问，对于其他包是不可见的，即它是"未导出的"或"私有的"。
4、users表CURD操作
go 复制代码 
// 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操作
go 复制代码 
// 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: "[email protected]"}
	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 = "[email protected]"
	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函数结束...")
}
相关推荐
ErizJ29 分钟前
Golang｜分布式索引架构
开发语言·分布式·后端·架构·golang
layman052832 分钟前
node.js 实战——mongoDB
数据库·mongodb·node.js
爱可生开源社区1 小时前
SQLShift 全新上线：Oracle→OceanBase 迁移利器
数据库
wkj0012 小时前
java 和 C#操作数据库对比
java·数据库·c#
编程在手天下我有2 小时前
Redis 数据类型全览：特性、场景与操作实例
数据库·redis·数据结构与算法
左灯右行的爱情2 小时前
缓存并发更新的挑战
jvm·数据库·redis·后端·缓存
Qiuner3 小时前
软件设计师速通其一：计算机内部数据表示
服务器·数据库·信号处理
文牧之4 小时前
PostgreSQL oracle_fdw 扩展解析
运维·数据库·postgresql
一个天蝎座 白勺 程序猿5 小时前
Python爬虫（9）Python数据存储实战：基于pymysql的MySQL数据库操作详解
数据库·python·mysql
Bug哆哆5 小时前
数据库安装和升级和双主配置
数据库
热门推荐
01西电B测-计算机网络综合实验(含验收问题)02KGG转MP3工具|非KGM文件|解密音频03从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑04Coze扣子平台完整体验和实践（附国内和国际版对比）05YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】06DeepSeek各版本说明与优缺点分析07最新 Kubernetes 集群部署 + flannel 网络插件（保姆级教程，最新 K8S 版本）08yolov8，yolo11，yolo12 服务器训练到部署全流程 笔记09我决定放弃搞 Java 了10yolov10来了！用yolov10训练自己的数据集（原理、训练、部署、应用）