소스 검색

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