From 1e717b5414af791df97eb763e9fb123a6b57629c Mon Sep 17 00:00:00 2001
From: Bojan Serafimov <bojan.serafimov7@gmail.com>
Date: Tue, 18 Jan 2022 15:15:38 -0500
Subject: [PATCH] Don't error on health check

---
 proxy/src/proxy.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/proxy/src/proxy.rs b/proxy/src/proxy.rs
index 26936159d0..ebdee14505 100644
--- a/proxy/src/proxy.rs
+++ b/proxy/src/proxy.rs
@@ -168,7 +168,7 @@ impl ProxyConnection {
         loop {
             let msg = match self.pgb.read_message()? {
                 Some(Fe::StartupPacket(msg)) => msg,
-                None => bail!("connection is lost"),
+                None => return Ok(None), // Probably load balancer health check
                 bad => bail!("unexpected message type: {:?}", bad),
             };
             println!("got message: {:?}", msg);