Pārlūkot izejas kodu

add remote access tags in migration

abhishek9686 10 mēneši atpakaļ
vecāks
revīzija
14152a9484
1 mainītis faili ar 11 papildinājumiem un 1 dzēšanām
  1. 11 1
      migrate/migrate.go

+ 11 - 1
migrate/migrate.go

@@ -21,12 +21,12 @@ import (
 func Run() {
 	updateEnrollmentKeys()
 	assignSuperAdmin()
+	createDefaultTags()
 	removeOldUserGrps()
 	syncUsers()
 	updateHosts()
 	updateNodes()
 	updateAcls()
-	createDefaultTags()
 }
 
 func assignSuperAdmin() {
@@ -167,6 +167,16 @@ func updateNodes() {
 		return
 	}
 	for _, node := range nodes {
+		node := node
+		if node.IsIngressGateway {
+			tagID := models.TagID(fmt.Sprintf("%s.%s", node.Network,
+				models.RemoteAccessTagName))
+			if _, ok := node.Tags[tagID]; !ok {
+				node.Tags[tagID] = struct{}{}
+				logic.UpsertNode(&node)
+			}
+
+		}
 		if node.IsEgressGateway {
 			egressRanges, update := removeInterGw(node.EgressGatewayRanges)
 			if update {