From 48722ebe367e9d6d939fe8fc73ef602f68904804 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Thu, 9 Mar 2023 14:34:58 +0800 Subject: [PATCH] script --- pkg/common/cmd/rpc.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/pkg/common/cmd/rpc.go b/pkg/common/cmd/rpc.go index 359c93998..3e7ee4f30 100644 --- a/pkg/common/cmd/rpc.go +++ b/pkg/common/cmd/rpc.go @@ -3,6 +3,7 @@ package cmd import ( "OpenIM/internal/startrpc" "OpenIM/pkg/discoveryregistry" + "github.com/spf13/cobra" "google.golang.org/grpc" ) @@ -15,11 +16,11 @@ func NewAuthCmd() *AuthCmd { return authCmd } -func (a *AuthCmd) Execute() error { - //a.Command.Run = func(cmd *cobra.Command, args []string) { - // a.port = a.getPortFlag(cmd) - // a.prometheusPort = a.getPrometheusPortFlag(cmd) - //} +func (a *AuthCmd) Exec() error { + a.Command.Run = func(cmd *cobra.Command, args []string) { + a.port = a.getPortFlag(cmd) + a.prometheusPort = a.getPrometheusPortFlag(cmd) + } return a.Execute() }