ing

servlet.go

@@ -4,5 +4,7 @@ type Servlet interface {
 	ServletCtx(serverCtx ServerCtx) ServletCtx
 
 	Init(serverCtx ServerCtx) error
+	OnStart(serverCtx ServerCtx) error
+	OnStop(serverCtx ServerCtx)
 	Destroy(serverCtx ServerCtx)
 }

socket/net/server-handler.go

@@ -36,6 +36,28 @@ func (sh *ServerHandlers) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (sh *ServerHandlers) OnStart(serverCtx server.ServerCtx) error {
+	if err := sh.ServerHandlers.OnStart(serverCtx); nil != err {
+		return err
+	}
+
+	if nil != sh.servlet {
+		if err := sh.servlet.OnStart(serverCtx); nil != err {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (sh *ServerHandlers) OnStop(serverCtx server.ServerCtx) {
+	if nil != sh.servlet {
+		sh.servlet.OnStop(serverCtx)
+	}
+
+	sh.ServerHandlers.OnStop(serverCtx)
+}
+
 func (sh *ServerHandlers) Destroy(serverCtx server.ServerCtx) {
 	if nil != sh.servlet {
 		sh.servlet.Destroy(serverCtx)

socket/net/servlet.go

@@ -25,6 +25,14 @@ func (s *Servlets) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (s *Servlets) OnStart(serverCtx server.ServerCtx) error {
+	return nil
+}
+
+func (s *Servlets) OnStop(serverCtx server.ServerCtx) {
+	//
+}
+
 func (s *Servlets) Destroy(serverCtx server.ServerCtx) {
 	//
 }

socket/web/server-handler.go

@@ -42,6 +42,32 @@ func (sh *ServerHandlers) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (sh *ServerHandlers) OnStart(serverCtx server.ServerCtx) error {
+	if err := sh.ServerHandlers.OnStart(serverCtx); nil != err {
+		return err
+	}
+
+	if nil != sh.servlets {
+		for _, servlet := range sh.servlets {
+			if err := servlet.OnStart(serverCtx); nil != err {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func (sh *ServerHandlers) OnStop(serverCtx server.ServerCtx) {
+	if nil != sh.servlets {
+		for _, servlet := range sh.servlets {
+			servlet.OnStop(serverCtx)
+		}
+	}
+
+	sh.ServerHandlers.OnStop(serverCtx)
+}
+
 func (sh *ServerHandlers) Destroy(serverCtx server.ServerCtx) {
 	if nil != sh.servlets {
 		for _, servlet := range sh.servlets {

socket/web/servlet.go

@@ -24,6 +24,14 @@ func (s *Servlets) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (s *Servlets) OnStart(serverCtx server.ServerCtx) error {
+	return nil
+}
+
+func (s *Servlets) OnStop(serverCtx server.ServerCtx) {
+	//
+}
+
 func (s *Servlets) Destroy(serverCtx server.ServerCtx) {
 	//
 }

web/fasthttp/server-handler.go

@@ -48,6 +48,32 @@ func (sh *ServerHandlers) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (sh *ServerHandlers) OnStart(serverCtx server.ServerCtx) error {
+	if err := sh.ServerHandlers.OnStart(serverCtx); nil != err {
+		return err
+	}
+
+	if nil != sh.servlets {
+		for _, servlet := range sh.servlets {
+			if err := servlet.OnStart(serverCtx); nil != err {
+				return err
+			}
+		}
+	}
+
+	return nil
+}
+
+func (sh *ServerHandlers) OnStop(serverCtx server.ServerCtx) {
+	if nil != sh.servlets {
+		for _, servlet := range sh.servlets {
+			servlet.OnStop(serverCtx)
+		}
+	}
+
+	sh.ServerHandlers.OnStop(serverCtx)
+}
+
 func (sh *ServerHandlers) Destroy(serverCtx server.ServerCtx) {
 	if nil != sh.servlets {
 		for _, servlet := range sh.servlets {

web/fasthttp/servlet.go

@@ -24,6 +24,14 @@ func (s *Servlets) Init(serverCtx server.ServerCtx) error {
 	return nil
 }
 
+func (s *Servlets) OnStart(serverCtx server.ServerCtx) error {
+	return nil
+}
+
+func (s *Servlets) OnStop(serverCtx server.ServerCtx) {
+	//
+}
+
 func (s *Servlets) Destroy(serverCtx server.ServerCtx) {
 	//
 }