Fix crash when a prefix family was empty

This commit is contained in:
Juan Font 2024-04-17 13:23:29 +00:00
parent 40953727cf
commit c4c8cfe5ea

View File

@ -377,14 +377,18 @@ func NewHeadscaleDatabase(
} }
} }
err = tx.Model(&types.Node{}).Where("id = ?", node.ID).Update("ipv4", v4.String()).Error if v4 != nil {
if err != nil { err = tx.Model(&types.Node{}).Where("id = ?", node.ID).Update("ipv4", v4.String()).Error
return fmt.Errorf("saving ip addresses to new columns: %w", err) if err != nil {
return fmt.Errorf("saving ip addresses to new columns: %w", err)
}
} }
err = tx.Model(&types.Node{}).Where("id = ?", node.ID).Update("ipv6", v6.String()).Error if v6 != nil {
if err != nil { err = tx.Model(&types.Node{}).Where("id = ?", node.ID).Update("ipv6", v6.String()).Error
return fmt.Errorf("saving ip addresses to new columns: %w", err) if err != nil {
return fmt.Errorf("saving ip addresses to new columns: %w", err)
}
} }
} }