实现节点下线功能,优化控制通道和数据通道的连接处理
This commit is contained in:
@@ -43,27 +43,43 @@ func Start() error {
|
||||
return fmt.Errorf("注册节点失败: %w", err)
|
||||
}
|
||||
|
||||
// 建立控制通道
|
||||
// 连接到网关
|
||||
var ctx, cancel = signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
|
||||
defer cancel()
|
||||
|
||||
var errCh = make(chan error)
|
||||
go func() {
|
||||
for {
|
||||
err = ctrl(ctx, id, host)
|
||||
if err == nil {
|
||||
errCh <- nil
|
||||
return
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
default:
|
||||
slog.Error("建立控制通道失败", "err", err)
|
||||
slog.Info(fmt.Sprintf("%d 秒后重试", core.RetryInterval))
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-time.After(time.Duration(core.RetryInterval) * time.Second):
|
||||
}
|
||||
slog.Error("建立控制通道失败", "err", err)
|
||||
slog.Info(fmt.Sprintf("%d 秒后重试", core.RetryInterval))
|
||||
time.Sleep(time.Duration(core.RetryInterval) * time.Second)
|
||||
}
|
||||
}()
|
||||
|
||||
// 等待退出
|
||||
select {
|
||||
case err := <-errCh:
|
||||
if err != nil {
|
||||
slog.Error("控制通道发生错误", "err", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 下线节点
|
||||
slog.Debug("下线节点...")
|
||||
err = report.Offline()
|
||||
|
||||
@@ -69,5 +69,27 @@ func Online(prov, city, isp string) (id int32, host string, err error) {
|
||||
}
|
||||
|
||||
func Offline() error {
|
||||
var bytes, err = json.Marshal(map[string]any{
|
||||
"name": env.Name,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var body = strings.NewReader(string(bytes))
|
||||
|
||||
req, err := http.NewRequest("POST", env.EndpointOffline, body)
|
||||
if err != nil {
|
||||
return fmt.Errorf("创建请求失败: %w", err)
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("执行请求失败: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return errors.New("状态码: " + resp.Status)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user