如何通过Juniper GraphQL传递Polars DataFrame？编译报错求助

你尝试在Juniper GraphQL中返回包含Polars DataFrame的HashMap，但遇到了 trait bound 不满足的编译错误，错误信息如下：

the trait bound `Result<HashMap<std::string::String, polars::prelude::DataFrame>, FieldError>: IntoResolvable<'_, __S, _, ()>` is not satisfied
the following other types implement trait `IntoResolvable<'a, S, T, C>`:
  <Result<(&'a <T as GraphQLValue<S2>>::Context, T), FieldError<S1>> as IntoResolvable<'a, S2, T, C>>
  <Result<T, E> as IntoResolvable<'a, S, T, C>>
  <Result<std::option::Option<(&'a <T as GraphQLValue<S2>>::Context, T)>, FieldError<S1>> as IntoResolvable<'a, S2, std::option::Option<T>, C>>

对应的Rust代码：

use juniper::{EmptySubscription,
            FieldResult,
            RootNode,
            EmptyMutation};
use polars::prelude::DataFrame;
use std::collections::HashMap;

pub struct QueryRoot;

#[juniper::graphql_object]
impl QueryRoot {
    fn fetch_data() -> FieldResult<HashMap<String, DataFrame>> {
        // ...
        let data: HashMap<String, DataFrame> = fetch_my_data();
        Ok(data)
    }
}

pub type Schema = RootNode<'static, QueryRoot, EmptyMutation, EmptySubscription>;

pub fn create_schema() -> Schema {
    Schema::new(QueryRoot {}, EmptyMutation::new(), EmptySubscription::new())
}

错误核心是Juniper无法自动识别Polars DataFrame作为GraphQL可解析类型，需要将DataFrame转换为Juniper支持的类型，或为DataFrame实现对应的GraphQL trait。

方案一：转换为JSON兼容的通用类型 #

将DataFrame序列化为Juniper原生支持的Vec<HashMap<String, serde_json::Value>>，无需额外定义类型：

1. 确保Cargo.toml中添加依赖并启用Polars的json特性：

[dependencies]
juniper = "0.15.10"
polars = { version = "0.35.4", features = ["json"] }
serde_json = "1.0.107"

2. 修改fetch_data函数：

use serde_json::Value;

#[juniper::graphql_object]
impl QueryRoot {
    fn fetch_data() -> FieldResult<HashMap<String, Vec<HashMap<String, Value>>>> {
        let data: HashMap<String, DataFrame> = fetch_my_data();
        let mut result = HashMap::new();
        
        for (key, df) in data {
            // 将DataFrame转为JSON字符串，再反序列化为通用JSON结构
            let json_str = df.to_json(Default::default())?;
            let rows: Vec<HashMap<String, Value>> = serde_json::from_str(&json_str)?;
            result.insert(key, rows);
        }
        
        Ok(result)
    }
}

方案二：自定义GraphQL类型包装DataFrame #

如果需要精细控制GraphQL中暴露的DataFrame字段，可以定义包装结构体并实现graphql_object：

1. 定义包装结构体并实现GraphQL对象：

use serde_json::Value;

pub struct PolarsDataFrame {
    inner: DataFrame,
}

#[juniper::graphql_object]
impl PolarsDataFrame {
    // 暴露列名列表
    fn columns(&self) -> Vec<String> {
        self.inner.get_column_names()
            .iter()
            .map(|name| name.to_string())
            .collect()
    }

    // 暴露数据行数
    fn row_count(&self) -> usize {
        self.inner.height()
    }

    // 暴露完整行数据
    fn rows(&self) -> FieldResult<Vec<HashMap<String, Value>>> {
        let json_str = self.inner.to_json(Default::default())?;
        let rows: Vec<HashMap<String, Value>> = serde_json::from_str(&json_str)?;
        Ok(rows)
    }
}

2. 修改QueryRoot的返回类型：

#[juniper::graphql_object]
impl QueryRoot {
    fn fetch_data() -> FieldResult<HashMap<String, PolarsDataFrame>> {
        let data: HashMap<String, DataFrame> = fetch_my_data();
        let result = data.into_iter()
            .map(|(key, df)| (key, PolarsDataFrame { inner: df }))
            .collect();
        Ok(result)
    }
}

内容的提问来源于stack exchange，提问作者921kiyo