Dockerfile
This commit is contained in:
@@ -5,6 +5,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
|
"net/http"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -21,6 +22,7 @@ var (
|
|||||||
reconnectMutex sync.RWMutex
|
reconnectMutex sync.RWMutex
|
||||||
reconnectDone chan struct{}
|
reconnectDone chan struct{}
|
||||||
connected bool
|
connected bool
|
||||||
|
httpClient *http.Client
|
||||||
)
|
)
|
||||||
|
|
||||||
// connectConsul 连接 Consul
|
// connectConsul 连接 Consul
|
||||||
@@ -45,9 +47,6 @@ func connectConsul(ctx context.Context) error {
|
|||||||
connected = true
|
connected = true
|
||||||
g.Log().Infof(ctx, "✅ Consul 初始化成功: %s", consulAddr)
|
g.Log().Infof(ctx, "✅ Consul 初始化成功: %s", consulAddr)
|
||||||
|
|
||||||
// 启动健康检查和自动重连(只启动一次)
|
|
||||||
go startHealthCheckAndReconnect()
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -58,11 +57,16 @@ func startHealthCheckAndReconnect() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
reconnectDone = make(chan struct{})
|
reconnectDone = make(chan struct{})
|
||||||
ticker := time.NewTicker(5 * time.Second)
|
ticker := time.NewTicker(30 * time.Second)
|
||||||
defer ticker.Stop()
|
defer ticker.Stop()
|
||||||
|
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
|
|
||||||
|
// 初始化HTTP客户端用于健康检查
|
||||||
|
httpClient = &http.Client{
|
||||||
|
Timeout: 5 * time.Second,
|
||||||
|
}
|
||||||
|
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-ticker.C:
|
case <-ticker.C:
|
||||||
@@ -80,7 +84,7 @@ func startHealthCheckAndReconnect() {
|
|||||||
reconnectMutex.Unlock()
|
reconnectMutex.Unlock()
|
||||||
|
|
||||||
if err := connectConsul(ctx); err != nil {
|
if err := connectConsul(ctx); err != nil {
|
||||||
g.Log().Errorf(ctx, "❌ Consul 重连失败: %v,5秒后重试...", err)
|
g.Log().Errorf(ctx, "❌ Consul 重连失败: %v,30秒后重试...", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
case <-reconnectDone:
|
case <-reconnectDone:
|
||||||
@@ -99,7 +103,22 @@ func checkConsulHealth(ctx context.Context) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// 简单检查注册器是否可用,避免频繁调用Search导致错误
|
// 使用consul原生API进行健康检查
|
||||||
|
// 调用 /v1/agent/self 接口检测连接状态
|
||||||
|
url := fmt.Sprintf("http://%s/v1/agent/self", consulAddr)
|
||||||
|
resp, err := httpClient.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
g.Log().Debugf(ctx, "Consul 健康检查失败: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
g.Log().Debugf(ctx, "Consul 健康检查失败,状态码: %d", resp.StatusCode)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
g.Log().Debugf(ctx, "✅ Consul 健康检查通过")
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -109,8 +128,12 @@ func init() {
|
|||||||
g.Log().Warning(context.Background(), "⚠️ Consul 配置未找到,跳过初始化")
|
g.Log().Warning(context.Background(), "⚠️ Consul 配置未找到,跳过初始化")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := connectConsul(context.Background()); err != nil {
|
if err := connectConsul(context.Background()); err != nil {
|
||||||
g.Log().Errorf(context.Background(), "❌ Consul 初始化失败: %v", err)
|
g.Log().Errorf(context.Background(), "❌ Consul 初始化失败: %v", err)
|
||||||
|
} else {
|
||||||
|
// 连接成功后启动健康检查和自动重连
|
||||||
|
go startHealthCheckAndReconnect()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
func getLocalIP() (string, error) {
|
func getLocalIP() (string, error) {
|
||||||
|
|||||||
Reference in New Issue
Block a user