lease keep alive return lease not found error (#87)

Author: messense

Cargo.toml

@@ -19,7 +19,7 @@ tls-roots = ["tls", "tonic/tls-roots"]
 pub-response-field = ["visible"]
 
 [dependencies]
-tonic = "0.12"
+tonic = "0.12.3"
 prost = "0.13"
 tokio = "1.38"
 tokio-stream = "0.1"
@@ -36,7 +36,7 @@ hyper-util = { version = "0.1", features = ["client-legacy"], optional = true }
 tokio = { version = "1.38", features = ["full"] }
 
 [build-dependencies]
-tonic-build = { version = "0.12", default-features = false, features = ["prost"] }
+tonic-build = { version = "0.12.3", default-features = false, features = ["prost"] }
 
 [package.metadata.docs.rs]
 features = ["tls", "tls-roots"]

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

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

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(())
 }