Function main

cmd/activator/main.go:45–189

()

)

func main() {
	var (
		port          int
		adminPort     int
		inCluster     bool
		autoscalerURL string
		namespace     string
	)

	flag.IntVar(&port, "port", 8000, "port where the activator server will be exposed")
	flag.IntVar(&adminPort, "admin-port", 15000, "port where the admin server will be exposed")
	flag.BoolVar(&inCluster, "in-cluster", false, "use when autoscaler runs in-cluster")
	flag.StringVar(&autoscalerURL, "autoscaler-url", "", "the URL for the cortex autoscaler endpoint")
	flag.StringVar(&namespace, "namespace", os.Getenv("CORTEX_NAMESPACE"),
		"kubernetes namespace where the cortex APIs are deployed "+
			"(can be set through the CORTEX_NAMESPACE env variable)",
	)
	flag.Parse()

	log := logging.GetLogger()
	defer func() {
		_ = log.Sync()
	}()

	switch {
	case autoscalerURL == "":
		log.Fatal("--autoscaler-url is a required option")
	case namespace == "":
		log.Fatal("--namespace is a required option")
	}

	awsClient, err := aws.New()
	if err != nil {
		exit(log, err)
	}

	_, userID, err := awsClient.CheckCredentials()
	if err != nil {
		exit(log, err)
	}

	telemetryEnabled := strings.ToLower(os.Getenv("CORTEX_TELEMETRY_DISABLE")) != "true"

	err = telemetry.Init(telemetry.Config{
		Enabled: telemetryEnabled,
		UserID:  userID,
		Properties: map[string]string{
			"kind":       userconfig.RealtimeAPIKind.String(),
			"image_type": "activator",
		},
		Environment: "operator",
		LogErrors:   true,
		BackoffMode: telemetry.BackoffDuplicateMessages,
	})
	if err != nil {
		log.Fatalw("failed to initialize telemetry", zap.Error(err))
	}
	defer telemetry.Close()