add Shutdown function

rookie-ninja · May 4, 2023 · baebf2c · baebf2c
1 parent 5c175cf
commit baebf2c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/boot.go b/boot.go
@@ -127,6 +127,13 @@ func (boot *Boot) Bootstrap(ctx context.Context) {
 func (boot *Boot) WaitForShutdownSig(ctx context.Context) {
 	rkentry.GlobalAppCtx.WaitForShutdownSig()
 
+	boot.Shutdown(ctx)
+}
+
+// Shutdown shutdown boot. for non-web application
+// 1: Call shutdown hook function added by user.
+// 2: Call interrupt function of entries in rkentry.GlobalAppCtx.
+func (boot *Boot) Shutdown(ctx context.Context) {
 	// Call shutdown hook function
 	for _, f := range rkentry.GlobalAppCtx.ListShutdownHooks() {
 		f()