storage-bigtable: Return error if attempting to write to table NotFound

storage-bigtable/src/bigtable.rs

@@ -6,7 +6,7 @@ use {
         compression::{compress_best, decompress},
         root_ca_certificate, CredentialType,
     },
-    backoff::{future::retry, ExponentialBackoff},
+    backoff::{future::retry, Error as BackoffError, ExponentialBackoff},
     log::*,
     std::{
         str::FromStr,
@@ -84,6 +84,15 @@ pub enum Error {
     Timeout,
 }
 
+fn check_table_not_found<T>(result: Result<T>) -> std::result::Result<T, BackoffError<Error>> {
+    if let Err(Error::Rpc(ref err)) = result {
+        if err.code() == tonic::Code::NotFound && err.message().starts_with("table") {
+            return Err(BackoffError::Permanent(Error::Rpc(err.clone())));
+        }
+    }
+    Ok(result?)
+}
+
 impl std::convert::From<std::io::Error> for Error {
     fn from(err: std::io::Error) -> Self {
         Self::Io(err)
@@ -265,7 +274,8 @@ impl BigTableConnection {
     {
         retry(ExponentialBackoff::default(), || async {
             let mut client = self.client();
-            Ok(client.put_bincode_cells(table, cells).await?)
+            let result = client.put_bincode_cells(table, cells).await;
+            check_table_not_found(result)
         })
         .await
     }
@@ -303,7 +313,8 @@ impl BigTableConnection {
     {
         retry(ExponentialBackoff::default(), || async {
             let mut client = self.client();
-            Ok(client.put_protobuf_cells(table, cells).await?)
+            let result = client.put_protobuf_cells(table, cells).await;
+            check_table_not_found(result)
         })
         .await
     }