refactor(application): reorganize lifecycle hooks and move interfaces to separate files

feat: add collectHooks method to automatically collect lifecycle hooks from loaded goners
refactor: move Default application instance and related methods to default.go
refactor: move lifecycle interfaces to stage_interface.go and update mock generation
test: add TestApplication_collectHooks to verify hook collection order

Author: Degfy

application.go

@@ -20,8 +20,6 @@ type Application struct {
 	signal chan os.Signal
 }
 
-var Default = NewApp()
-
 // NewApp creates and initializes a new Application instance.
 // It creates an empty Application struct and calls init() to:
 // 1. Initialize signal channel
@@ -80,10 +78,6 @@ func (s *Application) Load(goner Goner, options ...Option) *Application {
 	return s
 }
 
-func Load(goner Goner, options ...Option) *Application {
-	return Default.Load(goner, options...)
-}
-
 // Loads executes multiple LoadFuncs in sequence to load goner for Application
 // Parameters:
 //   - loads: Variadic LoadFunc parameters that will be executed in order
@@ -100,10 +94,6 @@ func (s *Application) Loads(loads ...LoadFunc) *Application {
 	return s
 }
 
-func Loads(loads ...LoadFunc) *Application {
-	return Default.Loads(loads...)
-}
-
 // BeforeStart registers a function to be called before starting the application.
 // The function will be executed before any daemons are started.
 // Returns the Application instance for method chaining.
@@ -167,13 +157,6 @@ func (s *Application) End() *Application {
 	return s
 }
 
-// End triggers application termination
-// It terminates the application by sending a SIGINT signal to the default Application instance
-// This is a convenience method equivalent to calling Default.End()
-func End() {
-	Default.End()
-}
-
 func (s *Application) start() {
 	for _, fn := range s.beforeStartHooks {
 		fn()
@@ -215,6 +198,34 @@ func (s *Application) install() {
 	}
 }
 
+func (s *Application) collectHooks() {
+	coffins := s.loader.iKeeper.getAllCoffins()
+	for _, co := range coffins {
+		if co.goner != nil {
+			if start, ok := co.goner.(BeforeStarter); ok {
+				s.beforeStart(func() {
+					start.BeforeStart()
+				})
+			}
+			if afterStart, ok := co.goner.(AfterStarter); ok {
+				s.afterStart(func() {
+					afterStart.AfterStart()
+				})
+			}
+			if stop, ok := co.goner.(BeforeStoper); ok {
+				s.beforeStop(func() {
+					stop.BeforeStop()
+				})
+			}
+			if afterStop, ok := co.goner.(AfterStoper); ok {
+				s.afterStop(func() {
+					afterStop.AfterStop()
+				})
+			}
+		}
+	}
+}
+
 // Run initializes the application, injects dependencies into the provided function,
 // executes it, and then performs cleanup.
 // The function can have dependencies that will be automatically injected.
@@ -224,6 +235,7 @@ func (s *Application) install() {
 //   - funcList: The function to execute with injected dependencies
 func (s *Application) Run(funcList ...any) {
 	s.install()
+	s.collectHooks()
 	s.start()
 
 	var options []RunOption
@@ -247,8 +259,11 @@ func (s *Application) Run(funcList ...any) {
 	s.stop()
 }
 
-func Run(fn ...any) {
-	Default.Run(fn...)
+// Serve initializes the application, starts all daemons, and waits for termination signal.
+// After receiving termination signal, performs cleanup by stopping all daemons.
+func (s *Application) Serve(funcList ...any) {
+	funcList = append(funcList, OpWaitEnd())
+	s.Run(funcList...)
 }
 
 type RunOption interface {
@@ -265,17 +280,6 @@ func OpWaitEnd() RunOption {
 	return waitEnd{}
 }
 
-// Serve initializes the application, starts all daemons, and waits for termination signal.
-// After receiving termination signal, performs cleanup by stopping all daemons.
-func (s *Application) Serve(funcList ...any) {
-	funcList = append(funcList, OpWaitEnd())
-	s.Run(funcList...)
-}
-
-func Serve() {
-	Default.Serve()
-}
-
 type TestFlag interface {
 	forTest()
 }
@@ -291,11 +295,6 @@ func (s *Application) Test(fn any) {
 	s.Run(fn)
 }
 
-// Test for run tests
-func Test(fn any) {
-	Default.Test(fn)
-}
-
 // RunTest Deprecated, use Test instead
 func RunTest(fn any, priests ...LoadFunc) {
 	NewApp(priests...).Test(fn)

application_test.go

@@ -2,6 +2,7 @@ package gone_test
 
 import (
 	"errors"
+	"go.uber.org/mock/gomock"
 	"reflect"
 	"strings"
 	"sync"
@@ -558,3 +559,61 @@ func TestEnd(t *testing.T) {
 		}).
 		Serve()
 }
+
+func TestApplication_collectHooks(t *testing.T) {
+	controller := gomock.NewController(t)
+	defer controller.Finish()
+
+	var order, beforeStartOrder, afterStartOrder, beforeStopOrder, afterStopOrder int
+
+	starter := gone.NewMockBeforeStarter(controller)
+	starter.EXPECT().BeforeStart().Do(func() {
+		order++
+		beforeStartOrder = order
+	})
+	afterStarter := gone.NewMockAfterStarter(controller)
+	afterStarter.EXPECT().AfterStart().Do(func() {
+		order++
+		afterStartOrder = order
+	})
+
+	stoper := gone.NewMockBeforeStoper(controller)
+	stoper.EXPECT().BeforeStop().Do(func() {
+		order++
+		beforeStopOrder = order
+	})
+
+	afterStoper := gone.NewMockAfterStoper(controller)
+
+	afterStoper.EXPECT().AfterStop().Do(func() {
+		order++
+		afterStopOrder = order
+	})
+	gone.
+		NewApp().
+		Load(stoper).
+		Load(afterStoper).
+		Load(starter).
+		Load(afterStarter).
+		Run(func() {
+			if beforeStartOrder != 1 {
+				t.Errorf("beforeStartOrder = %d, want %d", beforeStartOrder, 1)
+			}
+			if afterStartOrder != 2 {
+				t.Errorf("afterStartOrder = %d, want %d", afterStartOrder, 2)
+			}
+			if beforeStopOrder != 0 {
+				t.Errorf("beforeStopOrder = %d, want %d", beforeStopOrder, 0)
+			}
+			if afterStopOrder != 0 {
+				t.Errorf("afterStopOrder = %d, want %d", afterStopOrder, 0)
+			}
+		})
+
+	if beforeStopOrder != 3 {
+		t.Errorf("beforeStopOrder = %d, want %d", beforeStopOrder, 3)
+	}
+	if afterStopOrder != 4 {
+		t.Errorf("afterStopOrder = %d, want %d", afterStopOrder, 4)
+	}
+}

default.go

@@ -0,0 +1,53 @@
+package gone
+
+// Default is a singleton instance of Application, used to simplify common usage patterns.
+var Default = NewApp()
+
+// Loads uses the default application instance to load multiple LoadFunc functions.
+// Parameters:
+//   - loads: One or more LoadFunc functions that will be executed in sequence
+//
+// Returns:
+//   - *Application: Returns the default application instance for method chaining
+func Loads(loads ...LoadFunc) *Application {
+	return Default.Loads(loads...)
+}
+
+// Load uses the default application instance to load a Goner with optional configuration options.
+// Parameters:
+//   - goner: The Goner instance to load
+//   - options: Optional configuration options for the Goner
+//
+// Returns:
+//   - *Application: Returns the default application instance for method chaining
+func Load(goner Goner, options ...Option) *Application {
+	return Default.Load(goner, options...)
+}
+
+// Run executes one or more functions using the default application instance.
+// These functions can have dependencies that will be automatically injected.
+// Parameters:
+//   - fn: A variadic list of functions to execute with injected dependencies
+func Run(fn ...any) {
+	Default.Run(fn...)
+}
+
+// Serve starts all daemons and waits for termination signal using the default application instance.
+// This function will start all registered daemons and block until a shutdown signal is received.
+func Serve() {
+	Default.Serve()
+}
+
+// End triggers application termination
+// It terminates the application by sending a SIGINT signal to the default Application instance
+// This is a convenience method equivalent to calling Default.End()
+func End() {
+	Default.End()
+}
+
+// Test runs tests using the default application instance.
+// Parameters:
+//   - fn: The test function to execute with injected dependencies
+func Test(fn any) {
+	Default.Test(fn)
+}

iface.go

@@ -82,6 +82,7 @@ type FuncInjectHook func(pt reflect.Type, i int, injected bool) any
 
 //go:generate mockgen  -source=./iface.go    -package=gone -destination=iface_mock.go
 //go:generate mockgen -source=./interface.go -package=gone -destination=interface_mock.go
+//go:generate mockgen -source=./stage_interface.go -package=gone -destination=stage_mock.go
 //go:generate mockgen -source=./logger.go    -package=gone -destination=./logger_mock.go
 //go:generate mockgen -source=./config.go    -package=gone -destination=./config_mock.go