1 files changed, 9 insertions, 4 deletions

diff --git a/clover/cloverctl/src/cloverctl/cmd/start_visibility.go b/clover/cloverctl/src/cloverctl/cmd/start_visibility.go
index 18f8aac..bbc25d8 100644
--- a/clover/cloverctl/src/cloverctl/cmd/start_visibility.go
+++ b/clover/cloverctl/src/cloverctl/cmd/start_visibility.go
@@ -9,6 +9,7 @@ package cmd
 
 import (
     "fmt"
+    "os"
     "io/ioutil"
     "gopkg.in/resty.v1"
     "github.com/ghodss/yaml"
@@ -18,7 +19,7 @@ import (
 
 var visibilitystartCmd = &cobra.Command{
     Use:   "visibility",
-    Short: "Start visibility data collection",
+    Short: "Start visibility collector process",
     Long: ``,
     Run: func(cmd *cobra.Command, args []string) {
         startCollector()
@@ -36,23 +37,27 @@ func startCollector() {
     if cloverFile != "" {
         in, err := ioutil.ReadFile(cloverFile)
         if err != nil {
-            panic(err.Error())
+            fmt.Printf("Cannot read file: %v\n", err)
+            os.Exit(1)
         }
         out_json, err := yaml.YAMLToJSON(in)
         message_body = string(out_json)
         if err != nil {
-            panic(err.Error())
+            fmt.Printf("Invalid yaml: %v\n", err)
+            os.Exit(1)
         }
     } else {
         message_body = `{"sample_interval":"10", "t_port":"80", "t_host":"jaeger-query.istio-system"}`
     }
+    checkControllerIP()
     url := controllerIP + "/collector/start"
     resp, err := resty.R().
     SetHeader("Content-Type", "application/json").
     SetBody(message_body).
     Post(url)
     if err != nil {
-        panic(err.Error())
+        fmt.Printf("Cannot connect to controller: %v\n", err)
+        os.Exit(1)
     }
     fmt.Printf("\n%v\n", resp)
 }