add flag: --sqldb

shotizam.go

@@ -30,6 +30,7 @@ import (
 var (
 	base    = flag.String("base", "", "base file to diff from; must be in json format")
 	mode    = flag.String("mode", "sql", "output mode; tsv, json, sql, nameinfo")
+	output  = flag.String("o", "", "output filename; default to stdout")
 	sqlite  = flag.Bool("sqlite", false, "launch SQLite on data (when true, mode flag is ignored)")
 	verbose = flag.Bool("verbose", false, "verbose logging of file parsing")
 )
@@ -251,12 +252,19 @@ func main() {
 	}
 
 	var w io.WriteCloser = os.Stdout
+	if !*sqlite && *output != "" && *output != "-" {
+		var err error
+		w, err = os.Create(*output)
+		if err != nil {
+			log.Fatalf("output fail: %s", err)
+		}
+	}
 	switch *mode {
 	case "sql":
 	case "json":
 	case "tsv":
 	case "nameinfo":
-		w = nopWriteCloser()
+
 	default:
 		log.Fatalf("unknown mode %q", *mode)
 	}
@@ -341,6 +349,7 @@ func main() {
 			newm := recMap(recs)
 			recs = diffMap(oldm, newm)
 		}
+		// TODO: output write new or diff?
 		je := json.NewEncoder(w)
 		je.SetIndent("", "\t")
 		if err := je.Encode(recs); err != nil {
@@ -359,8 +368,8 @@ func main() {
 				skip += len(name)
 			}
 		}
-		log.Printf("                          total length of func names: %d", totNames)
-		log.Printf("bytes of func names which are prefixes of other func: %d", skip)
+		fmt.Fprintf(w, "                          total length of func names: %d\n", totNames)
+		fmt.Fprintf(w, "bytes of func names which are prefixes of other func: %d\n", skip)
 		return
 	}
 
@@ -369,6 +378,31 @@ func main() {
 		if err := cmd.Wait(); err != nil {
 			log.Fatal(err)
 		}
+
+		if *output != "" {
+			(func() { // use func for defer close
+				r, err := os.Open(dbPath)
+				if err != nil {
+					log.Fatalf("output fail: %s", err)
+				}
+				defer r.Close()
+
+				var w io.WriteCloser = os.Stdout
+				if *output != "-" {
+					var err error
+					w, err = os.Create(*output)
+					if err != nil {
+						log.Fatalf("output fail: %s", err)
+					}
+				}
+				defer w.Close()
+
+				if _, err := io.Copy(w, r); err != nil {
+					log.Fatalf("output fail: %s", err)
+				}
+			})()
+		}
+
 		if err := syscall.Exec(cmd.Path, cmd.Args, cmd.Env); err != nil {
 			log.Fatal(err)
 		}