Fix issues regarding serialization and error handling

build.gradle.kts

@@ -10,7 +10,8 @@ import org.gradle.api.tasks.testing.logging.TestLogEvent
 
 plugins {
     // Apply the org.jetbrains.kotlin.jvm Plugin to add support for Kotlin.
-    id("org.jetbrains.kotlin.jvm") version "1.8.20"
+    kotlin("jvm") version "1.9.20"
+    kotlin("plugin.serialization") version "1.9.20"
 
     // Apply the java-library plugin for API and implementation separation.
     `java-library`

src/main/kotlin/com/featurevisor/sdk/DatafileReader.kt

@@ -6,7 +6,6 @@ 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(
     datafileJson: DatafileContent,
@@ -34,7 +33,7 @@ class DatafileReader constructor(
         return attributes.find { attribute -> attribute.key == attributeKey }
     }
 
-    fun getSegment(segmentKey: SegmentKey): Segment? {
+    fun getSegment(segmentKey: String): Segment? {
         return segments.find { segment -> segment.key == segmentKey }
     }
 

src/main/kotlin/com/featurevisor/sdk/Instance+Evaluation.kt

@@ -612,7 +612,7 @@ private fun FeaturevisorInstance.getBucketKey(feature: Feature, context: Context
 
         is BucketBy.Or -> {
             type = "or"
-            attributeKeys = bucketBy.bucketBy.or
+            attributeKeys = bucketBy.bucketBy
         }
     }
 
@@ -632,9 +632,9 @@ private fun FeaturevisorInstance.getBucketKey(feature: Feature, context: Context
 
     bucketKey.add(AttributeValue.StringValue(featureKey))
 
-    val result = bucketKey.map {
+    val result = bucketKey.joinToString(separator = bucketKeySeparator) {
         it.toString()
-    }.joinToString(separator = bucketKeySeparator)
+    }
 
     configureBucketKey?.let { configureBucketKey ->
         return configureBucketKey(feature, context, result)

src/main/kotlin/com/featurevisor/sdk/Instance+Feature.kt

@@ -7,7 +7,7 @@ import com.featurevisor.types.Feature
 import com.featurevisor.types.Force
 import com.featurevisor.types.Traffic
 
-internal fun FeaturevisorInstance.getFeatureByKey(featureKey: String): Feature? {
+fun FeaturevisorInstance.getFeatureByKey(featureKey: String): Feature? {
     return datafileReader.getFeature(featureKey)
 }
 

src/main/kotlin/com/featurevisor/sdk/Instance+Fetch.kt

@@ -39,21 +39,36 @@ private fun fetchDatafileContentFromUrl(
     }
 }
 
-private inline fun <reified T> fetch(
+const val BODY_BYTE_COUNT = 1000000L
+private inline fun fetch(
     request: Request,
-    crossinline completion: (Result<T>) -> Unit,
+    crossinline completion: (Result<DatafileContent>) -> Unit,
 ) {
     val client = OkHttpClient()
     val call = client.newCall(request)
     call.enqueue(object : Callback {
         override fun onResponse(call: Call, response: Response) {
-            val responseBody = response.body
-            if (response.isSuccessful && responseBody != null) {
-                val json = Json { ignoreUnknownKeys = true }
-                val content = json.decodeFromString<T>(responseBody.string())
+            val responseBody = response.peekBody(BODY_BYTE_COUNT)
+            if (response.isSuccessful) {
+                val json = Json {
+                    ignoreUnknownKeys = true
+                    serializersModule = this.serializersModule.apply {
+
+                    }
+                }
+                val responseBodyString = responseBody.string()
+                FeaturevisorInstance.companionLogger?.debug(responseBodyString)
+                val content = json.decodeFromString<DatafileContent>(responseBodyString)
                 completion(Result.success(content))
             } else {
-                completion(Result.failure(FeaturevisorError.UnparsableJson(responseBody?.string(), response.message)))
+                completion(
+                    Result.failure(
+                        FeaturevisorError.UnparsableJson(
+                            responseBody.string(),
+                            response.message
+                        )
+                    )
+                )
             }
         }
 

src/main/kotlin/com/featurevisor/sdk/Instance.kt

@@ -30,6 +30,8 @@ class FeaturevisorInstance private constructor(options: InstanceOptions) {
         fun createInstance(options: InstanceOptions): FeaturevisorInstance {
             return FeaturevisorInstance(options)
         }
+
+        var companionLogger: Logger? = null
     }
 
     private val on: (EventName, Listener) -> Unit
@@ -58,6 +60,7 @@ class FeaturevisorInstance private constructor(options: InstanceOptions) {
 
     init {
         with(options) {
+            companionLogger = logger
             if (onReady != null) {
                 emitter.addListener(event = READY, listener = onReady)
             }
@@ -96,7 +99,7 @@ class FeaturevisorInstance private constructor(options: InstanceOptions) {
                         if (result.isSuccess) {
                             datafileReader = DatafileReader(result.getOrThrow())
                             statuses.ready = true
-                            emitter.emit(READY)
+                            emitter.emit(READY, result.getOrThrow())
                             if (refreshInterval != null) startRefreshing()
                         } else {
                             logger?.error("Failed to fetch datafile: $result")