Skip to content

Commit

Permalink
fix(proxy): adjusted proxy events duplex
Browse files Browse the repository at this point in the history
  • Loading branch information
paulobressan committed Mar 5, 2024
1 parent 4aadb37 commit 90dff55
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 4 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
/target
proxy.socket
25 changes: 21 additions & 4 deletions proxy/src/proxy.rs
Original file line number Diff line number Diff line change
Expand Up @@ -47,18 +47,35 @@ impl ProxyApp {
) {
let mut upstream_buf = [0; 1024];
let mut downstream_buf = [0; 1024];

loop {
let downstream_read = server_session.read(&mut upstream_buf);
let upstream_read = client_session.read(&mut downstream_buf);
let event: DuplexEvent;
select! {
n = downstream_read => event
= DuplexEvent::DownstreamRead(n.unwrap()),
n = upstream_read => event
= DuplexEvent::UpstreamRead(n.unwrap()),
n = downstream_read => {
if let Err(err) = &n {
error!(error=err.to_string(), "Downstream error");
return;
}
event = DuplexEvent::DownstreamRead(n.unwrap())
},
n = upstream_read => {
if let Err(err) = &n {
error!(error=err.to_string(), "Upstream error");
return;
}
event = DuplexEvent::UpstreamRead(n.unwrap())
},
}

match event {
DuplexEvent::DownstreamRead(0) => {
return;
}
DuplexEvent::UpstreamRead(0) => {
return;
}
DuplexEvent::DownstreamRead(n) => {
state.metrics.count_total_packages_bytes(&consumer, n);

Expand Down

0 comments on commit 90dff55

Please sign in to comment.