From 1613f830e6501255e41dd0f19dd1517b5b2c538b Mon Sep 17 00:00:00 2001 From: Cobalt Date: Mon, 28 Mar 2022 11:20:44 +0200 Subject: [PATCH] feat: Add visibilty to codegen of route struct --- actix-web-codegen/src/route.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/actix-web-codegen/src/route.rs b/actix-web-codegen/src/route.rs index cb1ba1ef6..97aae1ec9 100644 --- a/actix-web-codegen/src/route.rs +++ b/actix-web-codegen/src/route.rs @@ -291,6 +291,7 @@ impl ToTokens for Route { resource_type, doc_attributes, } = self; + let visibility = &ast.vis; let resource_name = resource_name .as_ref() .map_or_else(|| name.to_string(), LitStr::value); @@ -316,7 +317,7 @@ impl ToTokens for Route { let stream = quote! { #(#doc_attributes)* #[allow(non_camel_case_types, missing_docs)] - pub struct #name; + #visibility struct #name; impl ::actix_web::dev::HttpServiceFactory for #name { fn register(self, __config: &mut actix_web::dev::AppService) {