百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

golang系列:database/sql 访问数据库

nanshan 2024-11-02 12:12 14 浏览 0 评论

Golang database/sql是一套统一的访问数据库的抽象接口,真正与数据库打交道的是各个数据库对应的驱动实现,在使用时需要先注册对应驱动,然后就可以使用sql中定义的接口统一操作数据库。

连接池sql.DB

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql"
)
func main() {
	db, err := sql.Open("mysql",
		"user:password@tcp(127.0.0.1:3306)/hello")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()
        err = db.Ping()
        if err != nil {
	     // do something here
        }
}

sql.DB刚开始建立时是懒加载的,不会自动创建新的连接,只有使用Ping()或者运行查询时才会自动生成一个新的连接然后去连接数据库,只有这个时候才能确定数据库是否真的OK,所以建议一定要在sql.Open后运行Ping()确定数据连接正常运行

sql.DB是连接后初始化的一个连接池,通常全局就初始化这一个连接池,并且长期运行,所有后续数据库操作都使用该连接池进行。sql.DB内部自动维护连接池,当需要连接时自动选择一个空闲的连接,如果没有空闲就建立一个新的连接,当连接不再使用时放回连接池中,内部会自动管理空闲回收。

数据库的连接是一个比较大的耗时和资源消耗操作,首选需要经典的TCP三次握手,tcp连接后数据库需要分配连接资源,同时根据连接信息鉴权等,所以建议使用长连接。对应到我们的go中,sql.DB会自动管理连接池,最好全局使用一个连接池,不要重复的open或者close。

查询

sql.DB支持4种查询:

db.Query()    
db.QueryRow()
db.Prepare(sql)   stmt.Query(args)
db.Exec()
  • db.Query() 返回多行数据,需要依次遍历,并且需要自己关闭查询结果集
  • db.QueryRow() 是专门查询一行数据的一个语法糖,返回ErrNoRow或者一行数据,不需要自己关闭结果集
  • db.Prepare() 是预先将一个数据库连接(con)和一个条sql语句绑定并返回stmt结构体代表这个绑定后的连接,然后运行stmt.Query()或者stmt.QueryRow();stmt是并发安全的。之所以这样设计,是因为每次直接调用db.Prepare都会自动选择一个可用的con,每次选择的可能不是同一个con
  • db.Exec() 适用于执行insert、update、delete等不需要返回结果集的操作

结果集

只有db.Query()返回结果集

var (
	id int
	name string
)
rows, err := db.Query("select id, name from users where id = ?", 1)
if err != nil {
	log.Fatal(err)
}
defer rows.Close()
for rows.Next() {
	err := rows.Scan(&id, &name)
	if err != nil {
		log.Fatal(err)
	}
	log.Println(id, name)
}
err = rows.Err()
if err != nil {
	log.Fatal(err)
}
  • 首先检查结果集返回时是否有error
  1. 如果有error,直接操作error
  2. 如果无error,defer rows.Close(),确保结果集最终关闭,这样才能释放连接回连接池,Close是可以重复调用的,关闭已经关闭的结果集不会报错
  • 遍历结果集
  1. 使用for rows.Next()遍历结果集,这样迭代一行一行处理结果,节约内存分配,同时防止出现OOM的问题
  2. 使用rows.Scan将一行数据填入指定的变量中,scan会自动根据目标变量的类型处理类型转换的问题,比如数据库中是varchar,但目标变量是int,那么scan会自动转换,当然如果转化出现error会返回error

事务

tx := db.Begin()
tx.Commit()
tx.Rollback()

事务是使用db.begin开始,以db.commit/db.rollback结束

普通的db.Query/db.QueryRow自动从连接池中选择一个可用连接,运行结束后会自动将连接放回连接池,下次运行再次重复这个过程

db.begin会自动从连接池中选择一个连接并返回一直持有该连接的tx(和db.Prepare有点像),后续所有事务操作都用tx,这样能保证是在用一个连接内运行事务,只有commit/rollback才会释放连接

Error处理

  • 结果集遍历后error,每次for rows.Next结束后要跟一个rows.Err()检测
for rows.Next() {
	// ...
}
if err = rows.Err(); err != nil {
	// handle the error here
}
  • 结果集遍历close error
if err = rows.Close(); err != nil {
	// but what should we do if there's an error?
	log.Println(err)
}
  • QueryRow() Error
err = db.QueryRow("select name from users where id = ?", 1).Scan(&name)
if err != nil {
	log.Fatal(err)
}

QueryRow的结果是在Scan时才会出现

  • Mysql特定Error
if driverErr, ok := err.(*mysql.MySQLError); ok { // Now the error number is accessible directly
	if driverErr.Number == 1045 {
		// Handle the permission-denied error
	}
}
  • 连接没释放问题
  1. 事务没有commit或者rollback
  2. 查询集没有close
  • 查询参数问题
  1. Mysql使用?做参数,防止sql注入
  2. 既然是参数,就只能当参数,不可以用于其他部分,也不能做插值

标准库sql不支持但常用的特性

  • 不支持多条sql执行
  • 不支持返回多个结果集
  • 不支持存储过程(Mysql驱动目前不支持)
  • 不支持Scan到map、struct
  • 不建议uint64

第三方扩展库github.com/jmoiron/sqlx

简化连接

db = sqlx.MustConnect("sqlite3", ":memory:")  // sql.Open+db.Ping

便捷功能

  • StructScan 、SliceScan、MapScan
  • Get、Select
  • MustExec
  • 支持命名参数
  • 支持IN查询自动展开参数

Get、Select类似于QueryRow和Query,但Get和Select可以自动将结果Scan到特定数据

Struct字段必须能导出,且根据db tag(如果没有就用小写字段名)进行匹配

相关推荐

CentOS 7 搭建 Harbor2.4.1 Docker镜像仓库

上一篇文章我们使用了registry镜像来搭建Docker私有镜像仓库,但是使用体验不是很好,没有一个可管理的UI界面,管理很麻烦。本篇文章将介绍一个新的搭建Docker镜像仓库的工具叫做Har...

简单认识认识mqtt及mosquitto(mqtt报文解析)

某项目中使用了MQTT作为进程间的通信方式,之前没用过,这两篇笔记我们就来一起学习一下这种方式。MQTT的一些介绍以下介绍内容来自《[野火]《LwIP应用开发实战指南》MQTT协议全称是Messa...

全源码打造高性能 LNMP 架构: 实战教程(2025最新版)

适用场景:企业生产环境、自建Web服务、深度性能调优操作系统:CentOS7/8、RockyLinux、Debian、Ubuntu(本文以CentOSStream9为例)技术栈:N...

Nacos3.0重磅来袭!全面拥抱AI,单机及集群模式安装详细教程!

之前和大家分享过JDK17的多版本管理及详细安装过程,然后在项目升级完jdk17后又发现之前的注册和配置中心nacos又用不了,原因是之前的nacos1.3版本的,版本太老了,已经无法适配当前新的JD...

Ubuntu24.04.2 企业级MinIO存储系统部署指南

一、概要1.1MinIO架构解析MinIO是一款高性能的云原生对象存储系统,采用Golang开发并遵循ApacheLicensev2.0协议。其核心架构基于纠删码(ErasureCode)技...

从零打造自己的 国产鸿蒙(OpenHarmony)定制系统-完整可落地流程

适用版本:OpenHarmony4.0/5.0Standard目标人群:想在x86PC、RK3568开发板或自有硬件上裁剪、加品牌、预装应用并生成可刷机镜像的开发者/团队目录环境准...

一次暂未成功的dify安装经历(dify怎么安装)

前几天在阿里云买了一台机,这几天一直在尝试安装dify,到现在还没安装上我是按这个教程装的https://blog.csdn.net/2401_82469710/article/details/14...

ZLMediaKit教程(五)支持webrtc(webrtc lib)

ZLMediaKit系列文章(共六篇):ZLMediaKit流媒体(一)编译安装ZLMediaKit教程(二)主程序和配置文件解析ZLMediaKit教程(三)URL规则ZLMediaKit教程...

Linux程序安装与管理指南(linux程序安装命令大全)

在Linux系统中,安装和管理程序主要通过包管理器和手动编译安装两种主要方式实现。以下是详细的操作指南,涵盖常见发行版(如Ubuntu/Debian、CentOS/RHEL、Fedora等)的用法。一...

离线状态下安装 Nginx 各个模块?这篇攻略让你轻松搞定

你是不是也在为离线状态下安装Nginx各个模块而发愁?在互联网大厂后端开发工作中,我们常常会遇到一些特殊的网络环境,比如公司内部的离线服务器,或是处于隔离状态的测试环境。当需要在这些离线环境中安装...

Rust实践:Win10环境下的openssl交叉编译

Rust支持跨平台,可以指定生成目标平台,交叉编译也是支持的。当然,想要交叉编译成功,还需要指定平台的编译器(如:msvc、gcc等)。openssl是C语言开发的库,如果在Rust代码中用到open...

Linux下Blackwell架构显卡(RTX5070/5090)编译PaddlePaddle指南

Blackwell显卡架构如RTX5070\5090等显卡当前Paddle预编译版本中包含的GPU架构(即SM架构)是有限的,比如常见的SM75(T4)、SM86(A10)、SM89(...

突破操作系统界限,掌握Linux的必备指南

#头条创作挑战赛#简介Linux是一种开源的操作系统,它的核心思想是自由和开放。Linux以其稳定性、可靠性和安全性而闻名,被广泛用于服务器和嵌入式设备中。Linux创始人Linux安装在安装Linu...

Linux日常高频使用的100条命令,强烈建议收藏

查看系统信息如何查看系统版本:uname-alsb_release-acat/etc/os-release如何查看系统内核信息:uname-r如何查看系统CPU信息:lscpucat...

Linux文件系统结构全解析(linux文件结构详解)

对Linux新手而言,“一切皆文件”的设计哲学常让人既兴奋又困惑——打开终端输入ls/,看到的bin、etc、var等目录到底有什么用?如何快速定位关键文件?本文将从Linux文件系统的底层逻...

取消回复欢迎 发表评论: