Add feature: 新增'-s'选项以支持对数据库、表、字段按字典序排列。#28

Author: voidint

config/config.go

@@ -12,4 +12,5 @@ type Config struct {
 	Viewer   string
 	Output   string
 	Debug    bool
+	Sorted   bool
 }

main.go

@@ -5,7 +5,9 @@ import (
 	"io"
 	"os"
 	"os/user"
+	"sort"
 	"strings"
+	"time"
 
 	"github.com/urfave/cli"
 	"github.com/voidint/tsdump/build"
@@ -67,11 +69,12 @@ COPYRIGHT:
 }
 
 func main() {
+	now := time.Now()
 	app := cli.NewApp()
 	app.Name = "tsdump"
 	app.Usage = "Database table structure dump tool."
 	app.Version = build.Version()
-	app.Copyright = "Copyright (c) 2017-2020, voidint. All rights reserved."
+	app.Copyright = fmt.Sprintf("Copyright (c) 2017-%d, voidint. All rights reserved.", now.Year())
 	app.Authors = []cli.Author{
 		cli.Author{
 			Name:  "voidint",
@@ -127,6 +130,11 @@ func main() {
 			Usage:       "write to a file, instead of STDOUT",
 			Destination: &c.Output,
 		},
+		cli.BoolFlag{
+			Name:        "s, sorted",
+			Usage:       "sort table columns",
+			Destination: &c.Sorted,
+		},
 	}
 
 	app.Before = func(ctx *cli.Context) (err error) {
@@ -155,6 +163,10 @@ func main() {
 			return cli.NewExitError(fmt.Sprintf("[tsdump] %s", err.Error()), 1)
 		}
 
+		if c.Sorted {
+			sortedDBs(dbs)
+		}
+
 		if c.Output != "" {
 			var f *os.File
 			if f, err = os.Create(c.Output); err != nil {
@@ -231,3 +243,28 @@ func getMetadata(repo model.IRepo, db string, tables ...string) (dbs []model.DB,
 	}
 	return dbs, nil
 }
+
+func sortedDBs(dbs []model.DB) {
+	for i := range dbs {
+		sortedTables(dbs[i].Tables)
+	}
+
+	sort.Slice(dbs, func(i, j int) bool {
+		return dbs[i].Name < dbs[j].Name
+	})
+}
+
+func sortedTables(tables []model.Table) {
+	for i := range tables {
+		sortedColumns(tables[i].Columns)
+	}
+	sort.Slice(tables, func(i, j int) bool {
+		return tables[i].Name < tables[j].Name
+	})
+}
+
+func sortedColumns(columns []model.Column) {
+	sort.Slice(columns, func(i, j int) bool {
+		return columns[i].Name < columns[j].Name
+	})
+}