rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: route issue

Browse Source
This commit is contained in:
Rogee
2025-01-10 14:27:40 +08:00
parent 312b30ce42
commit 2eef4c35f4
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/ast/provider/provider.go
View File

@@ -275,7 +275,7 @@ func Parse(source string) []Provider {
if providerDoc.Mode == "event" { if providerDoc.Mode == "event" {
provider.Mode = "event" provider.Mode = "event"
modePkg := gomod.GetModuleName() + "/providers/events" modePkg := gomod.GetModuleName() + "/providers/event"
provider.Imports["git.ipao.vip/rogeecn/atom"] = "" provider.Imports["git.ipao.vip/rogeecn/atom"] = ""
provider.Imports["git.ipao.vip/rogeecn/atom/contracts"] = "" provider.Imports["git.ipao.vip/rogeecn/atom/contracts"] = ""
@@ -288,7 +288,7 @@ func Parse(source string) []Provider {
Star: "*", Star: "*",
Type: "PubSub", Type: "PubSub",
Package: modePkg, Package: modePkg,
PackageAlias: "events", PackageAlias: "event",
} }
} }

10
pkg/ast/route/render.go
View File

@@ -71,7 +71,7 @@ func Render(path string, routes []RouteDefinition) error {
case PositionURI: case PositionURI:
return fmt.Sprintf(`URI[%s]("%s")`, item.Type, item.Name), true return fmt.Sprintf(`URI[%s]("%s")`, item.Type, item.Name), true
case PositionQuery: case PositionQuery:
return fmt.Sprintf(`Query[%s]("%s")`, item.Type, item.Name), true return fmt.Sprintf(`Query%s[%s]("%s")`, isScalarType(item.Type), item.Type, item.Name), true
case PositionHeader: case PositionHeader:
return fmt.Sprintf(`Header[%s]("%s")`, item.Type, item.Name), true return fmt.Sprintf(`Header[%s]("%s")`, item.Type, item.Name), true
case PositionCookie: case PositionCookie:
@@ -108,3 +108,11 @@ func Render(path string, routes []RouteDefinition) error {
} }
return nil return nil
} }
func isScalarType(t string) string {
switch t {
case "string", "int", "int32", "int64", "float32", "float64", "bool":
return "Param"
}
return ""
}