extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2024-12-13 18:31:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add network routers in account object

Browse Source 
Signed-off-by: bcmmbaga <bethuelmbaga12@gmail.com>
This commit is contained in:
bcmmbaga 2024-12-10 14:59:55 +01:00
parent f9be8f829e
commit 21586acc16
No known key found for this signature in database
GPG Key ID: 511EED5C928AD547
2 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
management/server/account.go
View File

@ -278,7 +278,8 @@ type Account struct {
// Settings is a dictionary of Account settings
Settings *Settings `gorm:"embedded;embeddedPrefix:settings_"`
Networks []*networks.Network `gorm:"foreignKey:AccountID;references:id"`
Networks []*networks.Network `gorm:"foreignKey:AccountID;references:id"`
NetworkRouters []*networks.NetworkRouter `gorm:"foreignKey:AccountID;references:id"`
}
// Subclass used in gorm to only load settings and not whole account
@ -887,6 +888,11 @@ func (a *Account) Copy() *Account {
nets = append(nets, network.Copy())
}
networkRouters := []*networks.NetworkRouter{}
for _, router := range a.NetworkRouters {
networkRouters = append(networkRouters, router.Copy())
}
return &Account{
Id: a.Id,
CreatedBy: a.CreatedBy,
@ -906,6 +912,7 @@ func (a *Account) Copy() *Account {
PostureChecks: postureChecks,
Settings: settings,
Networks: nets,
NetworkRouters: networkRouters,
}
}

12
management/server/networks/network_router.go
View File

@ -31,3 +31,15 @@ func NewNetworkRouter(accountID string, networkID string, peer string, peerGroup
Metric: metric,
}, nil
}
func (n *NetworkRouter) Copy() *NetworkRouter {
return &NetworkRouter{
ID: n.ID,
NetworkID: n.NetworkID,
AccountID: n.AccountID,
Peer: n.Peer,
PeerGroups: n.PeerGroups,
Masquerade: n.Masquerade,
Metric: n.Metric,
}
}