瀏覽代碼

add migration func for old upstream servers

abhishek9686 2 周之前
父節點
當前提交
b043ed8cb7
共有 1 個文件被更改,包括 7 次插入1 次删除
  1. 7 1
      migrate/migrate.go

+ 7 - 1
migrate/migrate.go

@@ -43,6 +43,10 @@ func Run() {
 
 
 func migrateNameservers() {
 func migrateNameservers() {
 	nets, _ := logic.GetNetworks()
 	nets, _ := logic.GetNetworks()
+	user, err := logic.GetSuperAdmin()
+	if err != nil {
+		return
+	}
 	for _, netI := range nets {
 	for _, netI := range nets {
 		if len(netI.NameServers) > 0 {
 		if len(netI.NameServers) > 0 {
 			ns := schema.Nameserver{
 			ns := schema.Nameserver{
@@ -56,12 +60,14 @@ func migrateNameservers() {
 					"*": struct{}{},
 					"*": struct{}{},
 				},
 				},
 				Status:    true,
 				Status:    true,
-				CreatedBy: "auto",
+				CreatedBy: user.UserName,
 			}
 			}
 			for _, ip := range netI.NameServers {
 			for _, ip := range netI.NameServers {
 				ns.Servers = append(ns.Servers, ip)
 				ns.Servers = append(ns.Servers, ip)
 			}
 			}
 			ns.Create(db.WithContext(context.TODO()))
 			ns.Create(db.WithContext(context.TODO()))
+			netI.NameServers = []string{}
+			logic.SaveNetwork(&netI)
 		}
 		}
 	}
 	}
 }
 }