使用gofreetds连接Microsoft SQL Server数据库

gofreetds是一个用于Go语言连接Microsoft SQL Server数据库的第三方驱动库，它基于FreeTDS实现。下面我将详细介绍如何使用gofreetds连接和操作MSSQL数据库。

安装gofreetds

首先需要安装gofreetds库：

go get github.com/minus5/gofreetds

基本连接示例

package main

import (
	"fmt"
	"log"
	"github.com/minus5/gofreetds"
)

func main() {
	// 配置连接参数
	connStr := "user=sa;password=your_password;server=localhost;database=your_database"
	
	// 创建连接池
	pool, err := gofreetds.NewConnPool(connStr)
	if err != nil {
		log.Fatal(err)
	}
	defer pool.Close()

	// 从连接池获取连接
	conn, err := pool.Get()
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	// 执行查询
	rst, err := conn.Exec("select * from users where id = ?", 1)
	if err != nil {
		log.Fatal(err)
	}

	// 处理结果
	for _, row := range rst[0].Rows {
		fmt.Printf("ID: %d, Name: %s\n", row.Int("id"), row.String("name"))
	}
}

连接参数说明

连接字符串支持的参数：

• user 或 username: 数据库用户名
• password 或 pwd: 数据库密码
• server 或 host: 服务器地址
• database 或 db: 数据库名称
• port: 端口号(默认1433)
• mirror: 镜像服务器地址
• timeout: 连接超时(秒)
• encrypt: 是否加密(disable/require)
• tls: TLS配置

执行SQL查询

查询单行

func querySingleRow(conn *gofreetds.Conn) {
	rst, err := conn.Exec("select top 1 * from users where age > ?", 18)
	if err != nil {
		log.Fatal(err)
	}
	
	if len(rst) > 0 && len(rst[0].Rows) > 0 {
		row := rst[0].Rows[0]
		fmt.Printf("User: %s, Age: %d\n", row.String("name"), row.Int("age"))
	}
}

查询多行

func queryMultipleRows(conn *gofreetds.Conn) {
	rst, err := conn.Exec("select * from products where price > ?", 100)
	if err != nil {
		log.Fatal(err)
	}
	
	for _, row := range rst[0].Rows {
		fmt.Printf("Product: %s, Price: %.2f\n", 
			row.String("name"), 
			row.Float64("price"))
	}
}

执行存储过程

func callStoredProcedure(conn *gofreetds.Conn) {
	rst, err := conn.Exec("exec usp_get_employee_by_id ?", 123)
	if err != nil {
		log.Fatal(err)
	}
	
	for _, row := range rst[0].Rows {
		fmt.Printf("Employee: %s %s\n", 
			row.String("first_name"), 
			row.String("last_name"))
	}
}

事务处理

func transactionExample(conn *gofreetds.Conn) {
	// 开始事务
	tx, err := conn.Begin()
	if err != nil {
		log.Fatal(err)
	}
	
	// 确保事务回滚或提交
	defer func() {
		if err != nil {
			tx.Rollback()
			return
		}
		tx.Commit()
	}()
	
	// 执行事务中的SQL
	_, err = tx.Exec("update accounts set balance = balance - ? where id = ?", 100, 1)
	if err != nil {
		return
	}
	
	_, err = tx.Exec("update accounts set balance = balance + ? where id = ?", 100, 2)
	if err != nil {
		return
	}
	
	fmt.Println("Transaction completed successfully")
}

使用连接池

gofreetds内置了连接池功能，可以高效管理数据库连接：

func connectionPoolExample() {
	// 创建连接池
	pool, err := gofreetds.NewConnPool("user=sa;password=123;server=localhost;database=test")
	if err != nil {
		log.Fatal(err)
	}
	defer pool.Close()
	
	// 设置连接池参数
	pool.SetMaxOpenConns(10)  // 最大连接数
	pool.SetMaxIdleConns(5)   // 最大空闲连接数
	
	// 从连接池获取连接
	conn, err := pool.Get()
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()
	
	// 使用连接执行查询
	_, err = conn.Exec("select * from users")
	if err != nil {
		log.Fatal(err)
	}
}

性能注意事项

1. 尽量使用连接池而不是频繁创建新连接
2. 对于大量数据的查询，考虑使用分页
3. 复杂的查询可以转换为存储过程提高性能
4. 使用参数化查询而不是字符串拼接，防止SQL注入

错误处理

gofreetds返回的错误通常包含详细的SQL Server错误信息：

func errorHandlingExample(conn *gofreetds.Conn) {
	_, err := conn.Exec("select * from non_existent_table")
	if err != nil {
		if mssqlErr, ok := err.(*gofreetds.SqlError); ok {
			fmt.Printf("SQL Error: %d, State: %d, Message: %s\n",
				mssqlErr.Number,
				mssqlErr.State,
				mssqlErr.Message)
		} else {
			fmt.Printf("Other error: %v\n", err)
		}
	}
}

gofreetds是一个功能齐全的MSSQL驱动，适合大多数Go应用程序与SQL Server交互的需求。对于更高级的需求，也可以考虑官方支持的go-mssqldb驱动。