feat: Return StreamInfo from BasinClient::create_stream

Signed-off-by: Vaibhav Rabber <[email protected]>

examples/basic.rs

@@ -66,8 +66,8 @@ async fn main() {
     let basin_client = client.basin_client(basin.clone());
 
     match basin_client.create_stream(create_stream_req).await {
-        Ok(()) => {
-            println!("Stream created");
+        Ok(info) => {
+            println!("Stream created: {info:?}");
         }
         Err(ClientError::Service(status)) => {
             if status.code() == tonic::Code::AlreadyExists {

src/client.rs

@@ -472,7 +472,10 @@ impl BasinClient {
     }
 
     #[sync_docs]
-    pub async fn create_stream(&self, req: types::CreateStreamRequest) -> Result<(), ClientError> {
+    pub async fn create_stream(
+        &self,
+        req: types::CreateStreamRequest,
+    ) -> Result<types::StreamInfo, ClientError> {
         self.inner
             .send_retryable(CreateStreamServiceRequest::new(
                 self.inner.basin_service_client(),

src/service/basin.rs

@@ -107,7 +107,7 @@ impl CreateStreamServiceRequest {
 
 impl ServiceRequest for CreateStreamServiceRequest {
     type ApiRequest = api::CreateStreamRequest;
-    type Response = ();
+    type Response = types::StreamInfo;
     type ApiResponse = api::CreateStreamResponse;
     const IDEMPOTENCY_LEVEL: IdempotencyLevel = IdempotencyLevel::Idempotent;
 
@@ -127,9 +127,9 @@ impl ServiceRequest for CreateStreamServiceRequest {
 
     fn parse_response(
         &self,
-        _resp: tonic::Response<Self::ApiResponse>,
+        resp: tonic::Response<Self::ApiResponse>,
     ) -> Result<Self::Response, types::ConvertError> {
-        Ok(())
+        resp.into_inner().try_into()
     }
 }
 

src/types.rs

@@ -458,6 +458,16 @@ impl From<api::StreamInfo> for StreamInfo {
     }
 }
 
+impl TryFrom<api::CreateStreamResponse> for StreamInfo {
+    type Error = ConvertError;
+
+    fn try_from(value: api::CreateStreamResponse) -> Result<Self, Self::Error> {
+        let api::CreateStreamResponse { info } = value;
+        let info = info.ok_or("missing stream info")?;
+        Ok(info.into())
+    }
+}
+
 #[sync_docs]
 #[derive(Debug, Clone)]
 pub struct ListStreamsResponse {