From eb3bed151d94c93646c4a4338ad996b0177d302e Mon Sep 17 00:00:00 2001
From: Radu Ioan Fericean <radu@fericean.ro>
Date: Thu, 2 Jan 2014 22:37:58 +0200
Subject: [PATCH] write pid file from cgr-engine

---
 cmd/cgr-engine/cgr-engine.go | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/cmd/cgr-engine/cgr-engine.go b/cmd/cgr-engine/cgr-engine.go
index 0490cd573..e28b62ea7 100644
--- a/cmd/cgr-engine/cgr-engine.go
+++ b/cmd/cgr-engine/cgr-engine.go
@@ -23,16 +23,19 @@ import (
 	"flag"
 	"fmt"
 	"io"
+	"log"
 	"net"
 	"net/rpc"
 	"net/rpc/jsonrpc"
+	"os"
 	"runtime"
+	"strconv"
 	"time"
 
 	"github.com/cgrates/cgrates/apier/v1"
 	"github.com/cgrates/cgrates/balancer2go"
-	"github.com/cgrates/cgrates/cdrs"
 	"github.com/cgrates/cgrates/cdrc"
+	"github.com/cgrates/cgrates/cdrs"
 	"github.com/cgrates/cgrates/config"
 	"github.com/cgrates/cgrates/engine"
 	"github.com/cgrates/cgrates/history"
@@ -60,8 +63,9 @@ var (
 	version      = flag.Bool("version", false, "Prints the application version.")
 	raterEnabled = flag.Bool("rater", false, "Enforce starting of the rater daemon overwriting config")
 	schedEnabled = flag.Bool("scheduler", false, "Enforce starting of the scheduler daemon overwriting config")
-	cdrsEnabled = flag.Bool("cdrs", false, "Enforce starting of the cdrs daemon overwriting config")
-	cdrcEnabled = flag.Bool("cdrc", false, "Enforce starting of the cdrc service overwriting config")
+	cdrsEnabled  = flag.Bool("cdrs", false, "Enforce starting of the cdrs daemon overwriting config")
+	cdrcEnabled  = flag.Bool("cdrc", false, "Enforce starting of the cdrc service overwriting config")
+	writepid     = flag.Bool("writepid", false, "Write pid file")
 	bal          = balancer2go.NewBalancer()
 	exitChan     = make(chan bool)
 	sm           sessionmanager.SessionManager
@@ -303,12 +307,26 @@ func checkConfigSanity() error {
 	return nil
 }
 
+func writePid() {
+	f, err := os.Create("/var/run/cgr-engine.pid")
+	if err != nil {
+		log.Fatal("Could not write pid file: ", err)
+	}
+	f.WriteString(strconv.Itoa(os.Getpid()))
+	if err := f.Close(); err != nil {
+		log.Fatal("Could not write pid file: ", err)
+	}
+}
+
 func main() {
 	flag.Parse()
 	if *version {
 		fmt.Println("CGRateS " + utils.VERSION)
 		return
 	}
+	if *writepid {
+		writePid()
+	}
 	runtime.GOMAXPROCS(runtime.NumCPU())
 
 	cfg, err = config.NewCGRConfig(cfgPath)
@@ -324,7 +342,7 @@ func main() {
 		return
 	}
 	var ratingDb engine.RatingStorage
-	var accountDb  engine.AccountingStorage
+	var accountDb engine.AccountingStorage
 	var logDb engine.LogStorage
 	var loadDb engine.LoadStorage
 	var cdrDb engine.CdrStorage