Support create pid file.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

config/config.go

@@ -18,15 +18,16 @@ type ConfYaml struct {
 
 // SectionCore is sub seciont of config.
 type SectionCore struct {
-	Port            string `yaml:"port"`
-	MaxNotification int    `yaml:"max_notification"`
-	WorkerNum       int    `yaml:"worker_num"`
-	QueueNum        int    `yaml:"queue_num"`
-	Mode            string `yaml:"mode"`
-	SSL             bool   `yaml:"ssl"`
-	CertPath        string `yaml:"cert_path"`
-	KeyPath         string `yaml:"key_path"`
-	HTTPProxy       string `yaml:"http_proxy"`
+	Port            string     `yaml:"port"`
+	MaxNotification int        `yaml:"max_notification"`
+	WorkerNum       int        `yaml:"worker_num"`
+	QueueNum        int        `yaml:"queue_num"`
+	Mode            string     `yaml:"mode"`
+	SSL             bool       `yaml:"ssl"`
+	CertPath        string     `yaml:"cert_path"`
+	KeyPath         string     `yaml:"key_path"`
+	HTTPProxy       string     `yaml:"http_proxy"`
+	PID             SectionPID `yaml:"pid"`
 }
 
 // SectionAPI is sub seciont of config.
@@ -88,6 +89,13 @@ type SectionBuntDB struct {
 	Path string `yaml:"path"`
 }
 
+// SectionPID is sub seciont of config.
+type SectionPID struct {
+	Enabled  bool   `yaml:"enabled"`
+	Path     string `yaml:"path"`
+	Override bool   `yaml:"override"`
+}
+
 // BuildDefaultPushConf is default config setting.
 func BuildDefaultPushConf() ConfYaml {
 	var conf ConfYaml
@@ -102,6 +110,9 @@ func BuildDefaultPushConf() ConfYaml {
 	conf.Core.KeyPath = "key.pem"
 	conf.Core.MaxNotification = 100
 	conf.Core.HTTPProxy = ""
+	conf.Core.PID.Enabled = false
+	conf.Core.PID.Path = "gorush.pid"
+	conf.Core.PID.Override = false
 
 	// Api
 	conf.API.PushURI = "/api/push"

config/config.yml

@@ -8,6 +8,10 @@ core:
   cert_path: "cert.pem"
   key_path: "key.pem"
   http_proxy: ""
+  pid:
+    enabled: false
+    path: "gorush.pid"
+    override: true
 
 api:
   push_uri: "/api/push"

config/config_test.go

@@ -63,6 +63,10 @@ func (suite *ConfigTestSuite) TestValidateConfDefault() {
 	assert.Equal(suite.T(), "key.pem", suite.ConfGorushDefault.Core.KeyPath)
 	assert.Equal(suite.T(), 100, suite.ConfGorushDefault.Core.MaxNotification)
 	assert.Equal(suite.T(), "", suite.ConfGorushDefault.Core.HTTPProxy)
+	// Pid
+	assert.Equal(suite.T(), false, suite.ConfGorushDefault.Core.PID.Enabled)
+	assert.Equal(suite.T(), "gorush.pid", suite.ConfGorushDefault.Core.PID.Path)
+	assert.Equal(suite.T(), false, suite.ConfGorushDefault.Core.PID.Override)
 
 	// Api
 	assert.Equal(suite.T(), "/api/push", suite.ConfGorushDefault.API.PushURI)
@@ -111,6 +115,10 @@ func (suite *ConfigTestSuite) TestValidateConf() {
 	assert.Equal(suite.T(), "key.pem", suite.ConfGorush.Core.KeyPath)
 	assert.Equal(suite.T(), 100, suite.ConfGorush.Core.MaxNotification)
 	assert.Equal(suite.T(), "", suite.ConfGorush.Core.HTTPProxy)
+	// Pid
+	assert.Equal(suite.T(), false, suite.ConfGorush.Core.PID.Enabled)
+	assert.Equal(suite.T(), "gorush.pid", suite.ConfGorush.Core.PID.Path)
+	assert.Equal(suite.T(), true, suite.ConfGorush.Core.PID.Override)
 
 	// Api
 	assert.Equal(suite.T(), "/api/push", suite.ConfGorush.API.PushURI)