Improve error handling

Author: avinassh

Cargo.lock

@@ -27,6 +27,7 @@ tracing-subscriber = { version = "0.3.16", features = ["env-filter"] }
 async-trait = "0.1.80"
 serde = "1.0.203"
 thiserror = "1.0.61"
+prost = "0.12.6"
 
 [features]
 foundation-db = ["foundationdb"]

libsql-storage-server/Cargo.toml

@@ -1,4 +1,6 @@
-use tonic::Status;
+use libsql_storage::rpc;
+use prost::Message;
+use tonic::{Code, Status};
 
 #[derive(Debug, thiserror::Error)]
 pub enum Error {
@@ -8,11 +10,34 @@ pub enum Error {
     Other(#[from] anyhow::Error),
 }
 
+impl Error {
+    fn code(&self) -> rpc::ErrorCode {
+        match self {
+            Error::WriteConflict => rpc::ErrorCode::WriteConflict,
+            Error::Other(_) => rpc::ErrorCode::InternalError,
+        }
+    }
+}
+
 impl From<Error> for Status {
     fn from(error: Error) -> Self {
-        match error {
-            Error::WriteConflict => Status::aborted("write conflict"),
-            Error::Other(err) => Status::internal(err.to_string()),
+        let status_code = match error.code() {
+            rpc::ErrorCode::InternalError => Code::Internal,
+            rpc::ErrorCode::WriteConflict => Code::Aborted,
+        };
+        let details = rpc::ErrorDetails {
+            message: error.to_string(),
+            code: error.code() as i32,
+        };
+
+        let mut details_buf = Vec::new();
+        if let Err(e) = details.encode(&mut details_buf) {
+            Status::new(
+                Code::Internal,
+                format!("failed to encode error details: {}", e),
+            )
+        } else {
+            Status::with_details(status_code, error.to_string(), details_buf.into())
         }
     }
 }

libsql-storage-server/src/errors.rs

@@ -6,11 +6,12 @@ use std::path::Path;
 use std::sync::{Arc, Mutex};
 
 use crate::local_cache::LocalCache;
-use crate::rpc::Frame;
+use crate::rpc::{ErrorCode, Frame};
 use libsql_sys::ffi::{SQLITE_ABORT, SQLITE_BUSY};
 use libsql_sys::name::{NamespaceName, NamespaceResolver};
 use libsql_sys::rusqlite;
 use libsql_sys::wal::{Result, Vfs, Wal, WalManager};
+use prost::Message;
 use rpc::storage_client::StorageClient;
 use tonic::transport::Channel;
 use tracing::{error, trace, warn};
@@ -365,10 +366,13 @@ impl Wal for DurableWal {
         let resp = binding.insert_frames(req);
         let resp = tokio::task::block_in_place(|| rt.block_on(resp));
         if let Err(e) = resp {
-            if e.code() == tonic::Code::Aborted {
-                return Err(rusqlite::ffi::Error::new(SQLITE_BUSY));
-            }
-            return Err(rusqlite::ffi::Error::new(SQLITE_ABORT));
+            let error_code = rpc::ErrorDetails::decode(e.details())
+                .ok()
+                .and_then(|details| rpc::ErrorCode::try_from(details.code).ok());
+            return match error_code {
+                Some(ErrorCode::WriteConflict) => Err(rusqlite::ffi::Error::new(SQLITE_BUSY)),
+                _ => Err(rusqlite::ffi::Error::new(SQLITE_ABORT)),
+            };
         }
         // TODO: fix parity with storage server frame num with local cache
         self.local_cache