diff --git a/cmd/kratos/go.mod b/cmd/kratos/go.mod index d7fa7268d..f75933470 100644 --- a/cmd/kratos/go.mod +++ b/cmd/kratos/go.mod @@ -8,4 +8,5 @@ require ( github.com/fatih/color v1.13.0 github.com/spf13/cobra v1.3.0 golang.org/x/mod v0.5.0 + golang.org/x/text v0.3.7 ) diff --git a/cmd/kratos/internal/proto/server/server.go b/cmd/kratos/internal/proto/server/server.go index 8fa131c35..b5707dac5 100644 --- a/cmd/kratos/internal/proto/server/server.go +++ b/cmd/kratos/internal/proto/server/server.go @@ -7,10 +7,15 @@ import ( "path" "strings" + "golang.org/x/text/cases" + "golang.org/x/text/language" + "github.com/emicklei/proto" "github.com/spf13/cobra" ) +var caser = cases.Title(language.English) + // CmdServer the service command. var CmdServer = &cobra.Command{ Use: "server", @@ -60,7 +65,7 @@ func run(cmd *cobra.Command, args []string) { r, ok := e.(*proto.RPC) if ok { cs.Methods = append(cs.Methods, &Method{ - Service: s.Name, Name: r.Name, Request: r.RequestType, + Service: s.Name, Name: caser.String(r.Name), Request: r.RequestType, Reply: r.ReturnsType, Type: getMethodType(r.StreamsRequest, r.StreamsReturns), }) }