Make sure group or collection cannot be removed if used by users or r…

…esources
kassisol · Apr 6, 2018 · f2eacc8 · f2eacc8
1 parent 04bd9ab
commit f2eacc8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/object/collection/collection.go b/object/collection/collection.go
@@ -55,6 +55,10 @@ func (c *Config) Remove(name string) error {
 		return fmt.Errorf("collection '%s' does not exist", name)
 	}
 
+	if len(c.Storage.ListResources(map[string]string{"elem": name})) > 0 {
+		return fmt.Errorf("collection '%s' cannot be removed. It is used by at least one resource", name)
+	}
+
 	if err := c.Storage.RemoveCollection(name); err != nil {
 		return err
 	}

diff --git a/object/group/group.go b/object/group/group.go
@@ -59,6 +59,10 @@ func (c *Config) Remove(name string) error {
 		return fmt.Errorf("group '%s' does not exist", name)
 	}
 
+	if len(c.Storage.ListUsers(map[string]string{"elem": name})) > 0 {
+		return fmt.Errorf("group '%s' cannot be removed. It is used by at least one user", name)
+	}
+
 	if err := c.Storage.RemoveGroup(name); err != nil {
 		return err
 	}