Golang实现的URL重定向服务MySQL连接池未正确释放的问题咨询

首先澄清误解：你之前担心的“QueryRow查不到匹配结果就不释放连接”是错误的。Go标准库的QueryRow在调用Scan读取时，会自动遍历内部关联的结果集，哪怕返回sql.ErrNoRows，也会主动把连接归还到池里，连接泄漏和QueryRow的使用方式本身无关。

代码中存在的资源泄漏点 #

• 请求循环内重复预编译SQL
  每一次HTTP请求都调用db.Prepare()生成新的预编译语句。Go的database/sql包中，预编译语句是绑定到底层物理数据库连接的，每次Prepare操作都会从连接池申请连接完成预编译握手。哪怕后续调用了Stmt.Close()，高并发下的连接申请速率会远高于释放速率，快速打满数据库连接上限。
• 错误路径下的资源未释放
  代码中只要碰到数据库操作错误就直接调用log.Panic()终止当前协程，后续写在错误处理逻辑后的Stmt.Close()根本不会被执行，对应的连接会被永久占用无法归还连接池。
• 用Query()执行非结果集操作
  插入重定向日志时，代码使用insertStmt.Query(id)执行INSERT语句。Query()方法是为返回结果集的查询设计的，只要调用就会从连接池取连接、关联结果集指针，哪怕返回的*sql.Rows实例被直接丢弃，连接也不会自动释放，必须显式调用rows.Close()读完所有结果才会归还。
• 连接池配置缺失
  代码仅设置了最大打开连接数，没有配置最大空闲连接数、连接最大存活时间。MySQL服务端会默认关闭空闲超时的连接，但应用端连接池无法感知，下次申请连接时会新建物理连接，长期运行后残留大量无效连接。

正确的实现方式 #

1. 启动阶段一次性初始化全局预编译语句 #

把所有需要重复执行的SQL预编译过程放在程序启动逻辑里，不要在请求处理循环中做预编译操作，从根源减少不必要的连接申请。

2. 资源释放用defer兜底 #

所有需要手动关闭的资源（比如预编译语句、结果集），在拿到实例后第一时间用defer声明关闭逻辑，哪怕后续流程出现panic，也能保证资源被正确释放。

3. 非结果集操作统一用Exec() #

INSERT/UPDATE/DELETE这类不返回查询结果的操作，用Stmt.Exec()执行，该方法执行完成后会自动归还连接，不需要手动处理结果集。

4. 补全连接池基础配置 #

除了最大打开连接数，配套设置最大空闲连接数、连接最大存活时间，在程序启动时调用db.Ping()确认连接可用性，避免拿取已经被数据库端断开的无效连接。

核心代码修正示例 #

import "time"

// 全局预编译语句，启动时初始化
var (
	queryRewriteStmt *sql.Stmt
	insertLogStmt    *sql.Stmt
)

func main() {
	// ... 原有加载环境变量、初始化DB的逻辑

	// 启动时预编译所有SQL
	var err error
	queryRewriteStmt, err = db.Prepare("SELECT id, newPath FROM url_rewrites where oldPath = ?")
	if err != nil {
		log.Fatal("初始化URL查询预编译失败:", err)
	}
	insertLogStmt, err = db.Prepare("INSERT INTO redirect_logs (url_rewrite_id) values (?)")
	if err != nil {
		log.Fatal("初始化日志插入预编译失败:", err)
	}

	// ... 原有路由、服务启动逻辑
}

func handleMapper(w http.ResponseWriter, r *http.Request) {
	urlPath := strings.TrimPrefix(r.URL.Path, "/")

	var id int
	var newPath string
	// 直接使用全局预编译语句查询
	err := queryRewriteStmt.QueryRow(urlPath).Scan(&id, &newPath)
	if err == sql.ErrNoRows {
		// 未匹配重定向规则，直接转发
		forwardRequest(w, r)
		return
	}
	if err != nil {
		log.Printf("查询URL映射失败: %v", err)
		http.Error(w, "内部服务错误", http.StatusInternalServerError)
		return
	}

	// 插入重定向日志，用Exec避免结果集漏连接
	_, err = insertLogStmt.Exec(id)
	if err != nil {
		log.Printf("写入重定向日志失败: %v", err)
	}

	// 301跳转
	targetURL := os.Getenv("NEWDOMAIN") + newPath
	http.Redirect(w, r, targetURL, http.StatusMovedPermanently)
}

func dbConn() *sql.DB {
	// ... 原有连接串拼接逻辑
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		log.Panic(err)
	}
	// 补全连接池配置
	db.SetMaxOpenConns(100)
	db.SetMaxIdleConns(20)
	db.SetConnMaxLifetime(10 * time.Minute) // 配置值比MySQL默认wait_timeout短
	// 启动时校验连接可用性
	if err = db.Ping(); err != nil {
		log.Panic("数据库连接不可用:", err)
	}
	return db
}

额外优化点：反向代理实例不需要每次请求新建，全局初始化一次httputil.NewSingleHostReverseProxy()实例即可复用，减少不必要的对象创建开销。

内容的提问来源于stack exchange，提问作者develobster