Still support Arc

github · tclem · Aug 20, 2024 · Aug 20, 2024 · Aug 20, 2024 · fcf5a5753be3c7585179f5fd2a9e9292fb9ca5d0
commit fcf5a5753be3c7585179f5fd2a9e9292fb9ca5d0
diff --git a/crates/twirp-build/src/lib.rs b/crates/twirp-build/src/lib.rs
@@ -38,6 +38,23 @@ impl prost_build::ServiceGenerator for ServiceGenerator {
         }
         writeln!(buf, "}}").unwrap();
 
+        writeln!(buf, "#[twirp::async_trait::async_trait]").unwrap();
+        writeln!(buf, "impl<T> {service_name} for std::sync::Arc<T>").unwrap();
+        writeln!(buf, "where").unwrap();
+        writeln!(buf, "    T: {service_name} + Sync + Send").unwrap();
+        writeln!(buf, "{{").unwrap();
+        for m in &service.methods {
+            writeln!(
+                buf,
+                "    async fn {}(&self, ctx: twirp::Context, req: {}) -> Result<{}, twirp::TwirpErrorResponse> {{",
+                m.name, m.input_type, m.output_type,
+            )
+                .unwrap();
+            writeln!(buf, "        (*self).{}(ctx, req).await", m.name).unwrap();
+            writeln!(buf, "    }}").unwrap();
+        }
+        writeln!(buf, "}}").unwrap();
+
         // add_service
         writeln!(
             buf,

diff --git a/example/src/main.rs b/example/src/main.rs
@@ -44,6 +44,7 @@ pub async fn main() {
     }
 }
 
+// Note: If your server type can't be Clone, consider wrapping it in `std::sync::Arc`.
 #[derive(Clone)]
 struct HaberdasherApiServer;
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,6 +44,7 @@ pub async fn main() { @@
         }
     }
+    // Note: If your server type can't be Clone, consider wrapping it in `std::sync::Arc`.
     #[derive(Clone)]
     struct HaberdasherApiServer;
@@ Expand Down @@