Merge pull request #9 from ZachCheung/master

minor update in README example code

Author: BrandonRoehl

README.md

@@ -6,48 +6,57 @@ Create MYSQL dumps in Go without the `mysqldump` CLI as a dependancy.
 package main
 
 import (
-  "database/sql"
-  "fmt"
+	"database/sql"
+	"fmt"
+	"os"
 
-  "github.com/jamf/go-mysqldump"
-  "github.com/go-sql-driver/mysql"
+	"github.com/go-sql-driver/mysql"
+	"github.com/jamf/go-mysqldump"
 )
 
 func main() {
-  // Open connection to database
-  config := mysql.NewConfig()
-  config.User = "your-user"
-  config.Passwd = "your-pw"
-  config.DBName = "your-db"
-  config.Net = "tcp"
-  config.Addr = "your-hostname:your-port"
-
-  dumpDir := "dumps"  // you should create this directory
-  dumpFilenameFormat := fmt.Sprintf("%s-20060102T150405", config.DBName)   // accepts time layout string and add .sql at the end of file
-
-  db, err := sql.Open("mysql", config.FormatDSN())
-  if err != nil {
-    fmt.Println("Error opening database: ", err)
-    return
-  }
-
-  // Register database with mysqldump
-  dumper, err := mysqldump.Register(db, dumpDir, dumpFilenameFormat)
-  if err != nil {
-    fmt.Println("Error registering databse:", err)
-    return
-  }
-
-  // Dump database to file
-  err := dumper.Dump()
-  if err != nil {
-    fmt.Println("Error dumping:", err)
-    return
-  }
-  fmt.Printf("File is saved to %s", dumpFilenameFormat)
-
-  // Close dumper, connected database and file stream.
-  dumper.Close()
+	// Open connection to database
+	config := mysql.NewConfig()
+	config.User = "your-user"
+	config.Passwd = "your-pw"
+	config.DBName = "your-db"
+	config.Net = "tcp"
+	config.Addr = "your-hostname:your-port"
+
+	dumpDir := "dumps"
+	dumpFilenameFormat := fmt.Sprintf("%s-20060102T150405", config.DBName) // accepts time layout string and add .sql at the end of file
+
+	if err := os.MkdirAll(dumpDir, 0755); err != nil {
+		fmt.Println("Error mkdir:", err)
+		return
+	}
+
+	db, err := sql.Open("mysql", config.FormatDSN())
+	if err != nil {
+		fmt.Println("Error opening database:", err)
+		return
+	}
+
+	// Register database with mysqldump
+	dumper, err := mysqldump.Register(db, dumpDir, dumpFilenameFormat)
+	if err != nil {
+		fmt.Println("Error registering databse:", err)
+		return
+	}
+
+	// Dump database to file
+	if err := dumper.Dump(); err != nil {
+		fmt.Println("Error dumping:", err)
+		return
+	}
+	if file, ok := dumper.Out.(*os.File); ok {
+		fmt.Println("File is saved to", file.Name())
+	} else {
+		fmt.Println("It's not part of *os.File, but dump is done")
+	}
+
+	// Close dumper, connected database and file stream.
+	dumper.Close()
 }
 ```