tstest/integration/testcontrol: add Server.nodeLocked

This commit is contained in:
Brad Fitzpatrick 2021-06-04 08:19:23 -07:00
parent aa6abc98f3
commit 58cc2cc921
1 changed files with 7 additions and 0 deletions

View File

@ -176,6 +176,13 @@ func (s *Server) serveMachine(w http.ResponseWriter, r *http.Request) {
func (s *Server) Node(nodeKey tailcfg.NodeKey) *tailcfg.Node {
s.mu.Lock()
defer s.mu.Unlock()
return s.nodeLocked(nodeKey)
}
// nodeLocked returns the node for nodeKey. It's always nil or cloned memory.
//
// s.mu must be held.
func (s *Server) nodeLocked(nodeKey tailcfg.NodeKey) *tailcfg.Node {
return s.nodes[nodeKey].Clone()
}