Update DatafileReader.kt

featurevisor · Jul 30, 2024 · 3733bf0 · 3733bf0
1 parent 9ac0175
commit 3733bf0
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/main/kotlin/com/featurevisor/sdk/DatafileReader.kt b/src/main/kotlin/com/featurevisor/sdk/DatafileReader.kt
@@ -1,8 +1,14 @@
 package com.featurevisor.sdk
 
-import com.featurevisor.types.*
-
-class DatafileReader(
+import com.featurevisor.types.Attribute
+import com.featurevisor.types.AttributeKey
+import com.featurevisor.types.DatafileContent
+import com.featurevisor.types.Feature
+import com.featurevisor.types.FeatureKey
+import com.featurevisor.types.Segment
+import com.featurevisor.types.SegmentKey
+
+class DatafileReader constructor(
     datafileContent: DatafileContent,
 ) {
 
@@ -20,8 +26,8 @@ class DatafileReader(
         return schemaVersion
     }
 
-    fun getAllAttributes(): Map<AttributeKey, Attribute> {
-        return attributes
+    fun getAllAttributes(): List<Attribute> {
+        return attributes.values.toList()
     }
 
     fun getAttribute(attributeKey: AttributeKey): Attribute? {