mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-30 17:04:24 +08:00
22 lines
491 B
Go
22 lines
491 B
Go
package common
|
|
|
|
import (
|
|
"github.com/gravitl/netmaker/nm-proxy/models"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
)
|
|
|
|
func GetPeer(peerKey wgtypes.Key) (*models.Conn, bool) {
|
|
var peerInfo *models.Conn
|
|
var found bool
|
|
peerInfo, found = WgIfaceMap.PeerMap[peerKey.String()]
|
|
peerInfo.Mutex.RLock()
|
|
defer peerInfo.Mutex.RUnlock()
|
|
return peerInfo, found
|
|
|
|
}
|
|
|
|
func UpdatePeer(peer *models.Conn) {
|
|
peer.Mutex.Lock()
|
|
defer peer.Mutex.Unlock()
|
|
WgIfaceMap.PeerMap[peer.Key.String()] = peer
|
|
}
|