@microsoft/rayfin-connector-fabric-semanticmodel 1.36.0-beta.0 → 1.36.1

package/dist/arrow.js

@@ -56,9 +56,48 @@ function coerceBigInt(raw, columnName) {
     }
     return Number(v);
 }
+/**
+ * Locate the chunk holding the logical `index` of a chunked vector, returning
+ * `undefined` when the index is out of range.
+ */
+function locateChunk(chunks, index) {
+    let local = index;
+    for (const data of chunks) {
+        if (local < data.length)
+            return { data, index: local };
+        local -= data.length;
+    }
+    return undefined;
+}
+/**
+ * Read the Union type id of a row, or `undefined` when the row is null or
+ * cannot be resolved.
+ *
+ * Power BI returns Variant columns (for example most measures, or
+ * `IF(cond, 5, "text")`) dictionary-encoded as `Dictionary<Int8, Union<...>>`.
+ * The schema-level type is unwrapped by {@link unwrapType}, but the row-level
+ * data then carries the dictionary keys in `values` and the Union itself in
+ * `dictionary`, so the type id must be read from the referenced dictionary
+ * entry rather than from the row data.
+ */
+function unionTypeIdAt(column, rowIndex) {
+    const row = locateChunk(column.data, rowIndex);
+    if (!row || !row.data.getValid(row.index))
+        return undefined;
+    let union = row;
+    if (row.data.dictionary) {
+        const key = Number(row.data.values[row.index]);
+        const entry = locateChunk(row.data.dictionary.data, key);
+        if (!entry || !entry.data.getValid(entry.index))
+            return undefined;
+        union = entry;
+    }
+    return union.data.typeIds?.[union.index];
+}
 /**
  * Resolve the active child coercer for a Union row so Decimal, Date/Timestamp,
  * and Int64 values follow the same conversion paths as non-Union columns.
+ * Handles both plain and dictionary-encoded Union columns.
  */
 function makeUnionCoercer(field) {
     const { type } = unwrapType(field);
@@ -68,17 +107,11 @@ function makeUnionCoercer(field) {
             return null;
         if (!column || rowIndex === undefined)
             return raw;
-        let chunkIndex = rowIndex;
-        for (const data of column.data) {
-            if (chunkIndex < data.length) {
-                const typeId = data.typeIds[chunkIndex];
-                const childIndex = type.typeIdToChildIndex[typeId];
-                const childCoercer = childCoercers[childIndex];
-                return childCoercer ? childCoercer(raw, columnName) : raw;
-            }
-            chunkIndex -= data.length;
-        }
-        return raw;
+        const typeId = unionTypeIdAt(column, rowIndex);
+        if (typeId === undefined)
+            return raw;
+        const childCoercer = childCoercers[type.typeIdToChildIndex[typeId]];
+        return childCoercer ? childCoercer(raw, columnName) : raw;
     };
 }
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@microsoft/rayfin-connector-fabric-semanticmodel",
-  "version": "1.36.0-beta.0",
+  "version": "1.36.1",
   "description": "Typed connector marker for the Fabric semantic-model (Power BI dataset) connector category. Pair with @microsoft/rayfin-client to type `client.connectors.<name>.executeQuery(...)`.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -28,7 +28,7 @@
   "dependencies": {
     "apache-arrow": "^21.1.0",
     "lz4js": "^0.2.0",
-    "@microsoft/rayfin-connectors": "1.36.0-beta.0"
+    "@microsoft/rayfin-connectors": "1.36.1"
   },
   "publishConfig": {
     "registry": "https://npm.pkg.github.com",