extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-08-17 02:21:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

Save Peer Status separately in the FileStore (#554)

Browse Source 
Due to peer reconnects when restarting the Management service,
there are lots of SaveStore operations to update peer status.

Store.SavePeerStatus stores peer status separately and the
FileStore implementation stores it in memory.
This commit is contained in:
Misha Bragin
2022-11-08 10:46:12 +01:00
committed by GitHub
parent 7e262572a4
commit f37b43a542
13 changed files with 150 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
management/server/file_store_test.go
View File

@ -93,7 +93,7 @@ func TestStore(t *testing.T) {
return
}
restored, err := NewStore(store.storeFile)
restored, err := NewFileStore(store.storeFile)
if err != nil {
return
}
@ -129,7 +129,7 @@ func TestRestore(t *testing.T) {
t.Fatal(err)
}
store, err := NewStore(storeDir)
store, err := NewFileStore(storeDir)
if err != nil {
return
}
@ -161,7 +161,7 @@ func TestGetAccountByPrivateDomain(t *testing.T) {
t.Fatal(err)
}
store, err := NewStore(storeDir)
store, err := NewFileStore(storeDir)
if err != nil {
return
}
@ -190,7 +190,7 @@ func TestFileStore_GetAccount(t *testing.T) {
t.Fatal(err)
}
store, err := NewStore(storeDir)
store, err := NewFileStore(storeDir)
if err != nil {
t.Fatal(err)
}
@ -218,8 +218,59 @@ func TestFileStore_GetAccount(t *testing.T) {
assert.Len(t, account.NameServerGroups, len(expected.NameServerGroups))
}
func TestFileStore_SavePeerStatus(t *testing.T) {
storeDir := t.TempDir()
err := util.CopyFileContents("testdata/store.json", filepath.Join(storeDir, "store.json"))
if err != nil {
t.Fatal(err)
}
store, err := NewFileStore(storeDir)
if err != nil {
return
}
account, err := store.getAccount("bf1c8084-ba50-4ce7-9439-34653001fc3b")
if err != nil {
t.Fatal(err)
}
// save status of non-existing peer
newStatus := PeerStatus{Connected: true, LastSeen: time.Now()}
err = store.SavePeerStatus(account.Id, "non-existing-peer", newStatus)
assert.Error(t, err)
// save new status of existing peer
account.Peers["testpeer"] = &Peer{
Key: "peerkey",
SetupKey: "peerkeysetupkey",
IP: net.IP{127, 0, 0, 1},
Meta: PeerSystemMeta{},
Name: "peer name",
Status: &PeerStatus{Connected: false, LastSeen: time.Now()},
}
err = store.SaveAccount(account)
if err != nil {
t.Fatal(err)
}
err = store.SavePeerStatus(account.Id, "testpeer", newStatus)
if err != nil {
t.Fatal(err)
}
account, err = store.getAccount(account.Id)
if err != nil {
t.Fatal(err)
}
actual := account.Peers["testpeer"].Status
assert.Equal(t, newStatus, *actual)
}
func newStore(t *testing.T) *FileStore {
store, err := NewStore(t.TempDir())
store, err := NewFileStore(t.TempDir())
if err != nil {
t.Errorf("failed creating a new store")
}