fix: rpc plugin

nailyjs · Oct 29, 2024 · 26c6fb8 · 26c6fb8
1 parent d179f4a
commit 26c6fb8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/packages/rpc/src/rpc-bootstrap.ts b/packages/rpc/src/rpc-bootstrap.ts
@@ -82,7 +82,7 @@ export class RpcBootstrap<Adapter> extends BackendBootstrap {
    * @return {this}
    * @memberof RpcBootstrap
    */
-  use(rpcPlugin: RpcServerPlugin): this {
+  async use(rpcPlugin: RpcServerPlugin): Promise<this> {
     this._rpcPlugins.push(rpcPlugin)
     return this
   }

diff --git a/packages/rpc/src/rpc-plugin-protocol.ts b/packages/rpc/src/rpc-plugin-protocol.ts
@@ -1,6 +1,7 @@
+import type { PluginProtocol } from '@nailyjs/ioc'
 import type { RpcBootstrap } from './rpc-bootstrap'
 
-export interface RpcServerPlugin {
+export interface RpcServerPlugin extends PluginProtocol {
   name: string
   beforeRun?(bootstrap: RpcBootstrap<any>): any
   beforeListen?(bootstrap: RpcBootstrap<any>): any