Skip to content

Commit d027e7a

Browse files
committed
refactor: added *Module type parameter to Module's LoadFn and RegisterRoutesFn
1 parent e1e5546 commit d027e7a

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

http/module.go

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ type (
1515
GetValidator() interface{}
1616
GetController() interface{}
1717
GetPath() string
18-
GetLoadFn() func()
18+
GetLoadFn() func(*Module)
19+
GetRegisterRoutesFn() func(*Module)
1920
GetSubmodules() *[]ModuleWrapper
2021
gonethttproute.RouterWrapper
2122
}
@@ -26,8 +27,8 @@ type (
2627
Service interface{}
2728
Validator interface{}
2829
Controller interface{}
29-
LoadFn func()
30-
RegisterRoutesFn func()
30+
LoadFn func(*Module)
31+
RegisterRoutesFn func(*Module)
3132
Submodules *[]ModuleWrapper
3233
gonethttproute.RouterWrapper
3334
}
@@ -52,7 +53,7 @@ func (m *Module) Create(
5253

5354
// Register the routes
5455
if m.RegisterRoutesFn != nil {
55-
m.RegisterRoutesFn()
56+
m.RegisterRoutesFn(m)
5657
}
5758

5859
// Create the submodules controllers router
@@ -67,7 +68,7 @@ func (m *Module) Create(
6768

6869
// Load the module
6970
if m.LoadFn != nil {
70-
m.LoadFn()
71+
m.LoadFn(m)
7172
}
7273
return nil
7374
}
@@ -103,11 +104,16 @@ func (m *Module) GetController() interface{} {
103104
}
104105

105106
// GetLoadFn is a function that returns the load function
106-
func (m *Module) GetLoadFn() func() {
107+
func (m *Module) GetLoadFn() func(*Module) {
107108
return m.LoadFn
108109
}
109110

110111
// GetSubmodules is a function that returns the submodules
111112
func (m *Module) GetSubmodules() *[]ModuleWrapper {
112113
return m.Submodules
113114
}
115+
116+
// GetRegisterRoutesFn is a function that returns the register routes function
117+
func (m *Module) GetRegisterRoutesFn() func(*Module) {
118+
return m.RegisterRoutesFn
119+
}

0 commit comments

Comments
 (0)