fix(config): correct the function signature of the rpcinfo.TimeoutPro…

…vider implementation
cloudwego · Jul 25, 2023 · d534906 · d534906
1 parent b26b928
commit d534906
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pkg/rpctimeout/item_rpc_timeout.go b/pkg/rpctimeout/item_rpc_timeout.go
@@ -27,8 +27,9 @@ import (
 )
 
 var (
-	_ iface.ConfigValueItem = (*RPCTimeout)(nil)
-	_ rpcinfo.Timeouts      = (*RPCTimeout)(nil)
+	_ iface.ConfigValueItem   = (*RPCTimeout)(nil)
+	_ rpcinfo.Timeouts        = (*RPCTimeout)(nil)
+	_ rpcinfo.TimeoutProvider = (*Container)(nil)
 )
 
 // TypeRPCTimeout is used as itemKey in ConfigValueImpl
@@ -134,7 +135,7 @@ func (c *Container) NotifyPolicyChange(configs map[string]*RPCTimeout) {
 }
 
 // Timeouts return the rpc timeout config by the method name of rpc info.
-func (c *Container) Timeouts(ri rpcinfo.RPCInfo) *RPCTimeout {
+func (c *Container) Timeouts(ri rpcinfo.RPCInfo) rpcinfo.Timeouts {
 	rtc := c.config.Load().(*rpcTimeoutConfig)
 	if config, ok := rtc.configs[ri.Invocation().MethodName()]; ok {
 		return config