serve.go

package cmd

import (
	"os"
	"os/signal"
	"sync"
	"syscall"

	"github.com/kubeshop/tracetest/server/app"
	"github.com/kubeshop/tracetest/server/config"
	"github.com/spf13/cobra"
)

func init() {
	rootCmd.AddCommand(serveCmd)
	serveCmd.Flags().StringVar(&provisioningFile, "provisioning-file", "", "path to a provisioning file")
}

var provisioningFile string

var serveCmd = &cobra.Command{
	Use:   "serve",
	Short: "Start Tracetest server",
	RunE: func(cmd *cobra.Command, args []string) error {
		cfg.Watch(func(updated *config.AppConfig) {
			appInstance.HotReload()
		})

		var wg sync.WaitGroup
		c := make(chan os.Signal)
		signal.Notify(c, os.Interrupt, syscall.SIGTERM)
		go func() {
			<-c
			wg.Done()
			appInstance.Stop()
			os.Exit(1)
		}()

		wg.Add(1)
		err := appInstance.Start(app.WithProvisioningFile(provisioningFile))
		if err != nil {
			return err
		}

		wg.Wait()

		return nil
	},
}