lease keep alive return lease not found error

build.rs

@@ -4,7 +4,7 @@ fn main() {
 
     tonic_build::configure()
         .build_server(false)
-        .compile(
+        .compile_protos(
             &[
                 "proto/auth.proto",
                 "proto/kv.proto",

examples/lease.rs

@@ -40,5 +40,10 @@ async fn main() -> Result<(), Error> {
     // revoke a lease
     let _resp = client.lease_revoke(id).await?;
     println!("revoke a lease with id {:?}", id);
+
+    // keep alive a revoked lease returns error
+    if let Err(err) = client.lease_keep_alive(id).await {
+        println!("revoked lease {:?} keep alive error: {:?}", id, err);
+    }
     Ok(())
 }

src/rpc/lease.rs

@@ -82,14 +82,17 @@ impl LeaseClient {
 
         let mut stream = self.inner.lease_keep_alive(receiver).await?.into_inner();
 
-        let id = match stream.message().await? {
-            Some(resp) => resp.id,
+        let (id, ttl) = match stream.message().await? {
+            Some(resp) => (resp.id, resp.ttl),
             None => {
                 return Err(Error::WatchError(
                     "failed to create lease keeper".to_string(),
                 ));
             }
         };
+        if ttl <= 0 {
+            return Err(Error::LeaseKeepAliveError("lease not found".to_string()));
+        }
 
         Ok((
             LeaseKeeper::new(id, sender),

tests/client.rs

@@ -294,6 +294,9 @@ async fn test_keep_alive() -> Result<()> {
     assert_eq!(resp.ttl(), 60);
 
     client.lease_revoke(id).await?;
+
+    // keep alive a revoked lease should return error
+    assert!(client.lease_keep_alive(id).await.is_err());
     Ok(())
 }