package main

import (
	"flag"
	"github.com/Runner-Go-Team/RunnerGo-collector-open/internal"
	"github.com/Runner-Go-Team/RunnerGo-collector-open/internal/pkg/conf"
	"github.com/Runner-Go-Team/RunnerGo-collector-open/internal/pkg/dal/redis"
	log2 "github.com/Runner-Go-Team/RunnerGo-collector-open/internal/pkg/log"
	"github.com/Runner-Go-Team/RunnerGo-collector-open/internal/pkg/server"
	"github.com/pyroscope-io/client/pyroscope"
	"net/http"
	"os"
	"os/signal"
	"runtime"
	"syscall"
	"time"
)

var mode int
var configFile string

func main() {

	flag.IntVar(&mode, "m", 0, "读取环境变量还是读取配置文件")
	flag.StringVar(&configFile, "c", "./open.yaml", "配置文件")
	if !flag.Parsed() {
		flag.Parse()
	}
	internal.InitProjects(mode, configFile)

	runtime.GOMAXPROCS(runtime.NumCPU())

	// 性能分析
	pyroscope.Start(
		pyroscope.Config{
			ApplicationName: "RunnerGo-engine-open",
			ServerAddress:   "http://192.168.1.205:4040/",
			//Logger:          pyroscope.StandardLogger,
			ProfileTypes: []pyroscope.ProfileType{
				pyroscope.ProfileCPU,
				pyroscope.ProfileAllocObjects,
				pyroscope.ProfileAllocSpace,
				pyroscope.ProfileInuseObjects,
				pyroscope.ProfileInuseSpace,
			},
		})
	// 发送心跳
	go redis.SendHeartBeatRedis(conf.Collector, conf.Duration)

	if mode != 0 {
		// 检查kafka是否启动
		kafkaAddress := os.Getenv("RG_KAFKA_ADDRESS")
		if kafkaAddress == "" {
			kafkaAddress = "kafka:9092"
		}
		time.Sleep(60 * time.Second)
		// docker版本，删除上次启动是的
		redis.ExitStressBelongPartition(conf.StressBelongPartition, conf.Collector)
	}

	collectorService := &http.Server{
		Addr: conf.Conf.Http.Host,
	}

	go server.Execute(conf.Conf.Kafka.Host)

	go func() {
		if err := collectorService.ListenAndServe(); err != nil {
			log2.Logger.Error("collector:", err)
			return
		}
	}()

	/// 接收终止信号
	quit := make(chan os.Signal)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit
	log2.Logger.Info("注销成功")

}