Error handling

dave · Oct 27, 2017 · dba25f0 · dba25f0
1 parent bba8f3e
commit dba25f0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/blaster.go b/blaster.go
@@ -11,6 +11,7 @@ import (
 	"time"
 
 	"github.com/leemcloughlin/gofarmhash"
+	"github.com/pkg/errors"
 	"github.com/spf13/viper"
 )
 
@@ -117,6 +118,10 @@ func (b *Blaster) Start(ctx context.Context) error {
 		return err
 	}
 
+	if b.config.Data == "" {
+		return errors.New("No data file specified. Use --config to view current config.")
+	}
+
 	if err := b.openDataFile(ctx); err != nil {
 		return err
 	}

diff --git a/cmd/blast/blast.go b/cmd/blast/blast.go
@@ -6,11 +6,15 @@ import (
 
 	"fmt"
 
+	"os"
+
 	"github.com/dave/blast"
 	"github.com/dave/blast/dummyworker"
 	"github.com/dave/blast/httpworker"
 )
 
+const DEBUG = false
+
 func main() {
 	ctx, cancel := context.WithCancel(context.Background())
 
@@ -21,6 +25,11 @@ func main() {
 	b.RegisterWorkerType("http", httpworker.New)
 
 	if err := b.Start(ctx); err != nil {
-		log.Fatal(fmt.Printf("%+v", err))
+		if DEBUG {
+			log.Fatal(fmt.Printf("%+v", err))
+		} else {
+			fmt.Println(err)
+			os.Exit(1)
+		}
 	}
 }