如何使用go with docker等待postgres db？

version: '3'
services:
  db:
    image: postgres
    restart: always
    environment:
      POSTGRES_USER: your_username
      POSTGRES_PASSWORD: your_password
      POSTGRES_DB: your_database_name
    ports:
      - "5432:5432"

package main

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

    _ "github.com/lib/pq"
)

func main() {
    // 连接数据库
    db, err := sql.Open("postgres", "postgres://your_username:your_password@localhost/your_database_name?sslmode=disable")
    if err != nil {
        log.Fatal(err)
    }

    // 等待数据库启动
    for {
        err = db.Ping()
        if err == nil {
            break
        }
        fmt.Println("Waiting for database to start...")
        time.Sleep(time.Second)
    }

    fmt.Println("Database started successfully!")
}

docker-compose up -d

go run main.go