基于服务状态调整HTTP路由：Go REST服务兼容数据库未就绪启动

嘿，这个需求太常见了——咱们做REST服务最怕的就是启动时被外部依赖卡脖子，尤其是数据库这种核心资源。针对你提到的DAO层用goroutine+布尔通道处理连接的场景，我给你一套非常实用的落地方案：

核心思路 #

把数据库连接的就绪状态和API请求处理完全解耦：

• 服务器启动后立刻对外提供服务，不用等数据库连接成功
• 后台goroutine持续尝试连接数据库，成功后标记就绪状态
• 每个API请求进来时先检查连接状态，未就绪就返回友好的维护提示，就绪后再执行业务逻辑

具体实现步骤 #

1. DAO层优化：带重试的连接管理 #

你原来的布尔通道可以保留，但建议再加一个原子布尔变量（sync/atomic.Bool）来持久化连接状态（通道是一次性的，没法反复检查），同时给连接失败加上指数退避重试（避免疯狂轰炸数据库）：

package dao

import (
    "context"
    "sync/atomic"
    "time"

    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

var (
    mongoClient *mongo.Client
    dbReady     atomic.Bool // 原子变量，保证并发安全的状态检查
)

// InitMongoDB 启动后台连接goroutine，无需阻塞主服务启动
func InitMongoDB(mongoURI string) {
    go func() {
        retryDelay := 1 * time.Second
        maxRetryDelay := 30 * time.Second // 最大重试间隔，避免无限增长

        for {
            // 每次重试都创建新的上下文，避免超时污染
            ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
            defer cancel()

            // 尝试连接MongoDB
            client, err := mongo.Connect(ctx, options.Client().ApplyURI(mongoURI))
            if err != nil {
                goto retry
            }

            // 验证连接是否真正可用
            if err := client.Ping(ctx, nil); err != nil {
                goto retry
            }

            // 连接成功，更新状态并退出goroutine
            mongoClient = client
            dbReady.Store(true)
            return

        retry:
            // 失败后等待重试，指数退避
            time.Sleep(retryDelay)
            if retryDelay < maxRetryDelay {
                retryDelay *= 2
            }
        }
    }()
}

// GetMongoClient 返回客户端和连接状态
func GetMongoClient() (*mongo.Client, bool) {
    return mongoClient, dbReady.Load()
}

2. API层适配：请求的状态分流 #

在每个需要数据库的API handler里，先检查连接状态，未就绪时返回503 Service Unavailable状态码（符合HTTP规范），就绪后再执行业务逻辑：

package handlers

import (
    "net/http"

    "your-project/dao"
)

func GetUserHandler(w http.ResponseWriter, r *http.Request) {
    client, isReady := dao.GetMongoClient()
    if !isReady {
        w.Header().Set("Content-Type", "text/plain")
        w.WriteHeader(http.StatusServiceUnavailable)
        w.Write([]byte("Service is initializing, please try again later."))
        return
    }

    // 正常执行业务逻辑，比如查询用户
    userColl := client.Database("your_db").Collection("users")
    // ... 后续的查询、返回逻辑
}

3. 额外优化：健康检查接口 #

加一个专门的健康检查接口，方便运维监控数据库状态：

func HealthCheckHandler(w http.ResponseWriter, r *http.Request) {
    _, isReady := dao.GetMongoClient()
    if isReady {
        w.WriteHeader(http.StatusOK)
        w.Write([]byte("OK - Database connected"))
    } else {
        w.WriteHeader(http.StatusServiceUnavailable)
        w.Write([]byte("NOT OK - Database not ready"))
    }
}

补充说明 #

• 针对你提到的「服务器启动后数据库宕机」场景，只需要在DAO层加一个定期Ping的goroutine，一旦发现连接断开，就把dbReady设为false，同时触发重试逻辑——API层的代码完全不用改，会自动返回503。
• 原子变量的使用保证了并发安全，多个请求同时检查状态时不会出现竞态问题。

内容的提问来源于stack exchange，提问作者nahsh