modify the service name from proto (#230)

This commit is contained in:
Keson
2020-11-28 11:48:44 +08:00
committed by GitHub
parent 3ebb1e0221
commit 2dfecda465
3 changed files with 15 additions and 8 deletions

View File

@@ -33,6 +33,7 @@ type (
GetSvc() Dir
GetPb() Dir
GetMain() Dir
GetServiceName() stringx.String
}
Dir struct {
@@ -41,7 +42,8 @@ type (
Package string
}
defaultDirContext struct {
inner map[string]Dir
inner map[string]Dir
serviceName stringx.String
}
)
@@ -110,8 +112,10 @@ func mkdir(ctx *ctx.ProjectContext, proto parser.Proto) (DirContext, error) {
return nil, err
}
}
serviceName := strings.TrimSuffix(proto.Name, filepath.Ext(proto.Name))
return &defaultDirContext{
inner: inner,
inner: inner,
serviceName: stringx.From(strings.ReplaceAll(serviceName, "-", "")),
}, nil
}
@@ -151,6 +155,10 @@ func (d *defaultDirContext) GetMain() Dir {
return d.inner[wd]
}
func (d *defaultDirContext) GetServiceName() stringx.String {
return d.serviceName
}
func (d *Dir) Valid() bool {
return len(d.Filename) > 0 && len(d.Package) > 0
}