소스 검색

migration funcs

abhishek9686 1 년 전
부모
커밋
54c540ef64
3개의 변경된 파일26개의 추가작업 그리고 0개의 파일을 삭제
  1. 6 0
      logic/user_mgmt.go
  2. 19 0
      migrate/migrate.go
  3. 1 0
      models/user_mgmt.go

+ 6 - 0
logic/user_mgmt.go

@@ -21,6 +21,12 @@ var AdminPermissionTemplate = models.UserRolePermissionTemplate{
 	FullAccess: true,
 }
 
+var ServiceUserPermissionTemplate = models.UserRolePermissionTemplate{
+	ID:         models.ServiceUser,
+	Default:    true,
+	FullAccess: false,
+}
+
 var NetworkAdminPermissionTemplate = models.UserRolePermissionTemplate{
 	ID:                 models.NetworkAdmin,
 	Default:            true,

+ 19 - 0
migrate/migrate.go

@@ -311,3 +311,22 @@ func MigrateEmqx() {
 	}
 
 }
+
+func SyncUsers() {
+	users, err := logic.GetUsersDB()
+	if err == nil {
+		for _, user := range users {
+			if user.IsSuperAdmin {
+				user.PlatformRoleID = models.SuperAdminRole
+				logic.UpsertUser(user)
+			} else if user.IsAdmin {
+				user.PlatformRoleID = models.AdminRole
+				logic.UpsertUser(user)
+			} else {
+				user.PlatformRoleID = models.ServiceUser
+				logic.UpsertUser(user)
+			}
+
+		}
+	}
+}

+ 1 - 0
models/user_mgmt.go

@@ -51,6 +51,7 @@ const (
 const (
 	SuperAdminRole UserRole = "super_admin"
 	AdminRole      UserRole = "admin"
+	ServiceUser    UserRole = "user"
 	NetworkAdmin   UserRole = "network_admin"
 	NetworkUser    UserRole = "network_user"
 )