Testcontainers-go 项目中使用 Docker Compose 连接 MySQL 的实践指南

在使用 Testcontainers-go 进行集成测试时，很多开发者会遇到如何正确连接 Docker Compose 中 MySQL 服务的问题。本文将详细介绍解决方案和最佳实践。

问题现象

开发者在使用 Testcontainers-go 的 Docker Compose 模块时，虽然成功启动了 MySQL 容器，但在尝试通过 GORM 连接时遇到了"driver: bad connection"错误。容器日志显示 MySQL 服务已健康，但连接仍然失败。

核心问题分析

经过排查，发现问题的根源在于连接字符串的构造方式不正确。开发者只获取了容器的映射端口，但没有正确获取容器的主机地址。在 Docker 环境中，特别是使用 Testcontainers 时，不能简单地假设服务运行在 localhost 上。

解决方案

正确的做法是同时获取容器的主机地址和映射端口：

// 获取MySQL容器的映射端口
mysqlPort, err := mysqlContainer.MappedPort(ctx, "3306")
require.NoError(t, err)

// 获取MySQL容器的主机地址
mysqlHost, err := mysqlContainer.Host(ctx)
require.NoError(t, err)

// 构造正确的DSN连接字符串
dsn := fmt.Sprintf("root:12345678@tcp(%s:%s)/test?charset=utf8mb4&parseTime=True&loc=Local", 
    mysqlHost, mysqlPort.Port())

技术要点解析

1. Host()方法：Testcontainers 提供的 Host() 方法能够返回容器在宿主机上的访问地址，这个地址会根据运行环境(Docker Desktop、Linux Docker等)自动适配。

2. MappedPort()方法：获取容器端口在宿主机上的映射端口，特别是在使用随机端口映射时尤为重要。

3. 连接字符串构造：必须使用从容器获取的实际主机地址和端口，而不是硬编码的localhost。

最佳实践建议

1. 总是检查容器服务的健康状态，确保服务完全启动后再尝试连接。

2. 考虑添加连接重试逻辑，以应对服务启动延迟的情况。

3. 在测试结束后，确保正确清理容器资源，避免资源泄漏。

4. 对于数据库测试，可以使用专门的测试数据库模块，它提供了更多针对数据库测试的便利方法。

总结

通过这个案例我们可以看到，在使用 Testcontainers-go 进行容器化测试时，理解容器网络和连接方式至关重要。正确获取容器的主机地址和端口是成功连接服务的关键。Testcontainers 提供了简洁的API来获取这些信息，开发者应该充分利用这些API而不是做硬编码假设。