Mirrors/wg-portal
1
0
Fork 0
mirror of https://github.com/h44z/wg-portal.git synced 2025-12-16 03:26:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

migrate peer database

Browse Source
This commit is contained in:
Christoph Haas
2021-03-22 13:00:02 +01:00
parent f4edc55851
commit f95c692aed
3 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/wireguard/peermanager.go
View File

@@ -274,13 +274,23 @@ type PeerManager struct {
}
func NewPeerManager(db *gorm.DB, wg *Manager) (*PeerManager, error) {
um := &PeerManager{db: db, wg: wg}
pm := &PeerManager{db: db, wg: wg}
if err := um.db.AutoMigrate(&Peer{}, &Device{}); err != nil {
if err := pm.db.AutoMigrate(&Peer{}, &Device{}); err != nil {
return nil, errors.WithMessage(err, "failed to migrate peer database")
}
return um, nil
// check if peers without device name exist (from version <= 1.0.3), if so assign them to the default device.
peers := make([]Peer, 0)
pm.db.Find(&peers)
for i := range peers {
if peers[i].DeviceName == "" {
peers[i].DeviceName = wg.Cfg.GetDefaultDeviceName()
pm.db.Save(&peers[i])
}
}
return pm, nil
}
// InitFromPhysicalInterface read all WireGuard peers from the WireGuard interface configuration. If a peer does not