use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
};
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone)]
pub struct RecordSearchBuilder {
config: Config,
namespace: String,
search: String,
fields: Vec<String>,
page: Option<u32>,
page_size: Option<u32>,
}
impl RecordSearchBuilder {
pub fn new(config: Config, namespace: impl Into<String>, search: impl Into<String>) -> Self {
Self {
config,
namespace: namespace.into(),
search: search.into(),
fields: Vec::new(),
page: None,
page_size: None,
}
}
pub fn field(mut self, field: impl Into<String>) -> Self {
self.fields.push(field.into());
self
}
pub fn fields(mut self, fields: impl IntoIterator<Item = impl Into<String>>) -> Self {
self.fields.extend(fields.into_iter().map(Into::into));
self
}
pub fn page(mut self, page: u32) -> Self {
self.page = Some(page);
self
}
pub fn page_size(mut self, page_size: u32) -> Self {
self.page_size = Some(page_size);
self
}
pub async fn execute(self) -> SDKResult<RecordSearchResponse> {
self.execute_with_options(RequestOption::default()).await
}
pub async fn execute_with_options(
self,
option: RequestOption,
) -> SDKResult<RecordSearchResponse> {
let url = format!(
"/open-apis/apaas/v1/applications/{}/objects/search",
self.namespace
);
let request = RecordSearchRequest {
search: self.search,
fields: self.fields,
page: self.page,
page_size: self.page_size,
};
let req: ApiRequest<RecordSearchResponse> =
ApiRequest::post(&url).body(serde_json::to_value(&request)?);
let resp = Transport::request(req, &self.config, Some(option)).await?;
resp.data
.ok_or_else(|| openlark_core::error::validation_error("Operation", "响应数据为空"))
}
}
#[derive(Debug, Clone, Deserialize, Serialize)]
struct RecordSearchRequest {
#[serde(rename = "search")]
search: String,
#[serde(rename = "fields", skip_serializing_if = "Vec::is_empty")]
fields: Vec<String>,
#[serde(rename = "page", skip_serializing_if = "Option::is_none")]
page: Option<u32>,
#[serde(rename = "page_size", skip_serializing_if = "Option::is_none")]
page_size: Option<u32>,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct SearchedRecord {
#[serde(rename = "id")]
id: String,
#[serde(rename = "data")]
data: serde_json::Value,
#[serde(rename = "score")]
score: f64,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct RecordSearchResponse {
#[serde(rename = "items")]
items: Vec<SearchedRecord>,
#[serde(rename = "has_more")]
has_more: bool,
#[serde(rename = "page")]
page: u32,
#[serde(rename = "page_size")]
page_size: u32,
}
impl ApiResponseTrait for RecordSearchResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[cfg(test)]
mod tests {
use serde_json;
#[test]
fn test_serialization_roundtrip() {
let json = r#"{"test": "value"}"#;
assert!(serde_json::from_str::<serde_json::Value>(json).is_ok());
}
#[test]
fn test_deserialization_from_json() {
let json = r#"{"field": "data"}"#;
let value: serde_json::Value = serde_json::from_str(json).expect("JSON 反序列化失败");
assert_eq!(value["field"], "data");
}
}