func init() {
// try to resolve rancher-metadata before going further
// the resolution indicates that the network has been set
_, err := metadata.NewMetadataClient()
if err != nil {
logrus.Fatalf("Failed to configure rancher-metadata client: %v", err)
}
}
func setEnv() {
flag.Parse()
provider = providers.GetProvider(*providerName)
if *debug {
logrus.SetLevel(logrus.DebugLevel)
}
if *logFile != "" {
if output, err := os.OpenFile(*logFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666); err != nil {
logrus.Fatalf("Failed to log to file %s: %v", *logFile, err)
} else {
logrus.SetOutput(output)
formatter := &logrus.TextFormatter{
FullTimestamp: true,
}
logrus.SetFormatter(formatter)
}
}
// configure metadata client
mClient, err := metadata.NewMetadataClient()
if err != nil {
logrus.Fatalf("Failed to configure rancher-metadata client: %v", err)
}
m = mClient
cattleUrl := os.Getenv("CATTLE_URL")
if len(cattleUrl) == 0 {
logrus.Fatalf("CATTLE_URL is not set")
}
cattleApiKey := os.Getenv("CATTLE_ACCESS_KEY")
if len(cattleApiKey) == 0 {
logrus.Fatalf("CATTLE_ACCESS_KEY is not set")
}
cattleSecretKey := os.Getenv("CATTLE_SECRET_KEY")
if len(cattleSecretKey) == 0 {
logrus.Fatalf("CATTLE_SECRET_KEY is not set")
}
//configure cattle client
c, err = NewCattleClient(cattleUrl, cattleApiKey, cattleSecretKey)
if err != nil {
logrus.Fatalf("Failed to configure cattle client: %v", err)
}
}