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/readwrite-handler.go

@@ -18,11 +18,11 @@ type ReadWriteHandler interface {
 type ReadWriteHandlers struct {
 	server.ReadWriteHandlers
 
-	PongTimeout time.Duration	`json:"pongTimeout"`
+	PongTimeout time.Duration `json:"pongTimeout"`
-	PingTimeout time.Duration	`json:"pingTimeout"`
+	PingTimeout time.Duration `json:"pingTimeout"`
-	PingPeriod  time.Duration	`json:"pingPeriod"`
+	PingPeriod  time.Duration `json:"pingPeriod"`
 
-	EnableCompression bool		`json:"enableCompression"`
+	EnableCompression bool `json:"enableCompression"`
 }
 
 func (rwh *ReadWriteHandlers) GetPongTimeout() time.Duration {

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) {
 	//
 }