From d7c0b29556f19a623218de317e4c6ab9f7e0d9a0 Mon Sep 17 00:00:00 2001 From: "mayuhang03@meituan.com" Date: Fri, 22 Mar 2024 12:47:12 +0800 Subject: [PATCH] fix node state export --- memberlist.go | 1 + util_test.go | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/memberlist.go b/memberlist.go index c496ecd8b..509222d4b 100644 --- a/memberlist.go +++ b/memberlist.go @@ -611,6 +611,7 @@ func (m *Memberlist) Members() []*Node { nodes := make([]*Node, 0, len(m.nodes)) for _, n := range m.nodes { if !n.DeadOrLeft() { + n.Node.State = n.State nodes = append(nodes, &n.Node) } } diff --git a/util_test.go b/util_test.go index f15b4d216..7565cbc2f 100644 --- a/util_test.go +++ b/util_test.go @@ -255,13 +255,14 @@ func TestKRandomNodes(t *testing.T) { nodes = append(nodes, &nodeState{ Node: Node{ Name: fmt.Sprintf("test%d", i), + State: state, }, State: state, }) } filterFunc := func(n *nodeState) bool { - if n.Name == "test0" || n.State != StateAlive { + if n.Name == "test0" || n.Node.State != StateAlive { return true } return false