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"
	"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", "./dev.yaml", "配置文件")
	if !flag.Parsed() {
		flag.Parse()
	}
	internal.InitProjects(mode, configFile)

	runtime.GOMAXPROCS(runtime.NumCPU())

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

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

	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("注销成功")

}