mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-28 06:51:34 +01:00
3735add87c
The log stream displayed in the dashboard was stopping to work when a node was rebooted. Rework the log data source to establish a per-node connection and use a retry loop to always reconnect until the dashboard is terminated. Print the connection errors in the log stream in red color. Closes siderolabs/talos#8388. Signed-off-by: Utku Ozdemir <utku.ozdemir@siderolabs.com>
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package util provides utility functions for the dashboard.
|
|
package util
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc/metadata"
|
|
|
|
"github.com/siderolabs/talos/pkg/machinery/client"
|
|
)
|
|
|
|
// NodeContext contains the context.Context for a single node and the node name.
|
|
type NodeContext struct {
|
|
Ctx context.Context //nolint:containedctx
|
|
Node string
|
|
}
|
|
|
|
// NodeContexts returns a list of NodeContexts from the given context.
|
|
//
|
|
// It extracts the node names from the outgoing GRPC context metadata.
|
|
// If the node name is not present in the metadata, context will be returned as-is with an empty node name.
|
|
func NodeContexts(ctx context.Context) []NodeContext {
|
|
md, mdOk := metadata.FromOutgoingContext(ctx)
|
|
if !mdOk {
|
|
return []NodeContext{{Ctx: ctx}}
|
|
}
|
|
|
|
nodeVal := md.Get("node")
|
|
if len(nodeVal) > 0 {
|
|
return []NodeContext{{Ctx: ctx, Node: nodeVal[0]}}
|
|
}
|
|
|
|
nodesVal := md.Get("nodes")
|
|
if len(nodesVal) == 0 {
|
|
return []NodeContext{{Ctx: ctx}}
|
|
}
|
|
|
|
nodeContexts := make([]NodeContext, 0, len(nodesVal))
|
|
|
|
for _, node := range nodesVal {
|
|
nodeContexts = append(nodeContexts, NodeContext{Ctx: client.WithNode(ctx, node), Node: node})
|
|
}
|
|
|
|
return nodeContexts
|
|
}
|