From 9a68cbc02b1cdabef2c340ba11f3e40ee3eb58b7 Mon Sep 17 00:00:00 2001
From: crusader <crusader@loafle.com>
Date: Thu, 19 Apr 2018 15:35:55 +0900
Subject: [PATCH] ing

---
 server/server-handler.go | 79 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/server/server-handler.go b/server/server-handler.go
index 1b991db..3f876fb 100644
--- a/server/server-handler.go
+++ b/server/server-handler.go
@@ -1,6 +1,14 @@
 package server
 
 import (
+	"fmt"
+	"io/ioutil"
+	"net"
+	"os"
+	"strconv"
+
+	logging "git.loafle.net/commons/logging-go"
+	"git.loafle.net/commons/server-go"
 	cssn "git.loafle.net/commons/server-go/socket/net"
 )
 
@@ -10,4 +18,75 @@ type ServerHandler interface {
 
 type ServerHandlers struct {
 	cssn.ServerHandlers
+
+	PIDFilePath string
+
+	port int
+}
+
+func (sh *ServerHandlers) Init(serverCtx server.ServerCtx) error {
+	if err := sh.ServerHandlers.Init(serverCtx); nil != err {
+		return err
+	}
+
+	return nil
+}
+
+func (sh *ServerHandlers) OnStart(serverCtx server.ServerCtx) error {
+	if err := sh.ServerHandlers.OnStart(serverCtx); nil != err {
+		return err
+	}
+
+	if _, err := os.Stat(sh.PIDFilePath); os.IsExist(err) {
+		if err := os.Remove(sh.PIDFilePath); nil != err {
+			logging.Logger().Errorf("Container[%s]: Removing pid file has been failed [%v]", sh.Name, err)
+		}
+	}
+
+	s := strconv.FormatInt(int64(sh.port), 10)
+	if err := ioutil.WriteFile(sh.PIDFilePath, []byte(s), os.ModePerm); nil != err {
+		return err
+	}
+
+	return nil
+}
+
+func (sh *ServerHandlers) OnStop(serverCtx server.ServerCtx) {
+	if _, err := os.Stat(sh.PIDFilePath); os.IsExist(err) {
+		if err := os.Remove(sh.PIDFilePath); nil != err {
+			logging.Logger().Errorf("Container: Removing pid file has been failed [%v]", err)
+		}
+	}
+
+	sh.ServerHandlers.OnStop(serverCtx)
+}
+
+func (sh *ServerHandlers) Destroy(serverCtx server.ServerCtx) {
+
+	sh.ServerHandlers.Destroy(serverCtx)
+}
+
+func (sh *ServerHandlers) Listener(serverCtx server.ServerCtx) (net.Listener, error) {
+	for i := 60000; i < 61000; i++ {
+		addr := fmt.Sprintf("localhost:%d", i)
+		l, err := net.Listen("tcp", addr)
+		if nil == err {
+			sh.port = i
+			return l, nil
+		}
+	}
+
+	return nil, fmt.Errorf("Container[%s]: Cannot find availrable port", sh.Name)
+}
+
+func (sh *ServerHandlers) Validate() error {
+	if err := sh.ServerHandlers.Validate(); nil != err {
+		return err
+	}
+
+	if "" == sh.PIDFilePath {
+		return fmt.Errorf("Container[%s]: The path of pid file must be specified", sh.Name)
+	}
+
+	return nil
 }