seal.go

package main

import (
	"flag"
	"log"
	"os"
	"runtime"
	"seal/conf"
	"sync"
	"time"

	"github.com/calabashdad/utiltools"
)

const sealVersion = "seal: 1.0.0 build-2018052801"

var (
	configFile  = flag.String("c", "./seal.yaml", "configure filename")
	showVersion = flag.Bool("v", false, "show version of seal")
)

var (
	gGuards sync.WaitGroup
)

func init() {
	log.SetFlags(log.Lshortfile | log.Ldate | log.Lmicroseconds)
	flag.Parse()
}

func main() {
	defer func() {
		if err := recover(); err != nil {
			log.Println(utiltools.PanicTrace())
			time.Sleep(1 * time.Second)
		}
	}()

	if len(os.Args) < 2 {
		log.Println("Show usage : ./seal --help.")
		return
	}

	if *showVersion {
		log.Println(sealVersion)
		return
	}

	err := conf.GlobalConfInfo.Loads(*configFile)
	if err != nil {
		log.Println("conf loads failed.err=", err)
		return
	}

	log.Printf("load conf file success, conf=%+v\n", conf.GlobalConfInfo)

	cpuNums := runtime.NumCPU()
	if 0 == conf.GlobalConfInfo.System.CPUNums {
		runtime.GOMAXPROCS(cpuNums)
		log.Println("app run on auto cpu nums=", cpuNums)
	} else {
		runtime.GOMAXPROCS(int(conf.GlobalConfInfo.System.CPUNums))
		log.Println("app run on cpu nums set by config, num=", conf.GlobalConfInfo.System.CPUNums)
	}

	gGuards.Add(1)
	if true {
		rtmpSrv := rtmpServer{}
		go rtmpSrv.Start()
	}

	gGuards.Add(1)
	if true {
		hlsSrv := hlsServer{}
		go hlsSrv.Start()
	}

	gGuards.Wait()
	log.Println("seal quit gracefully.")
}