Răsfoiți Sursa

remove v1 user groups

abhishek9686 11 luni în urmă
părinte
comite
3f25d5e136
1 a modificat fișierele cu 19 adăugiri și 1 ștergeri
  1. 19 1
      migrate/migrate.go

+ 19 - 1
migrate/migrate.go

@@ -21,11 +21,11 @@ import (
 func Run() {
 	updateEnrollmentKeys()
 	assignSuperAdmin()
+	removeOldUserGrps()
 	syncUsers()
 	updateHosts()
 	updateNodes()
 	updateAcls()
-
 }
 
 func assignSuperAdmin() {
@@ -124,6 +124,24 @@ func updateEnrollmentKeys() {
 	}
 }
 
+func removeOldUserGrps() {
+	rows, err := database.FetchRecords(database.USER_GROUPS_TABLE_NAME)
+	if err != nil {
+		return
+	}
+	for key, row := range rows {
+		userG := models.UserGroup{}
+		err = json.Unmarshal([]byte(row), &userG)
+		if err == nil {
+			if userG.ID == "" {
+				database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, key)
+			}
+		} else {
+			database.DeleteRecord(database.USER_GROUPS_TABLE_NAME, key)
+		}
+	}
+}
+
 func updateHosts() {
 	rows, err := database.FetchRecords(database.HOSTS_TABLE_NAME)
 	if err != nil {