syncbase: sb51: Print confirmation on success of make-demo command.
Change-Id: I37e5100618cd67ff49f6d8f36131523e8321e498
diff --git a/cmd/sb51/shell.go b/cmd/sb51/shell.go
index fb28d3c..e97419a 100644
--- a/cmd/sb51/shell.go
+++ b/cmd/sb51/shell.go
@@ -82,7 +82,7 @@
}
if flagMakeDemoTables {
- if err := makeDemoDB(ctx, d); err != nil {
+ if err := makeDemoDB(ctx, env.Stdout, d); err != nil {
return err
}
}
@@ -122,7 +122,7 @@
err = dumpDB(ctx, env.Stdout, d)
case "make-demo":
// TODO(jkline): add an "Are you sure prompt" to give the user a 2nd chance.
- err = makeDemoDB(ctx, d)
+ err = makeDemoDB(ctx, env.Stdout, d)
case "select":
err = queryExec(ctx, env.Stdout, d, q)
default:
@@ -191,8 +191,10 @@
return nil
}
-func makeDemoDB(ctx *context.T, d nosql.Database) error {
- if err := demodb.PopulateDemoDB(ctx, d); err != nil {
+func makeDemoDB(ctx *context.T, w io.Writer, d nosql.Database) error {
+ if err := demodb.PopulateDemoDB(ctx, d); err == nil {
+ fmt.Fprintln(w, "Demo tables created and populated.")
+ } else {
return fmt.Errorf("failed making demo tables: %v", err)
}
return nil
