优化结构
This commit is contained in:
@@ -6,7 +6,6 @@ package module
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
"kra/pkg/database/migration"
|
||||
"kra/pkg/task"
|
||||
)
|
||||
|
||||
type Menu struct {
|
||||
@@ -33,7 +32,6 @@ type Definition struct {
|
||||
Migrations []migration.Step
|
||||
Surface Surface
|
||||
TimedTasks []TimedTask
|
||||
Tasks []task.Method
|
||||
}
|
||||
|
||||
type Catalog struct {
|
||||
@@ -48,15 +46,6 @@ func (c Catalog) MigrationSteps() []migration.Step {
|
||||
return steps
|
||||
}
|
||||
|
||||
func (c Catalog) Surface() Surface {
|
||||
var surface Surface
|
||||
for _, item := range c.Definitions {
|
||||
surface.Menus = append(surface.Menus, item.Surface.Menus...)
|
||||
surface.APIs = append(surface.APIs, item.Surface.APIs...)
|
||||
}
|
||||
return surface
|
||||
}
|
||||
|
||||
func (c Catalog) DefaultTimedTasks() []TimedTask {
|
||||
var tasks []TimedTask
|
||||
for _, item := range c.Definitions {
|
||||
@@ -65,41 +54,6 @@ func (c Catalog) DefaultTimedTasks() []TimedTask {
|
||||
return tasks
|
||||
}
|
||||
|
||||
func (c Catalog) TaskMethods() []task.Method {
|
||||
var methods []task.Method
|
||||
for _, item := range c.Definitions {
|
||||
methods = append(methods, item.Tasks...)
|
||||
}
|
||||
return methods
|
||||
}
|
||||
|
||||
type RouteRegistrar interface {
|
||||
RegisterRoutes(public, private *gin.RouterGroup, engine *gin.Engine)
|
||||
}
|
||||
|
||||
type Runtime struct {
|
||||
routes []RouteRegistrar
|
||||
}
|
||||
|
||||
func NewRuntime(routes ...RouteRegistrar) *Runtime {
|
||||
return &Runtime{routes: append([]RouteRegistrar(nil), routes...)}
|
||||
}
|
||||
|
||||
func (r *Runtime) Add(routes ...RouteRegistrar) {
|
||||
if r == nil {
|
||||
return
|
||||
}
|
||||
r.routes = append(r.routes, routes...)
|
||||
}
|
||||
|
||||
func (r *Runtime) RegisterRoutes(public, private *gin.RouterGroup, engine *gin.Engine) {
|
||||
if r == nil {
|
||||
return
|
||||
}
|
||||
for _, registrar := range r.routes {
|
||||
if registrar == nil {
|
||||
continue
|
||||
}
|
||||
registrar.RegisterRoutes(public, private, engine)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user