rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-05-26 09:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pg protocol over SQL vaguely-semi-working

Browse Source
This commit is contained in:
George MacKerron
2023-05-16 07:42:51 +01:00
parent 2e05a9b652
commit b932c14f94
1 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
proxy/src/http/websocket.rs
View File

@@ -188,11 +188,31 @@ impl AsyncRead for MyObject {
cx: &mut Context<'_>, cx: &mut Context<'_>,
buf: &mut ReadBuf<'_>, buf: &mut ReadBuf<'_>,
) -> Poll<Result<(), io::Error>> { ) -> Poll<Result<(), io::Error>> {
let data = &self.get_mut().data; let data = &self.get_mut().data;
// let len: usize = data.len();
// let mut i: usize = 0;
// let mut zcount = 0;
// while len >= i + 5 {
// let cmd = data[i];
// info!("cmd: {}", cmd);
// if cmd == 0x5a { zcount += 1; }
// let size = u32::from_be_bytes(data[i..(i + 4)].try_into().unwrap());
// i += usize::try_from(size).unwrap();
// }
// if zcount < 2 {
// Poll::Pending
// } else {
// let mut reader = &data[..];
// Pin::new(&mut reader).poll_read(cx, buf)
//}
let mut reader = &data[..]; let mut reader = &data[..];
Pin::new(&mut reader).poll_read(cx, buf) Pin::new(&mut reader).poll_read(cx, buf)
} }
} }
impl AsyncWrite for MyObject { impl AsyncWrite for MyObject {
fn poll_write( fn poll_write(
self: Pin<&mut Self>, self: Pin<&mut Self>,
@@ -270,7 +290,7 @@ async fn ws_handler(
} }
let mut my_object = MyObject::new(data); let mut my_object = MyObject::new(data);
let handle = tokio::spawn(async move { let my_object = tokio::spawn(async move {
let result = handle_ws_client( let result = handle_ws_client(
config, config,
&cancel_map, &cancel_map,
@@ -279,8 +299,9 @@ async fn ws_handler(
host, host,
).await; ).await;
my_object my_object
}); })
let my_object = handle.await.map_err(|e| ApiError::InternalServerError(e.into()))?; .await
.map_err(|e| ApiError::InternalServerError(e.into()))?;
let response = Response::builder() let response = Response::builder()
.header("Content-Type", "application/octet-stream") .header("Content-Type", "application/octet-stream")