fix: 修改 groupOperator 的方法为小些，内部方法

go-jarvis · Dec 8, 2021 · 73fd9ac · 73fd9ac
1 parent d193ed7
commit 73fd9ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/rum/group.go b/rum/group.go
@@ -12,7 +12,7 @@ var _ Operator = (*RouterGroup)(nil)
 var _ GroupOperator = (*RouterGroup)(nil)
 
 type GroupOperator interface {
-	RouterGroup() *RouterGroup
+	routerGroup() *RouterGroup
 }
 
 type RouterGroup struct {
@@ -40,7 +40,7 @@ func NewRouterGroup(path string) *RouterGroup {
 }
 
 // RouterGroup 实现 GroupOperator interface
-func (r *RouterGroup) RouterGroup() *RouterGroup {
+func (r *RouterGroup) routerGroup() *RouterGroup {
 	return r
 }
 
@@ -50,7 +50,7 @@ func (r *RouterGroup) Register(ops ...Operator) {
 	for _, op := range ops {
 		// 加入 子路由
 		if groupOp, ok := op.(GroupOperator); ok {
-			r.children[groupOp.RouterGroup()] = true
+			r.children[groupOp.routerGroup()] = true
 			continue
 		}
 

diff --git a/rum/launch.go b/rum/launch.go
@@ -0,0 +1,7 @@
+package rum
+
+func AddRootRoute(e *Engine, root Operator) *Engine {
+	e.Register(root)
+
+	return e
+}