prepare engine for node if calling info is needed (#634)

projecteru2 · Mar 7, 2024 · ab0c418 · ab0c418
1 parent c8b84e2
commit ab0c418
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/cluster/calcium/node.go b/cluster/calcium/node.go
@@ -111,7 +111,15 @@ func (c *Calcium) RemoveNode(ctx context.Context, nodename string) error {
 func (c *Calcium) ListPodNodes(ctx context.Context, opts *types.ListNodesOptions) (<-chan *types.Node, error) {
 	logger := log.WithFunc("calcium.ListPodNodes").WithField("podname", opts.Podname).WithField("labels", opts.Labels).WithField("all", opts.All).WithField("info", opts.CallInfo)
 	nf := &types.NodeFilter{Podname: opts.Podname, Labels: opts.Labels, All: opts.All}
-	nodes, err := c.store.GetNodesByPod(ctx, nf, store.WithoutEngineOption())
+	var (
+		nodes []*types.Node
+		err   error
+	)
+	if opts.CallInfo {
+		nodes, err = c.store.GetNodesByPod(ctx, nf)
+	} else {
+		nodes, err = c.store.GetNodesByPod(ctx, nf, store.WithoutEngineOption())
+	}
 	if err != nil {
 		logger.Error(ctx, err)
 		return nil, err