go使用mysql实现增删改查操作

程序员buddha2024-12-01 7:08

1、安装MySQL驱动

bash 复制代码 
go get -u github.com/go-sql-driver/mysql

2、go连接MySQL

go 复制代码 
import (
	"database/sql"
	"log"

	_ "github.com/go-sql-driver/mysql" // 导入 mysql 驱动
)

type Users struct {
	ID    int
	Name  string
	Email string
}

var db *sql.DB

func init() {
	// 打开或创建一个 SQLite 数据库文件
	var err error
	db, err = sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test?parseTime=true")
	if err != nil {
		log.Fatal(err)
	}
	// 测试连接
	if err = db.Ping(); err != nil {
		log.Fatal(err)
	}
	// 创建用户表
	createTableSQL := `
	CREATE TABLE IF NOT EXISTS users (
		id INT AUTO_INCREMENT PRIMARY KEY,
		name varchar(64) NOT NULL,
		email varchar(64) NOT NULL UNIQUE
	);
	`
	// 执行 SQL 语句
	_, err = db.Exec(createTableSQL)
	if err != nil {
		log.Fatal(err)
	}
}

3、users表增删改查SQL

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

4、操作增删改查操作

go 复制代码 
// 创建用户
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)

整体测试代码如下：

go 复制代码 
// main.go
package main

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

	_ "github.com/go-sql-driver/mysql" // 导入 mysql 驱动
)

type Users struct {
	ID    int
	Name  string
	Email string
}

var db *sql.DB

func init() {
	// 打开或创建一个 SQLite 数据库文件
	var err error
	db, err = sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test?parseTime=true")
	if err != nil {
		log.Fatal(err)
	}
	// 测试连接
	if err = db.Ping(); err != nil {
		log.Fatal(err)
	}
	// 创建用户表
	createTableSQL := `
	CREATE TABLE IF NOT EXISTS users (
		id INT AUTO_INCREMENT PRIMARY KEY,
		name varchar(64) NOT NULL,
		email varchar(64) 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函数结束...")
}
相关推荐
艾莉丝努力练剑3 分钟前
【C语言】学习过程教训与经验杂谈：思想准备、知识回顾（三）
c语言·开发语言·数据结构·学习·算法
叁沐21 分钟前
MySQL 09 普通索引和唯一索引，应该怎么选择？
mysql
witton1 小时前
Go语言网络游戏服务器模块化编程
服务器·开发语言·游戏·golang·origin·模块化·耦合
叹一曲当时只道是寻常2 小时前
Softhub软件下载站实战开发（十六）：仪表盘前端设计与实现
前端·golang
草履虫建模2 小时前
Redis：高性能内存数据库与缓存利器
java·数据库·spring boot·redis·分布式·mysql·缓存
苹果醋32 小时前
Vue3组合式API应用：状态共享与逻辑复用最佳实践
java·运维·spring boot·mysql·nginx
一个处女座的测试2 小时前
Python语言+pytest框架+allure报告+log日志+yaml文件+mysql断言实现接口自动化框架
python·mysql·pytest
枯萎穿心攻击2 小时前
ECS由浅入深第三节：进阶？System 的行为与复杂交互模式
开发语言·unity·c#·游戏引擎
Jerry Lau2 小时前
go go go 出发咯 - go web开发入门系列（一） helloworld
开发语言·前端·golang
nananaij2 小时前
【Python基础入门 re模块实现正则表达式操作】
开发语言·python·正则表达式
热门推荐
01集群聊天服务器---MySQL数据库的建立02Java学习第十五部分——MyBatis03Coze扣子平台完整体验和实践（附国内和国际版对比）04基于odoo17的设计模式详解---单例模式05使用Ruby接入实时行情API教程06扣子（coze）实战|我用扣子搭建了一个自动分析小红薯笔记内容的AI应用|详细步骤拆解07《深入设计模式》模式结构汇总08基于odoo17的设计模式详解---装饰模式09DeepSeek各版本说明与优缺点分析10Java类变量（静态变量）