Golang操作数据库简单示例

news2025/3/15 21:48:16

目录

  • 准备工作
  • 准备数据
  • 创建项目
  • 连接数据库
  • 查询数据
  • 修改数据
  • 插入数据
  • 删除数据
  • 释放资源
  • 完整代码
  • 最终执行结果

准备工作

在开始之前,你需要确保自己安装了Golang的编程环境,安装MySQL数据库,有一个可以用于编写代码的编辑器或IDE工具。我在这里使用的编辑器是Fleet。

准备数据

将下面的SQL语句在MySQL中执行,创建数据库并插入数据.。

drop table if exists album;
create table album (
    id int auto_increment not null,
    title varchar(128) not null,
    artist varchar(255) not null,
    price decimal(5,2) not null,
    primary key (`id`)
);
insert into album
    (title, artist, price)
values
    ('Blue Train', 'John Coltrane', 56.99),
    ('Giant Steps', 'Greey Mulligan', 63.99),
    ('Jeru', 'Gerry Mulligan', 17.99),
    ('Sarah Vaughan', 'Sarah Vaughan', 34.98);

创建项目

进入终端,输入以下命令。(当然,你也可以手动创建)

mkdir data-access
cd ./data-access
go mod init example/data-access

在上面创建的目录下创建文件main.go,将以下代码粘贴到文件中。

package main

import (
	"database/sql"
	"errors"
	"fmt"
	"github.com/go-sql-driver/mysql"
	"log"
)

终端输入以下命令下载上面代码中引入的MySQL的驱动包

go mod tidy

连接数据库

在main.go中输入以下代码,连接数据库。

var db *sql.DB

func main() {
	// 设置连接属性
	cfg := mysql.Config{
		User:   "root",
		Passwd: "123456",
		Net:    "tcp",
		Addr:   "127.0.0.1:3306",
		DBName: "recordings",
	}

	// 获取数据库句柄
	var err error
	db, err = sql.Open("mysql", cfg.FormatDSN())
	if err != nil {
		log.Fatal(err)
	}

	// 测试是否连接成功
	pingErr := db.Ping()
	if pingErr != nil {
		log.Fatal(pingErr)
	}
	log.Println("Connected!")
}

查询数据

对于数据库中的数据,我们需要定义一个结构体去接收。在main.go中输入以下代码。

// Album 记录实体结构体
type Album struct {
	ID     int64
	Title  string
	Artist string
	Price  float32
}

接着我们实现String函数让输出稍微美观一些。

func (album Album) String() string {
	return fmt.Sprintf(`{ "id": %d, "title": %q, "artist": %q, "price": %.2f }`, album.ID, album.Title, album.Artist, album.Price)
}

接下来,让我们编写一个函数,这个函数的功能是通过人名去查询数据库中的记录。

// 通过人名查询记录
func albumsByArtist(name string) ([]Album, error) {
	var albums []Album

	rows, err := db.Query("select * from album where artist = ?", name)
	if err != nil {
		return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
	}
	// 资源释放,defer关键字修饰的语句在其下方语句未执行完成前不会执行
	defer rows.Close()

	// 循环获取相关值
	for rows.Next() {
		var alb Album
		if err := rows.Scan(&alb.ID, &alb.Title, &alb.Artist, &alb.Price); err != nil {
			return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
		}
		albums = append(albums, alb)
	}
	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
	}
	return albums, nil
}

接着定义一个函数,通过id去查询记录。

// 通过ID查询记录
func albumsById(id int64) (Album, error) {
	var alb Album

	row := db.QueryRow("select * from album where id = ?", id)
	if err := row.Scan(&alb.ID, &alb.Title, &alb.Artist, &alb.Price); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return alb, fmt.Errorf("albumsById %d: no such album", id)
		}
		return alb, fmt.Errorf("albumsById %d: %v", id, err)
	}
	return alb, nil
}

然后,我们在main函数中调用。

// 查询数据
albums, err := albumsByArtist("John Coltrane")
if err != nil {
	log.Fatal(err)
}
log.Printf("Albums found: %v\n", albums)

alb, err := albumsById(2)
if err != nil {
	log.Fatal(err)
}
log.Printf("Album found: %v\n", alb)

修改数据

我们来定义一个函数用来修改数据,用id做为删选条件。

// 修改数据
func updateAlbumById(alb Album) (int64, error) {
	result, err := db.Exec("update album set title = ?, artist = ?, price = ? where id = ?", alb.Title, alb.Artist, alb.Price, alb.ID)
	if err != nil {
		return 0, fmt.Errorf("updateAlbumById: %v", err)
	}

	rows, err := result.RowsAffected()
	if err != nil {
		return 0, fmt.Errorf("updateAlbumById: %v", err)
	}
	return rows, nil
}

在main函数中调用它。

// 修改数据
updateRows, err := updateAlbumById(Album{
	ID: 1,
	Title: "White teddy bear",
	Artist: "John Thompson's",
	Price: 20.99,
})
if err != nil {
	log.Fatal(err)
}
log.Printf("Number of rows updated: %v\n", updateRows)

插入数据

定义一个函数用来插入数据并且返回插入数据在数据库中的id。

// 插入记录
func addAlbum(alb Album) (int64, error) {
	result, err := db.Exec("insert into album (title, artist, price) values (?, ?, ?)", alb.Title, alb.Artist, alb.Price)
	if err != nil {
		return 0, fmt.Errorf("addAlbum: %v", err)
	}
	
	id, err := result.LastInsertId()
	if err != nil {
		return 0, fmt.Errorf("addAlbum: %v", err)
	}
	return id, nil
}

在main函数中调用

// 插入数据
albId, err := addAlbum(Album{
	Title: "The Modern Sound of Betty Carter",
	Artist: "Betty Carter",
	Price: 49.99,
})
if err != nil {
	log.Fatal(err)
}
log.Printf("id of added alnum: %v\n", albId)

删除数据

定义一个函数用来删除数据。

// 删除记录,返回删除的行数
func deleteAlbum(id int64) (int64, error) {
	result, err := db.Exec("delete from album where id = ?", id)
	if err != nil {
		return 0, fmt.Errorf("deleteAlbum: %v", err)
	}
	
	rows, err := result.RowsAffected()
	if err != nil {
		return 0, fmt.Errorf("deleteAlbum: %v", err)
	}
	return rows, nil
}

在main函数中调用

// 删除数据
deleteRows, err := deleteAlbum(albId)
if err != nil {
	log.Fatal(err)
}
log.Printf("Number of rows deleted: %v\n", deleteRows)

释放资源

对于数据库的操作完成后,需要释放数据库连接,在main函数中输入以下代码释放资源。

// 释放资源
err = db.Close()
if err != nil {
	log.Fatal(err)
}

完整代码

完整代码如下。

package main

import (
	"database/sql"
	"errors"
	"fmt"
	"github.com/go-sql-driver/mysql"
	"log"
)

var db *sql.DB

// Album 记录实体结构体
type Album struct {
	ID     int64
	Title  string
	Artist string
	Price  float32
}

func (album Album) String() string {
	return fmt.Sprintf(`{ "id": %d, "title": %q, "artist": %q, "price": %.2f }`, album.ID, album.Title, album.Artist, album.Price)
}

func main() {
	// 设置连接属性
	cfg := mysql.Config{
		User:   "root",
		Passwd: "123456",
		Net:    "tcp",
		Addr:   "127.0.0.1:3306",
		DBName: "recordings",
	}

	// 获取数据库句柄
	var err error
	db, err = sql.Open("mysql", cfg.FormatDSN())
	if err != nil {
		log.Fatal(err)
	}

	// 测试是否连接成功
	pingErr := db.Ping()
	if pingErr != nil {
		log.Fatal(pingErr)
	}
	log.Println("Connected!")

	// 查询数据
	albums, err := albumsByArtist("John Coltrane")
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("Albums found: %v\n", albums)

	alb, err := albumsById(2)
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("Album found: %v\n", alb)
	
	// 修改数据
	updateRows, err := updateAlbumById(Album{
		ID: 1,
		Title: "White teddy bear",
		Artist: "John Thompson's",
		Price: 20.99,
	})
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("Number of rows updated: %v\n", updateRows)
	
	// 插入数据
	albId, err := addAlbum(Album{
		Title: "The Modern Sound of Betty Carter",
		Artist: "Betty Carter",
		Price: 49.99,
	})
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("id of added alnum: %v\n", albId)
	
	// 删除数据
	deleteRows, err := deleteAlbum(albId)
	if err != nil {
		log.Fatal(err)
	}
	log.Printf("Number of rows deleted: %v\n", deleteRows)
	
	// 释放资源
	err = db.Close()
	if err != nil {
		log.Fatal(err)
	}
}

// 通过人名查询记录
func albumsByArtist(name string) ([]Album, error) {
	var albums []Album

	rows, err := db.Query("select * from album where artist = ?", name)
	if err != nil {
		return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
	}
	// 资源释放,defer关键字修饰的语句在其下方语句未执行完成前不会执行
	defer rows.Close()

	// 循环获取相关值
	for rows.Next() {
		var alb Album
		if err := rows.Scan(&alb.ID, &alb.Title, &alb.Artist, &alb.Price); err != nil {
			return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
		}
		
		albums = append(albums, alb)
	}
	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("albumsByArtist %q: %v", name, err)
	}
	return albums, nil
}

// 通过ID查询记录
func albumsById(id int64) (Album, error) {
	var alb Album

	row := db.QueryRow("select * from album where id = ?", id)
	if err := row.Scan(&alb.ID, &alb.Title, &alb.Artist, &alb.Price); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return alb, fmt.Errorf("albumsById %d: no such album", id)
		}
		return alb, fmt.Errorf("albumsById %d: %v", id, err)
	}
	return alb, nil
}

// 修改数据
func updateAlbumById(alb Album) (int64, error) {
	result, err := db.Exec("update album set title = ?, artist = ?, price = ? where id = ?", alb.Title, alb.Artist, alb.Price, alb.ID)
	if err != nil {
		return 0, fmt.Errorf("updateAlbumById: %v", err)
	}

	rows, err := result.RowsAffected()
	if err != nil {
		return 0, fmt.Errorf("updateAlbumById: %v", err)
	}
	return rows, nil
}

// 插入记录
func addAlbum(alb Album) (int64, error) {
	result, err := db.Exec("insert into album (title, artist, price) values (?, ?, ?)", alb.Title, alb.Artist, alb.Price)
	if err != nil {
		return 0, fmt.Errorf("addAlbum: %v", err)
	}
	
	id, err := result.LastInsertId()
	if err != nil {
		return 0, fmt.Errorf("addAlbum: %v", err)
	}
	return id, nil
}

// 删除记录,返回删除的行数
func deleteAlbum(id int64) (int64, error) {
	result, err := db.Exec("delete from album where id = ?", id)
	if err != nil {
		return 0, fmt.Errorf("deleteAlbum: %v", err)
	}
	
	rows, err := result.RowsAffected()
	if err != nil {
		return 0, fmt.Errorf("deleteAlbum: %v", err)
	}
	return rows, nil
}

最终执行结果

在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1103714.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

大型公共建筑能耗监测与信息管理系统研究及产品选型

摘要:文章通过阐述大型公共建筑能耗现状,突出大型公共建筑实施节能监管的必要性,并在系统总结运用技术手段实施建筑能耗监测的基础上,介绍了江苏省建筑能耗监测系统研究过程中的技术创新和应用情况。 关键词:公共建筑…

深度学习——含并行连接的网络(GoogLeNet)

深度学习——含并行连接的网络(GoogLeNet) 文章目录 前言一、Inception块二、GoogLeNet模型三、训练模型总结 前言 上篇文章中学习了NIN,而GoogLeNet吸收了NIN中串联网络的思想,并在此基础上做了改进。该论文中的一个观点是&…

PyQt5基础学习(一)

从PyQt5最基础的内容开始学习 import sysfrom PyQt5.QtGui import QIcon from PyQt5.QtWidgets import QWidget, QApplication, QTextBrowserclass Example(QWidget):def __init__(self):super().__init__()self.initUI()def initUI(self):self.resize(300, 300)self.setWind…

数据结构之单链表的模拟实现

💕"你笑的次数越多越好,因为你只有用笑才能不怀恶意地消灭罪恶。"💕 作者:Mylvzi 文章主要内容:数据结构之单链表的模拟实现 MyArrayList /*** Created with IntelliJ IDEA.* Description:* User: 绿字* …

F5.5G落进现实:目标网带来的光之路

数字化与智能化的世界将走向何方?这个问题有着非常复杂的答案,但其中有一个答案已经十分清晰。那就是智能化的下一步,必将走向泛在万兆的世界。 网络是算力联接的底座,是智能演化的基础。纵观每一代数字化升级,都可以发…

2023秋招华为技术岗线上面试经历

2023/10/16 个人情况:博士,预计2024年毕业,参加了2023秋招,华为应聘到3面主管面。 下面按招聘流程顺序,记录我的面试经历。因为想写详细一点的独立文章,所以想来想去还是放到CSDN上。 1. 宣讲会 宣讲会…

kr 第三阶段(一)16 位汇编

为什么要学习 16 位汇编? 16 位汇编包含了大部分 32 位汇编的知识点。有助于在学习内核的两种模式。 实模式:访问真实的物理内存保护模式:访问虚拟内存 有助于提升调试能力,调试命令与 OllyDbg 和 WinDebug 通用。可以学习实现反…

spring boot整合MongoDB 一

MongoDB介绍 应用场景 传统的关系型数据库(如MySQL),在数据操作的“三高”需求以及应对Web2.0的网站需求面前,显得力不从心。 解释:“三高”需求: • High performance - 对数据库高并发读写的需求。 • …

制作.a静态库 (封盒)

//云库房间 1.GitHub上创建开源框架项目须包含文件: LICENSE:开源许可证;README.md:仓库说明文件;开源项目;(登录GitHub官网) 2. 云仓储库构建成功(此时云库中没有内容三方框架)!!! 3. 4.5. //…

数仓建设(二)

1) 指标梳理 指标口径的不一致使得数据使用的成本极高,经常出现口径打架、反复核对数据的问题。在数据治理中,我们将需求梳理到的所有指标进行进一步梳理,明确其口径,如果存在两个指标名称相同,但口径不一致&#xff0…

项目管理之生命周期管理

项目生命周期管理矩阵是项目管理中一个重要的概念,它包括了项目从准备到收尾的各个阶段。项目生命周期管理矩阵以四个主要管理阶段为基础,分别为准备阶段、启动阶段、执行阶段和收尾阶段。这四个阶段在项目管理中有着明确的目标和职责,贯穿了…

【LeetCode】35. 搜索插入位置

1 问题 给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。 请必须使用时间复杂度为 O(log n) 的算法。 示例 1: 输入: nums [1,3,5,6], target 5 输出: 2 示例…

GPT4 Plugins 插件 WebPilot 生成抖音文案

1. 生成抖音文案 1.1. 准备1篇优秀的抖音文案范例 1.2. Promept公式 你是一个有1000万粉丝的抖音主播, 请模仿下面的抖音脚本文案,重新改与一篇文章改写成2分钟的抖音视频脚本, 要求前一部分是十分有争议性的内容,并且能够引发…

linux进程间通讯--信号量

1.认识信号量 方便理解:信号量就是一个计数器。当它大于0能用,小于等于0,用不了,这个值自己给。 2.特点: 信号量用于进程间同步,若要在进程间传递数据需要结合共享内存。信号量基于操作系统的 PV 操作&am…

GitHub仓库的README文件无法显示图片问题-非域名污染原因

之前上自己仓库就偶然发现图片不显示现象,当时以为是网络问题就没有留意这事。但是一直不显示就有问题了!于是网上搜了一遭,看见大家遇到此现象的原因普遍归于DNS污染1而我的问题原来是MarkDown格式! 在图片语法前不要加分区语法…

linux部署gitlab

1. 配置yum源: vim /etc/yum.repos.d/gitlab-ce.repo [gitlab-ce] nameGitlab CE Repository baseurlhttps://mirrors.tuna.tsinghua.edu.cn/gitlab-ce/yum/el$releasever/ gpgcheck0 enabled1 2. 更新本地缓存 sudo yum install -y gitlab-ce 3. 安装相关依赖 yum …

RT-Thread学习笔记(三):线程管理

线程管理 线程管理相关概念什么是时间片轮转调度器锁线程运行机制线程的五种状态 动态和静态创建线程区别动态和静态创建线程优缺点RT-Thread动态线程管理函数动态创建线程动态删除线程 RT-Thread静态线程管理函数静态创建线程 线程其他操作线程启动线程延时获得当前执行的线程…

基于nodejs+vue小型企业银行账目管理系统

目 录 摘 要 I ABSTRACT II 目 录 II 第1章 绪论 1 1.1背景及意义 1 1.2 国内外研究概况 1 1.3 研究的内容 1 第2章 相关技术 3 2.1 nodejs简介 4 2.2 express框架介绍 6 2.4 MySQL数据库 4 第3章 系统分析 5 3.1 需求分析 5 3.2 系统可行性分析 5 3.2.1技术可行性:…

九阳真经之各大厂校招

大学计算机系的同学要怎么努力才能校招进大厂? 秋招的大公司非常多,也是非常好的,赶上了秋招,你基本工作就敲定了,在整个应届毕业生的人群中你就占据很大的优势了。 如何准备应届校招? 一、做好规划,把…

Python 网络爬虫

爬虫原理 计算机一次Request请求和服务器端的Response回应,即实现了网络连接。 爬虫需要做两件事:模拟计算机对服务器发起Request请求。 接受服务器的Response内容并解析、提取所需的信息。 多页面爬虫流程 ​​​​​​​多页面网页爬虫流程