66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
|
package repository
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
|
|||
|
|
"github.com/testcontainers/testcontainers-go"
|
|||
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
MySQLUser = "root"
|
|||
|
|
MySQLPassword = "password"
|
|||
|
|
MySQLDatabase = "testdb"
|
|||
|
|
MySQLPort = "3306"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 啟動 MySQL container
|
|||
|
|
func startMySQLContainer() (host string, port string, dsn string, tearDown func(), err error) {
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
req := testcontainers.ContainerRequest{
|
|||
|
|
Image: "mysql:8.0",
|
|||
|
|
Env: map[string]string{
|
|||
|
|
"MYSQL_ROOT_PASSWORD": MySQLPassword,
|
|||
|
|
"MYSQL_DATABASE": MySQLDatabase,
|
|||
|
|
},
|
|||
|
|
ExposedPorts: []string{"3306/tcp"},
|
|||
|
|
WaitingFor: wait.ForListeningPort("3306/tcp"),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mysqlC, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
|
|||
|
|
ContainerRequest: req,
|
|||
|
|
Started: true,
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
return "", "", "", nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mappedPort, err := mysqlC.MappedPort(ctx, "3306")
|
|||
|
|
if err != nil {
|
|||
|
|
return "", "", "", nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
containerHost, err := mysqlC.Host(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", "", "", nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 組成 DSN（Data Source Name）
|
|||
|
|
dsn = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true",
|
|||
|
|
MySQLUser,
|
|||
|
|
MySQLPassword,
|
|||
|
|
containerHost,
|
|||
|
|
mappedPort.Port(),
|
|||
|
|
MySQLDatabase,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
tearDown = func() {
|
|||
|
|
_ = mysqlC.Terminate(ctx)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
fmt.Printf("MySQL ready at: %s\n", dsn)
|
|||
|
|
return containerHost, mappedPort.Port(), dsn, tearDown, nil
|
|||
|
|
}
|