Golang微服务健康检查实现方法

package main

import (
    "encoding/json"
    "net/http"
    "time"
)

type HealthResponse struct {
    Status    string    `json:"status"`
    Timestamp time.Time `json:"timestamp"`
    Version   string    `json:"version"`
}

func healthHandler(w http.ResponseWriter, r *http.Request) {
    // 可在此加入数据库连接、缓存等依赖检查
    w.Header().Set("Content-Type", "application/json")
    resp := HealthResponse{
        Status:    "healthy",
        Timestamp: time.Now(),
        Version:   "v1.0.0",
    }
    json.NewEncoder(w).Encode(resp)
}

func main() {
    http.HandleFunc("/health", healthHandler)
    http.ListenAndServe(":8080", nil)
}

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

func checkDB(db *sql.DB) bool {
    if err := db.Ping(); err != nil {
        return false
    }
    return true
}

func healthHandler(w http.ResponseWriter, r *http.Request) {
    dbOK := checkDB(db)
    if !dbOK {
        w.WriteHeader(http.StatusServiceUnavailable)
        json.NewEncoder(w).Encode(map[string]string{"status": "unhealthy", "component": "database"})
        return
    }

    w.WriteHeader(http.StatusOK)
    json.NewEncoder(w).Encode(map[string]string{"status": "healthy"})
}

import "github.com/hashicorp/consul/api"

func registerService() error {
    config := api.DefaultConfig()
    config.Address = "127.0.0.1:8500"
    client, _ := api.NewClient(config)

    registration := &api.AgentServiceRegistration{
        ID:      "user-service-1",
        Name:    "user-service",
        Address: "127.0.0.1",
        Port:    8080,
        Check: &api.AgentServiceCheck{
            HTTP:                           "http://127.0.0.1:8080/health",
            Timeout:                        "5s",
            Interval:                       "10s",
            DeregisterCriticalServiceAfter: "30s",
        },
    }

    return client.Agent().ServiceRegister(registration)
}