Compare commits
1 Commits
f12e227df0
...
feat/http/
| Author | SHA1 | Date | |
|---|---|---|---|
| 6f05dd8995 |
@@ -4,7 +4,7 @@ use std::pin::Pin;
|
||||
use std::sync::Arc;
|
||||
|
||||
use alknet_core::types::Capabilities;
|
||||
use futures::stream::{self, Stream};
|
||||
use futures::stream::Stream;
|
||||
use serde_json::Value;
|
||||
|
||||
use super::context::{CompositionAuthority, OperationContext, ScopedPeerEnv};
|
||||
@@ -156,63 +156,6 @@ impl OperationRegistry {
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn invoke_streaming(
|
||||
&self,
|
||||
name: &str,
|
||||
input: Value,
|
||||
context: OperationContext,
|
||||
) -> ResponseStream {
|
||||
let request_id = context.request_id.clone();
|
||||
let name_owned = name.to_string();
|
||||
|
||||
let registration = match self.operations.get(name) {
|
||||
Some(r) => r,
|
||||
None => {
|
||||
return Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::not_found(request_id, &name_owned)
|
||||
}));
|
||||
}
|
||||
};
|
||||
|
||||
if registration.spec.visibility == Visibility::Internal && !context.internal {
|
||||
return Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::not_found(request_id, &name_owned)
|
||||
}));
|
||||
}
|
||||
|
||||
let acl = ®istration.spec.access_control;
|
||||
let identity = if context.internal {
|
||||
context
|
||||
.handler_identity
|
||||
.as_ref()
|
||||
.and_then(|ca| ca.as_identity())
|
||||
} else {
|
||||
context.identity.clone()
|
||||
};
|
||||
|
||||
if let AccessResult::Forbidden(message) = acl.check(identity.as_ref()) {
|
||||
return Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::forbidden(request_id, message)
|
||||
}));
|
||||
}
|
||||
|
||||
let streaming_handler = match ®istration.handler {
|
||||
HandlerKind::Stream(h) => Arc::clone(h),
|
||||
HandlerKind::Once(_) => {
|
||||
return Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::error(
|
||||
request_id,
|
||||
CallError::invalid_operation_type(
|
||||
"invoke_streaming() called on a Query/Mutation op; use invoke()",
|
||||
),
|
||||
)
|
||||
}));
|
||||
}
|
||||
};
|
||||
|
||||
streaming_handler(input, context)
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for OperationRegistry {
|
||||
@@ -1063,189 +1006,4 @@ mod tests {
|
||||
assert!(!err.retryable);
|
||||
assert!(err.details.is_none());
|
||||
}
|
||||
|
||||
async fn collect_stream(mut s: ResponseStream) -> Vec<ResponseEnvelope> {
|
||||
use futures::stream::StreamExt;
|
||||
let mut out = Vec::new();
|
||||
while let Some(env) = s.next().await {
|
||||
out.push(env);
|
||||
}
|
||||
out
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_on_subscription_dispatches_handler_stream() {
|
||||
let mut registry = OperationRegistry::new();
|
||||
registry
|
||||
.register(HandlerRegistration::new(
|
||||
subscription_spec("events/stream"),
|
||||
HandlerKind::Stream(echo_streaming_handler()),
|
||||
OperationProvenance::Local,
|
||||
None,
|
||||
None,
|
||||
Capabilities::new(),
|
||||
))
|
||||
.unwrap();
|
||||
let ctx = root_context("req-is-1", None, None, false, ScopedPeerEnv::empty());
|
||||
let stream = registry.invoke_streaming("events/stream", serde_json::json!({"v": 7}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
assert_eq!(items[0].request_id, "req-is-1");
|
||||
assert_eq!(items[0].result, Ok(serde_json::json!({"v": 7})));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_on_unknown_op_yields_single_not_found() {
|
||||
let registry = OperationRegistry::new();
|
||||
let ctx = root_context("req-is-2", None, None, false, ScopedPeerEnv::empty());
|
||||
let stream = registry.invoke_streaming("missing", serde_json::json!({}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
match &items[0].result {
|
||||
Err(e) => {
|
||||
assert_eq!(e.code, "NOT_FOUND");
|
||||
assert!(e.message.contains("missing"));
|
||||
}
|
||||
other => panic!("expected NOT_FOUND, got {other:?}"),
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_on_query_op_yields_invalid_operation_type() {
|
||||
let mut registry = OperationRegistry::new();
|
||||
registry
|
||||
.register(HandlerRegistration::new(
|
||||
external_spec("echo", AccessControl::default()),
|
||||
HandlerKind::Once(echo_handler()),
|
||||
OperationProvenance::Local,
|
||||
None,
|
||||
None,
|
||||
Capabilities::new(),
|
||||
))
|
||||
.unwrap();
|
||||
let ctx = root_context("req-is-3", None, None, false, ScopedPeerEnv::empty());
|
||||
let stream = registry.invoke_streaming("echo", serde_json::json!({}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
match &items[0].result {
|
||||
Err(e) => assert_eq!(e.code, "INVALID_OPERATION_TYPE"),
|
||||
other => panic!("expected INVALID_OPERATION_TYPE, got {other:?}"),
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_internal_op_from_external_yields_not_found() {
|
||||
let mut registry = OperationRegistry::new();
|
||||
registry
|
||||
.register(HandlerRegistration::new(
|
||||
internal_subscription_spec(AccessControl::default()),
|
||||
HandlerKind::Stream(echo_streaming_handler()),
|
||||
OperationProvenance::Local,
|
||||
None,
|
||||
None,
|
||||
Capabilities::new(),
|
||||
))
|
||||
.unwrap();
|
||||
let ctx = root_context("req-is-4", None, None, false, ScopedPeerEnv::empty());
|
||||
let stream = registry.invoke_streaming("events/stream", serde_json::json!({}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
match &items[0].result {
|
||||
Err(e) => {
|
||||
assert_eq!(e.code, "NOT_FOUND");
|
||||
assert!(e.message.contains("events/stream"));
|
||||
}
|
||||
other => panic!("expected NOT_FOUND, got {other:?}"),
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_acl_denied_yields_forbidden() {
|
||||
let mut registry = OperationRegistry::new();
|
||||
registry
|
||||
.register(HandlerRegistration::new(
|
||||
subscription_spec_with_acl(AccessControl {
|
||||
required_scopes: vec!["admin".to_string()],
|
||||
..Default::default()
|
||||
}),
|
||||
HandlerKind::Stream(echo_streaming_handler()),
|
||||
OperationProvenance::Local,
|
||||
None,
|
||||
None,
|
||||
Capabilities::new(),
|
||||
))
|
||||
.unwrap();
|
||||
let ctx = root_context(
|
||||
"req-is-5",
|
||||
Some(identity_with_scopes("caller", &["user"])),
|
||||
None,
|
||||
false,
|
||||
ScopedPeerEnv::empty(),
|
||||
);
|
||||
let stream = registry.invoke_streaming("events/stream", serde_json::json!({}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
match &items[0].result {
|
||||
Err(e) => {
|
||||
assert_eq!(e.code, "FORBIDDEN");
|
||||
assert!(e.message.contains("admin"));
|
||||
}
|
||||
other => panic!("expected FORBIDDEN, got {other:?}"),
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn invoke_streaming_internal_call_uses_handler_identity_for_acl() {
|
||||
let mut registry = OperationRegistry::new();
|
||||
let composing_authority = CompositionAuthority::new("agent-chat", ["admin".to_string()]);
|
||||
registry
|
||||
.register(HandlerRegistration::new(
|
||||
internal_subscription_spec(AccessControl {
|
||||
required_scopes: vec!["admin".to_string()],
|
||||
..Default::default()
|
||||
}),
|
||||
HandlerKind::Stream(echo_streaming_handler()),
|
||||
OperationProvenance::Local,
|
||||
None,
|
||||
None,
|
||||
Capabilities::new(),
|
||||
))
|
||||
.unwrap();
|
||||
let ctx = root_context(
|
||||
"req-is-6",
|
||||
Some(identity_with_scopes("user", &["user"])),
|
||||
Some(composing_authority),
|
||||
true,
|
||||
ScopedPeerEnv::empty(),
|
||||
);
|
||||
let stream = registry.invoke_streaming("events/stream", serde_json::json!({"ok": 1}), ctx);
|
||||
let items = collect_stream(stream).await;
|
||||
assert_eq!(items.len(), 1);
|
||||
assert_eq!(items[0].request_id, "req-is-6");
|
||||
assert_eq!(items[0].result, Ok(serde_json::json!({"ok": 1})));
|
||||
}
|
||||
|
||||
fn subscription_spec_with_acl(acl: AccessControl) -> OperationSpec {
|
||||
OperationSpec::new(
|
||||
"events/stream",
|
||||
OperationType::Subscription,
|
||||
Visibility::External,
|
||||
serde_json::json!({}),
|
||||
serde_json::json!({}),
|
||||
vec![],
|
||||
acl,
|
||||
)
|
||||
}
|
||||
|
||||
fn internal_subscription_spec(acl: AccessControl) -> OperationSpec {
|
||||
OperationSpec::new(
|
||||
"events/stream",
|
||||
OperationType::Subscription,
|
||||
Visibility::Internal,
|
||||
serde_json::json!({}),
|
||||
serde_json::json!({}),
|
||||
vec![],
|
||||
acl,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -18,13 +18,15 @@ use alknet_call::client::{AdapterError, OperationAdapter};
|
||||
use alknet_call::protocol::wire::{CallError, ResponseEnvelope};
|
||||
use alknet_call::registry::context::OperationContext;
|
||||
use alknet_call::registry::registration::{
|
||||
make_handler, HandlerKind, HandlerRegistration, OperationProvenance,
|
||||
make_handler, make_streaming_handler, HandlerKind, HandlerRegistration, OperationProvenance,
|
||||
ResponseStream,
|
||||
};
|
||||
use alknet_call::registry::spec::{
|
||||
AccessControl, ErrorDefinition, OperationSpec, OperationType, Visibility,
|
||||
};
|
||||
use alknet_core::types::Capabilities;
|
||||
use async_trait::async_trait;
|
||||
use futures::stream;
|
||||
use futures::StreamExt;
|
||||
use reqwest::header::{HeaderMap, HeaderName, HeaderValue, ACCEPT, AUTHORIZATION, CONTENT_TYPE};
|
||||
use reqwest::Method;
|
||||
@@ -440,38 +442,66 @@ impl FromOpenAPI {
|
||||
.map(|e| (e.http_status.unwrap_or(0), e.code.clone()))
|
||||
.collect();
|
||||
|
||||
let handler = make_handler(move |input: Value, context: OperationContext| {
|
||||
let path_template = path_template.clone();
|
||||
let method_upper = method_upper.clone();
|
||||
let auth_scheme = auth_scheme.clone();
|
||||
let default_headers = default_headers.clone();
|
||||
let base_url = base_url.clone();
|
||||
let namespace = namespace.clone();
|
||||
let http_client = Arc::clone(&http_client);
|
||||
let error_status_codes = error_status_codes.clone();
|
||||
let op_type = op_type;
|
||||
async move {
|
||||
forward(
|
||||
&http_client,
|
||||
&base_url,
|
||||
&path_template,
|
||||
&method_upper,
|
||||
&auth_scheme,
|
||||
&default_headers,
|
||||
&namespace,
|
||||
&error_status_codes,
|
||||
op_type,
|
||||
input,
|
||||
context,
|
||||
)
|
||||
.await
|
||||
}
|
||||
});
|
||||
let handler = if op_type == OperationType::Subscription {
|
||||
let stream_handler =
|
||||
make_streaming_handler(move |input: Value, context: OperationContext| {
|
||||
let path_template = path_template.clone();
|
||||
let method_upper = method_upper.clone();
|
||||
let auth_scheme = auth_scheme.clone();
|
||||
let default_headers = default_headers.clone();
|
||||
let base_url = base_url.clone();
|
||||
let namespace = namespace.clone();
|
||||
let http_client = Arc::clone(&http_client);
|
||||
let error_status_codes = error_status_codes.clone();
|
||||
forward_stream(
|
||||
&http_client,
|
||||
&base_url,
|
||||
&path_template,
|
||||
&method_upper,
|
||||
&auth_scheme,
|
||||
&default_headers,
|
||||
&namespace,
|
||||
&error_status_codes,
|
||||
input,
|
||||
context,
|
||||
)
|
||||
});
|
||||
HandlerKind::Stream(stream_handler)
|
||||
} else {
|
||||
let once_handler = make_handler(move |input: Value, context: OperationContext| {
|
||||
let path_template = path_template.clone();
|
||||
let method_upper = method_upper.clone();
|
||||
let auth_scheme = auth_scheme.clone();
|
||||
let default_headers = default_headers.clone();
|
||||
let base_url = base_url.clone();
|
||||
let namespace = namespace.clone();
|
||||
let http_client = Arc::clone(&http_client);
|
||||
let error_status_codes = error_status_codes.clone();
|
||||
let op_type = op_type;
|
||||
async move {
|
||||
forward(
|
||||
&http_client,
|
||||
&base_url,
|
||||
&path_template,
|
||||
&method_upper,
|
||||
&auth_scheme,
|
||||
&default_headers,
|
||||
&namespace,
|
||||
&error_status_codes,
|
||||
op_type,
|
||||
input,
|
||||
context,
|
||||
)
|
||||
.await
|
||||
}
|
||||
});
|
||||
HandlerKind::Once(once_handler)
|
||||
};
|
||||
|
||||
let capabilities = Capabilities::new();
|
||||
Ok(HandlerRegistration::new(
|
||||
spec,
|
||||
HandlerKind::Once(handler),
|
||||
handler,
|
||||
OperationProvenance::FromOpenAPI,
|
||||
None,
|
||||
None,
|
||||
@@ -666,10 +696,6 @@ async fn forward(
|
||||
|
||||
let status = response.status();
|
||||
|
||||
if op_type == OperationType::Subscription && status.is_success() {
|
||||
return stream_subscription(request_id, response).await;
|
||||
}
|
||||
|
||||
if !status.is_success() {
|
||||
let code = error_status_codes
|
||||
.iter()
|
||||
@@ -721,35 +747,136 @@ async fn forward(
|
||||
}
|
||||
}
|
||||
|
||||
async fn stream_subscription(request_id: String, response: reqwest::Response) -> ResponseEnvelope {
|
||||
let mut stream = response.bytes_stream();
|
||||
let mut buffer = String::new();
|
||||
let mut last_event: Option<Value> = None;
|
||||
while let Some(chunk_result) = stream.next().await {
|
||||
match chunk_result {
|
||||
Ok(chunk) => {
|
||||
buffer.push_str(&String::from_utf8_lossy(&chunk));
|
||||
let (events, remaining) = parse_sse_frames(&buffer);
|
||||
buffer = remaining;
|
||||
for event in events {
|
||||
let parsed = if event.data.trim().is_empty() {
|
||||
Value::Null
|
||||
} else {
|
||||
serde_json::from_str(&event.data)
|
||||
.unwrap_or(Value::String(event.data.clone()))
|
||||
};
|
||||
last_event = Some(parsed.clone());
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
fn forward_stream(
|
||||
http_client: &Arc<SharedHttpClient>,
|
||||
base_url: &str,
|
||||
path_template: &str,
|
||||
method: &str,
|
||||
auth_scheme: &Option<HttpAuthScheme>,
|
||||
default_headers: &HashMap<String, String>,
|
||||
namespace: &str,
|
||||
error_status_codes: &[(u16, String)],
|
||||
input: Value,
|
||||
context: OperationContext,
|
||||
) -> ResponseStream {
|
||||
let request_id = context.request_id.clone();
|
||||
|
||||
let (http_method, url, body, headers) = match build_request(
|
||||
base_url,
|
||||
path_template,
|
||||
method,
|
||||
auth_scheme,
|
||||
default_headers,
|
||||
namespace,
|
||||
&input,
|
||||
&context,
|
||||
) {
|
||||
Ok(parts) => parts,
|
||||
Err(err) => {
|
||||
return Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::error(request_id, err)
|
||||
}));
|
||||
}
|
||||
};
|
||||
|
||||
let http_client = Arc::clone(http_client);
|
||||
let error_status_codes = error_status_codes.to_vec();
|
||||
|
||||
let request_id_stream = request_id.clone();
|
||||
let error_status_codes_stream = error_status_codes.clone();
|
||||
|
||||
let init = async move {
|
||||
let request_builder = http_client
|
||||
.client()
|
||||
.request(http_method, url.as_str())
|
||||
.headers(headers)
|
||||
.header(ACCEPT, "text/event-stream");
|
||||
let request_builder = match body.as_ref() {
|
||||
Some(b) => {
|
||||
let serialized = serde_json::to_string(b).unwrap_or_else(|_| String::from("null"));
|
||||
request_builder.body(serialized)
|
||||
}
|
||||
None => request_builder,
|
||||
};
|
||||
request_builder.send().await
|
||||
};
|
||||
|
||||
let sse = stream::once(init).flat_map(move |result| {
|
||||
let request_id = request_id_stream.clone();
|
||||
let error_status_codes = error_status_codes_stream.clone();
|
||||
match result {
|
||||
Err(err) => Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::error(
|
||||
request_id,
|
||||
CallError::internal(format!("HTTP request failed: {err}")),
|
||||
)
|
||||
})) as ResponseStream,
|
||||
Ok(response) => {
|
||||
let status = response.status();
|
||||
if !status.is_success() {
|
||||
let code = error_status_codes
|
||||
.iter()
|
||||
.find(|(s, _)| *s == status.as_u16())
|
||||
.map(|(_, c)| c.clone())
|
||||
.unwrap_or_else(|| format!("HTTP_{}", status.as_u16()));
|
||||
let message = format!(
|
||||
"HTTP {}: {}",
|
||||
status.as_u16(),
|
||||
status.canonical_reason().unwrap_or("")
|
||||
);
|
||||
Box::pin(stream::once(async move {
|
||||
ResponseEnvelope::error(request_id, CallError::new(code, message, false))
|
||||
})) as ResponseStream
|
||||
} else {
|
||||
let request_id_inner = request_id.clone();
|
||||
Box::pin(
|
||||
stream::unfold(
|
||||
(response.bytes_stream(), String::new()),
|
||||
move |(mut bytes, mut buffer)| {
|
||||
let request_id = request_id_inner.clone();
|
||||
async move {
|
||||
match bytes.next().await {
|
||||
Some(Ok(chunk)) => {
|
||||
buffer.push_str(&String::from_utf8_lossy(&chunk));
|
||||
let (events, remaining) = parse_sse_frames(&buffer);
|
||||
let envelopes: Vec<ResponseEnvelope> = events
|
||||
.into_iter()
|
||||
.map(|e| {
|
||||
let parsed = if e.data.trim().is_empty() {
|
||||
Value::Null
|
||||
} else {
|
||||
serde_json::from_str(&e.data).unwrap_or(
|
||||
Value::String(e.data.clone()),
|
||||
)
|
||||
};
|
||||
ResponseEnvelope::ok(&request_id, parsed)
|
||||
})
|
||||
.collect();
|
||||
Some((envelopes, (bytes, remaining)))
|
||||
}
|
||||
Some(Err(err)) => {
|
||||
let error = CallError::internal(format!(
|
||||
"SSE stream error: {err}"
|
||||
));
|
||||
Some((
|
||||
vec![ResponseEnvelope::error(request_id, error)],
|
||||
(bytes, buffer),
|
||||
))
|
||||
}
|
||||
None => None,
|
||||
}
|
||||
}
|
||||
},
|
||||
)
|
||||
.flat_map(stream::iter),
|
||||
) as ResponseStream
|
||||
}
|
||||
}
|
||||
Err(err) => {
|
||||
return ResponseEnvelope::error(
|
||||
request_id,
|
||||
CallError::internal(format!("SSE stream error: {err}")),
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
ResponseEnvelope::ok(request_id, last_event.unwrap_or(Value::Null))
|
||||
});
|
||||
|
||||
Box::pin(sse)
|
||||
}
|
||||
|
||||
struct SseEvent {
|
||||
@@ -1194,6 +1321,34 @@ mod tests {
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn subscription_op_registration_is_handler_kind_stream() {
|
||||
let spec = OpenAPISpec::from_json(
|
||||
r##"{"openapi":"3.0.0","info":{"title":"T","version":"1"},
|
||||
"paths":{"/stream":{"post":{"operationId":"stream","responses":{"200":{"content":{"text/event-stream":{"schema":{}}}}}}}}}"##,
|
||||
)
|
||||
.unwrap();
|
||||
let bundles = adapter(spec, config("svc", "https://x", None))
|
||||
.import()
|
||||
.await
|
||||
.unwrap();
|
||||
assert!(matches!(bundles[0].handler, HandlerKind::Stream(_)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn query_op_registration_is_handler_kind_once() {
|
||||
let spec = OpenAPISpec::from_json(
|
||||
r#"{"openapi":"3.0.0","info":{"title":"T","version":"1"},
|
||||
"paths":{"/data":{"get":{"operationId":"data","responses":{"200":{"content":{"application/json":{"schema":{}}}}}}}}}"#,
|
||||
)
|
||||
.unwrap();
|
||||
let bundles = adapter(spec, config("svc", "https://x", None))
|
||||
.import()
|
||||
.await
|
||||
.unwrap();
|
||||
assert!(matches!(bundles[0].handler, HandlerKind::Once(_)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn integration_sse_subscription_streams_responded_events() {
|
||||
let sse_body = "data: {\"n\":1}\n\ndata: {\"n\":2}\n\n";
|
||||
@@ -1209,13 +1364,67 @@ mod tests {
|
||||
.unwrap();
|
||||
let registration = &bundles[0];
|
||||
let ctx = noop_context("req-12", Capabilities::new());
|
||||
let stream = match ®istration.handler {
|
||||
HandlerKind::Stream(h) => h(serde_json::json!({}), ctx),
|
||||
_ => panic!("expected Stream handler"),
|
||||
};
|
||||
let collected: Vec<ResponseEnvelope> = stream.collect().await;
|
||||
assert_eq!(collected.len(), 2);
|
||||
assert_eq!(collected[0].result, Ok(serde_json::json!({"n":1})));
|
||||
assert_eq!(collected[1].result, Ok(serde_json::json!({"n":2})));
|
||||
assert_eq!(collected[0].request_id, "req-12");
|
||||
assert_eq!(collected[1].request_id, "req-12");
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn integration_sse_subscription_http_error_returns_single_error_envelope() {
|
||||
let base = spawn_echo_server(404, r#"{"error":"missing"}"#, "application/json").await;
|
||||
let spec = OpenAPISpec::from_json(
|
||||
r##"{"openapi":"3.0.0","info":{"title":"T","version":"1"},
|
||||
"paths":{"/stream":{"post":{"operationId":"stream","responses":{
|
||||
"200":{"content":{"text/event-stream":{"schema":{}}}},
|
||||
"404":{"content":{"application/json":{"schema":{"type":"object"}}}}
|
||||
}}}}}"##,
|
||||
)
|
||||
.unwrap();
|
||||
let bundles = adapter(spec, config("svc", &base, None))
|
||||
.import()
|
||||
.await
|
||||
.unwrap();
|
||||
let registration = &bundles[0];
|
||||
let ctx = noop_context("req-err", Capabilities::new());
|
||||
let stream = match ®istration.handler {
|
||||
HandlerKind::Stream(h) => h(serde_json::json!({}), ctx),
|
||||
_ => panic!("expected Stream handler"),
|
||||
};
|
||||
let collected: Vec<ResponseEnvelope> = stream.collect().await;
|
||||
assert_eq!(collected.len(), 1);
|
||||
match &collected[0].result {
|
||||
Err(e) => assert_eq!(e.code, "HTTP_404"),
|
||||
other => panic!("expected HTTP_404 error, got {other:?}"),
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn integration_query_forwarding_unchanged_single_response() {
|
||||
let base = spawn_echo_server(200, r#"{"ok":true}"#, "application/json").await;
|
||||
let spec = OpenAPISpec::from_json(
|
||||
r#"{"openapi":"3.0.0","info":{"title":"T","version":"1"},
|
||||
"paths":{"/data":{"get":{"operationId":"data","responses":{"200":{"content":{"application/json":{"schema":{}}}}}}}}}"#,
|
||||
)
|
||||
.unwrap();
|
||||
let bundles = adapter(spec, config("svc", &base, None))
|
||||
.import()
|
||||
.await
|
||||
.unwrap();
|
||||
let registration = &bundles[0];
|
||||
let ctx = noop_context("req-q", Capabilities::new());
|
||||
let response = match ®istration.handler {
|
||||
HandlerKind::Once(h) => h(serde_json::json!({}), ctx).await,
|
||||
_ => panic!("expected Once handler"),
|
||||
};
|
||||
assert!(response.result.is_ok());
|
||||
let last = response.result.unwrap();
|
||||
assert_eq!(last, serde_json::json!({"n":2}));
|
||||
assert_eq!(response.request_id, "req-q");
|
||||
assert_eq!(response.result, Ok(serde_json::json!({"ok":true})));
|
||||
}
|
||||
|
||||
#[test]
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
---
|
||||
id: call/registry/invoke-streaming
|
||||
name: Implement OperationRegistry::invoke_streaming() returning ResponseStream
|
||||
status: completed
|
||||
status: pending
|
||||
depends_on: [call/registry/streaming-handler-handlerkind]
|
||||
scope: narrow
|
||||
risk: medium
|
||||
@@ -167,4 +167,4 @@ streams. The error envelope carries the `request_id` from the context.
|
||||
|
||||
## Summary
|
||||
|
||||
> Added OperationRegistry::invoke_streaming() in crates/alknet-call/src/registry/registration.rs — the streaming dispatch path for Subscription operations. Same visibility + ACL checks as invoke() (provably identical security axis), then dispatches the StreamingHandler and returns its ResponseStream. Pre-handler errors (not-found, forbidden, INVALID_OPERATION_TYPE for non-Subscription ops) yield a single error ResponseEnvelope via stream::once, then end. Added 6 unit tests covering all paths (subscription dispatch, unknown op, query op cross-kind error, internal op from external, ACL denied, internal call using handler_identity).
|
||||
> To be filled on completion
|
||||
Reference in New Issue
Block a user