Function flattenNoSQLQueryResult

frontend/src/utils/sqlResult.ts:267–332

(
  result: QueryResult
)

};

export const flattenNoSQLQueryResult = (
  result: QueryResult
): QueryResult | undefined => {
  if (
    result.columnNames.length !== 1 ||
    result.columnNames[0] !== "result" ||
    result.rows.length === 0
  ) {
    return undefined;
  }

  const { columns, columnIndexMap } = getNoSQLColumns(result.rows);
  if (columns.length === 0) {
    return undefined;
  }

  const rows: QueryRow[] = [];
  const columnTypeNames: string[] = Array.from({ length: columns.length }).map(
    () => "TEXT"
  );

  for (const row of result.rows) {
    if (row.values.length !== 1 || row.values[0].kind.case !== "stringValue") {
      return undefined;
    }

    // Use lossless-json to preserve precision for large integers (> 2^53-1)
    const data = losslessParse(
      row.values[0].kind.value,
      null,
      losslessReviver
    ) as Record<string, unknown>;
    const values: RowValue[] = Array.from({ length: columns.length }).map(() =>
      createProto(RowValueSchema, {
        kind: {
          case: "nullValue",
          value: NullValue.NULL_VALUE,
        },
      })
    );

    for (const [key, value] of Object.entries(data)) {
      const index = columnIndexMap.get(key) ?? 0;
      const { value: formatted, type } = convertAnyToRowValue(value, true);
      values[index] = formatted;
      columnTypeNames[index] = type;
    }

    rows.push(
      createProto(QueryRowSchema, {
        values,
      })
    );
  }

  return createProto(QueryResultSchema, {
    columnNames: columns,
    columnTypeNames,