Bài 2: Xây dựng Core Database Connection & Quản lý Connection Pool

Ở Bài 1, chúng ta đã thiết lập xong mô hình Repository Pattern giúp cô lập tầng nghiệp vụ khỏi câu lệnh SQL chi tiết. Tuy nhiên, trước khi viết các hàm truy vấn, chúng ta phải giải quyết bài toán hạ tầng nền móng: Làm thế nào để kết nối tới cơ sở dữ liệu và quản lý Connection Pool (Hồ bơi kết nối) sao cho tối ưu, không bị rò rỉ tài nguyên khi ứng dụng chịu tải cao?

Trong bài học này, chúng ta sẽ mổ xẻ cơ chế quản lý kết nối của Go và xây dựng một Core Database Connection chuẩn Production.

1. Bản chất của Connection Pool trong Golang (database/sql)

Nhiều lập trình viên mới thường lầm tưởng rằng mỗi câu lệnh truy vấn SQL sẽ mở một kết nối mới tới cơ sở dữ liệu rồi đóng lại ngay lập tức. Trên thực tế, việc mở và đóng kết nối TCP liên tục tiêu tốn rất nhiều tài nguyên CPU và thời gian (TCP Handshake, Authentication).

Standard library database/sql của Go đã tích hợp sẵn một Connection Pool quản lý tự động gồm ba trạng thái kết nối:

• Active (In-use): Các kết nối đang bận thực hiện câu lệnh SQL cho request.

• Idle: Các kết nối đang rảnh rỗi, được giữ sẵn trong pool để sẵn sàng phục vụ request tiếp theo ngay lập tức mà không cần khởi tạo mới.

• Closed: Các kết nối đã bị đóng (do hết hạn hoặc vượt quá giới hạn).

2. Các tham số cấu hình sống còn của Connection Pool

Nếu bạn không cấu hình tường minh, Go sẽ sử dụng các giá trị mặc định (ví dụ: MaxOpenConns mặc định là không giới hạn). Trên môi trường Production, điều này cực kỳ nguy hiểm vì có thể dẫn đến quá tải database (too many connections).

Bạn bắt buộc phải cấu hình 4 tham số vàng sau đây:

Tham số cấu hình

Ý nghĩa & Quy tắc cấu hình

SetMaxOpenConns(n)

Số lượng kết nối tối đa được phép mở đồng thời tới database. Cần được tính toán dựa trên cấu hình phần cứng của database server (ví dụ: max_connections của PostgreSQL).

SetMaxIdleConns(n)

Số lượng kết nối tối đa được giữ ở trạng thái rảnh rỗi (Idle) trong pool. Nên đặt giá trị này nhỏ hơn hoặc bằng MaxOpenConns để tránh lãng phí tài nguyên.

SetConnMaxLifetime(d)

Thời gian sống tối đa của một kết nối. Sau khoảng thời gian này, kết nối sẽ bị đóng và tái tạo mới hoàn toàn. Giúp giải phóng các kết nối bị kẹt hoặc stale connections.

SetConnMaxIdleTime(d)

Thời gian tối đa một kết nối được phép ở trạng thái rảnh rỗi trước khi bị đóng bớt để tiết kiệm tài nguyên khi hệ thống ít traffic.

3. Xây dựng Core Database Connection chuẩn Production

Dưới đây là đoạn mã hiện thực hóa việc khởi tạo kết nối PostgreSQL sử dụng database/sql kết hợp cấu hình Connection Pool hoàn chỉnh:

Tạo file pkg/database/postgres.go:

Go

package database

import (
	"context"
	"database/sql"
	"fmt"
	"time"

	_ "github.com/lib/pq" // Driver PostgreSQL
)

// Config chứa các tham số cấu hình kết nối database
type Config struct {
	Host            string
	Port            string
	User            string
	Password        string
	DBName          string
	SSLMode         string
	MaxOpenConns    int
	MaxIdleConns    int
	ConnMaxLifetime time.Duration
	ConnMaxIdleTime time.Duration
}

// NewPostgresDB khởi tạo kết nối cơ sở dữ liệu và cấu hình Connection Pool
func NewPostgresDB(cfg Config) (*sql.DB, error) {
	dsn := fmt.Sprintf(
		"host=%s port=%s user=%s password=%s dbname=%s sslmode=%s",
		cfg.Host, cfg.Port, cfg.User, cfg.Password, cfg.DBName, cfg.SSLMode,
	)

	// 1. Mở kết nối cơ bản (Lưu ý: sql.Open chưa thực sự kết nối xuống DB ngay lập tức)
	db, err := sql.Open("postgres", dsn)
	if err != nil {
		return nil, fmt.Errorf("lỗi khởi tạo driver sql: %w", err)
	}

	// 2. Cấu hình Connection Pool
	db.SetMaxOpenConns(cfg.MaxOpenConns)
	db.SetMaxIdleConns(cfg.MaxIdleConns)
	db.SetConnMaxLifetime(cfg.ConnMaxLifetime)
	db.SetConnMaxIdleTime(cfg.ConnMaxIdleTime)

	// 3. Kiểm tra kết nối thực tế bằng lệnh Ping (kèm timeout để tránh treo app)
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	if err := db.PingContext(ctx); err != nil {
		db.Close() // Đóng kết nối nếu ping thất bại tránh rò rỉ file descriptor
		return nil, fmt.Errorf("không thể kết nối tới cơ sở dữ liệu (Ping failed): %w", err)
	}

	return db, nil
}

4. Tích hợp vào vòng đời ứng dụng (Graceful Shutdown)

Khi ứng dụng của bạn nhận tín hiệu tắt (ví dụ nhấn Ctrl+C hoặc Kubernetes gửi lệnh termination), bạn phải đảm bảo gọi db.Close() để đóng toàn bộ Connection Pool một cách an toàn, tránh để lại các kết nối "treo" (orphaned connections) phía cơ sở dữ liệu:

Go

func main() {
	// Khởi tạo DB config...
	db, err := database.NewPostgresDB(cfg)
	if err != nil {
		log.Fatalf("Khởi tạo database thất bại: %v", err)
	}
	
	// Đảm bảo đóng Connection Pool khi ứng dụng kết thúc
	defer func() {
		if err := db.Close(); err != nil {
			log.Printf("Lỗi khi đóng database connection pool: %v", err)
		} else {
			log.Println("Đã đóng database connection pool thành công.")
		}
	}()

	// Chạy HTTP Server...
}

Tổng kết

Bằng cách thiết lập tường minh các tham số Connection Pool (MaxOpenConns, MaxIdleConns, ConnMaxLifetime) và thực hiện kiểm tra PingContext, chúng ta đã xây dựng được một nền móng kết nối cơ sở dữ liệu vững chắc, an toàn và sẵn sàng đối phó với tải cao trên môi trường Production.

Tuy nhiên, trong các bài toán thực tế, một thao tác nghiệp vụ hiếm khi chỉ đọc hoặc ghi một bảng đơn lẻ. Nó thường đòi hỏi thực hiện chuỗi nhiều câu lệnh liên tiếp; nếu một câu lệnh lỗi, toàn bộ phải được hoàn tác.

Làm thế nào để quản lý Transaction và truyền tải chúng xuyên suốt qua nhiều tầng Repository mà không làm bẩn mã nguồn?

Đó chính là nội dung của Bài 3 tiếp theo.