File tree Expand file tree Collapse file tree 6 files changed +13
-13
lines changed
rust-mcp-sdk/src/mcp_runtimes Expand file tree Collapse file tree 6 files changed +13
-13
lines changed Original file line number Diff line number Diff line change @@ -76,13 +76,13 @@ impl McpClient for ClientRuntime {
76
76
where
77
77
MessageDispatcher < ServerMessage > : McpDispatch < ServerMessage , MessageFromClient > ,
78
78
{
79
- ( self . transport . message_sender ( ) . await ) as _
79
+ ( self . transport . message_sender ( ) ) as _
80
80
}
81
81
82
82
async fn start ( self : Arc < Self > ) -> SdkResult < ( ) > {
83
83
let mut stream = self . transport . start ( ) . await ?;
84
84
85
- let mut error_io_stream = self . transport . error_stream ( ) . await . write ( ) . await ;
85
+ let mut error_io_stream = self . transport . error_stream ( ) . write ( ) . await ;
86
86
let error_io_stream = error_io_stream. take ( ) ;
87
87
88
88
let self_clone = Arc :: clone ( & self ) ;
Original file line number Diff line number Diff line change @@ -66,14 +66,14 @@ impl McpServer for ServerRuntime {
66
66
where
67
67
MessageDispatcher < ClientMessage > : McpDispatch < ClientMessage , MessageFromServer > ,
68
68
{
69
- ( self . transport . message_sender ( ) . await ) as _
69
+ ( self . transport . message_sender ( ) ) as _
70
70
}
71
71
72
72
/// Main runtime loop, processes incoming messages and handles requests
73
73
async fn start ( & self ) -> SdkResult < ( ) > {
74
74
let mut stream = self . transport . start ( ) . await ?;
75
75
76
- let sender = self . transport . message_sender ( ) . await . read ( ) . await ;
76
+ let sender = self . transport . message_sender ( ) . read ( ) . await ;
77
77
let sender = sender
78
78
. as_ref ( )
79
79
. ok_or ( schema_utils:: SdkError :: connection_closed ( ) ) ?;
@@ -124,7 +124,7 @@ impl McpServer for ServerRuntime {
124
124
}
125
125
126
126
async fn stderr_message ( & self , message : String ) -> SdkResult < ( ) > {
127
- let mut lock = self . transport . error_stream ( ) . await . write ( ) . await ;
127
+ let mut lock = self . transport . error_stream ( ) . write ( ) . await ;
128
128
if let Some ( IoStream :: Writable ( stderr) ) = lock. as_mut ( ) {
129
129
stderr. write_all ( message. as_bytes ( ) ) . await ?;
130
130
stderr. write_all ( b"\n " ) . await ?;
Original file line number Diff line number Diff line change @@ -316,11 +316,11 @@ where
316
316
Ok ( stream)
317
317
}
318
318
319
- async fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
319
+ fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
320
320
& self . message_sender as _
321
321
}
322
322
323
- async fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
323
+ fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
324
324
& self . error_stream as _
325
325
}
326
326
Original file line number Diff line number Diff line change @@ -139,11 +139,11 @@ where
139
139
* result
140
140
}
141
141
142
- async fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
142
+ fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
143
143
& self . message_sender as _
144
144
}
145
145
146
- async fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
146
+ fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
147
147
& self . error_stream as _
148
148
}
149
149
Original file line number Diff line number Diff line change @@ -254,11 +254,11 @@ where
254
254
* result
255
255
}
256
256
257
- async fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
257
+ fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > {
258
258
& self . message_sender as _
259
259
}
260
260
261
- async fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
261
+ fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > {
262
262
& self . error_stream as _
263
263
}
264
264
Original file line number Diff line number Diff line change @@ -113,8 +113,8 @@ where
113
113
async fn start ( & self ) -> TransportResult < Pin < Box < dyn Stream < Item = R > + Send > > >
114
114
where
115
115
MessageDispatcher < R > : McpDispatch < R , S > ;
116
- async fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > ;
117
- async fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > ;
116
+ fn message_sender ( & self ) -> & tokio:: sync:: RwLock < Option < MessageDispatcher < R > > > ;
117
+ fn error_stream ( & self ) -> & tokio:: sync:: RwLock < Option < IoStream > > ;
118
118
async fn shut_down ( & self ) -> TransportResult < ( ) > ;
119
119
async fn is_shut_down ( & self ) -> bool ;
120
120
}
You can’t perform that action at this time.
0 commit comments