Add async service protocol

This allows to implement a server using either synchronous or asynchronous functions
Cryptact · Sep 6, 2024 · 58f6cfc · 58f6cfc
1 parent 6c37ccf
commit 58f6cfc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/protoc-gen-twirpy/generator/template.go b/protoc-gen-twirpy/generator/template.go
@@ -33,11 +33,16 @@ class {{.Name}}ServiceProtocol(Protocol):
 	def {{.Name}}(self, ctx: Context, request: {{.QualifiedInput}}) -> {{.QualifiedOutput}}: ...
 	{{- end }}
 
+class Async{{.Name}}ServiceProtocol(Protocol):
+	{{- range .Methods }}
+	async def {{.Name}}(self, ctx: Context, request: {{.QualifiedInput}}) -> {{.QualifiedOutput}}: ...
+	{{- end }}
+
 class {{.Name}}Server(TwirpServer):
 	def __init__(
 		self,
 		*,
-		service: {{.Name}}ServiceProtocol,
+		service: {{.Name}}ServiceProtocol | Async{{.Name}}ServiceProtocol,
 		server_path_prefix: str = "/twirp"
 	):
 		super().__init__(service=service)