1 files changed, 31 insertions, 0 deletions
diff --git a/vendor/github.com/authzed/spicedb/pkg/datastore/stats.go b/vendor/github.com/authzed/spicedb/pkg/datastore/stats.go
new file mode 100644
index 0000000..f98c405
--- /dev/null
+++ b/vendor/github.com/authzed/spicedb/pkg/datastore/stats.go
@@ -0,0 +1,31 @@
+package datastore
+
+import (
+	"github.com/authzed/spicedb/pkg/namespace"
+	iv1 "github.com/authzed/spicedb/pkg/proto/impl/v1"
+)
+
+// ComputeObjectTypeStats creates a list of object type stats from an input list of
+// parsed object types.
+func ComputeObjectTypeStats(objTypes []RevisionedNamespace) []ObjectTypeStat {
+	stats := make([]ObjectTypeStat, 0, len(objTypes))
+
+	for _, objType := range objTypes {
+		var relations, permissions uint32
+
+		for _, rel := range objType.Definition.Relation {
+			if namespace.GetRelationKind(rel) == iv1.RelationMetadata_PERMISSION {
+				permissions++
+			} else {
+				relations++
+			}
+		}
+
+		stats = append(stats, ObjectTypeStat{
+			NumRelations:   relations,
+			NumPermissions: permissions,
+		})
+	}
+
+	return stats
+}