package main

import (
	"net/http"
    "flag"
	_ "net/http/pprof"
    "time"
	"taskpubsub/camera"
	"taskpubsub/sdk"
	"taskpubsub/tasktag"
	"taskpubsub/util"
	"basic.com/valib/logger.git"
	"github.com/spf13/viper"
)

var initchan = make(chan bool)
var useShm bool

const (
	configFilePath  = "/opt/vasystem/config/"
	configFileType = "yaml"
	LOGBASEPATH = "/data/disk1/valog/taskpubsub.log"
)

var envirment = flag.String("e", "pro", "")

func init(){
	viper.SetConfigType(configFileType)
	viper.SetConfigName(*envirment)
	viper.AddConfigPath(configFilePath)
	viper.AddConfigPath("./")
	err := viper.ReadInConfig()
	if err != nil {
		logger.Error("viper.ReadInConfig() Error")
		//panic(err)
	}

    var logFile = LOGBASEPATH
	if viper.GetString("LogBasePath") != "" {
		logFile = viper.GetString("LogBasePath") + "/taskpubsub.log"
	}

    // 日志初始化
	if viper.IsSet("LogLevel") &&
		viper.GetInt("LogLevel") >= logger.PanicLevel &&
		viper.GetInt("LogLevel") <= logger.DebugLevel {
		logger.Config(logFile, viper.GetInt("LogLevel"))
	}else{
		logger.Config(logFile, logger.DebugLevel)
	}
	var logSaveDays    =    15
    logger.SetSaveDays(logSaveDays)
	logger.Info("loginit success !")
	
	flag.BoolVar(&useShm, "shm", false, "use shm for performance")
}     
        
func main() {
    flag.Parse()
    time.Sleep(time.Second)

	if useShm{
		logger.Info("USE SHARE MEMORY")
	}else{
		logger.Info("USE PIPE")
	}
	// pprof 用于分析性能
	go func() {
		logger.Info(http.ListenAndServe("0.0.0.0:6061", nil))
	}()

	go util.Init(initchan)

	logger.Info("init ok !!!!, start sdk, task, camera init process ....", <-initchan)
	sdk.Init(useShm)        //  获取所有算法id ,建立 sdk 主题， 建立sdk server(send, recv 运行)
	tasktag.Init()   // 获取所有任务，建立任务标签， 在数据进入时， 打标签
	camera.Init(useShm)   //获取cid, taskid, sdkid ,关系
	select {}
}