github.com/castai/kvisor@v1.7.1-0.20240516114728-b3572a2607b5/cmd/event-generator/main.go (about)

     1  package main
     2  
     3  import (
     4  	"time"
     5  
     6  	"github.com/castai/kvisor/pkg/logging"
     7  	"github.com/sirupsen/logrus"
     8  	"github.com/spf13/pflag"
     9  
    10  	"github.com/castai/kvisor/cmd/event-generator/app"
    11  )
    12  
    13  var (
    14  	Version       = "local"
    15  	logLevel      = pflag.String("log-level", logrus.DebugLevel.String(), "log level")
    16  	mode          = pflag.String("mode", "controller", "Run as controller or single event generator")
    17  	thiefDelay    = pflag.Duration("thief-delay", 90*time.Second, "")
    18  	thiefInterval = pflag.Duration("thief-interval", 30*time.Second, "")
    19  )
    20  
    21  func main() {
    22  	pflag.Parse()
    23  
    24  	log := logging.New(&logging.Config{
    25  		Level: logging.MustParseLevel(*logLevel),
    26  	})
    27  
    28  	genapp, err := app.New(&app.Config{
    29  		Version:       Version,
    30  		Log:           log,
    31  		Kubeconfig:    "",
    32  		ThiefDelay:    *thiefDelay,
    33  		ThiefInterval: *thiefInterval,
    34  	})
    35  	if err != nil {
    36  		log.Fatal(err.Error())
    37  	}
    38  	if err := genapp.Run(*mode); err != nil {
    39  		log.Fatal(err.Error())
    40  	}
    41  }