/
md.go
80 lines (68 loc) · 1.83 KB
/
md.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package md
import (
"fmt"
"io"
"github.com/olekukonko/tablewriter"
"github.com/voidint/tsdump/model"
"github.com/voidint/tsdump/view"
)
func init() {
view.Register(Name, NewView())
}
const (
// Name 视图名称
Name = "md"
)
// View Markdown视图
type View struct {
}
// NewView 返回Markdown视图实例
func NewView() view.Viewer {
return new(View)
}
// Do 将数据库元数据以Markdown视图形式输出。
func (v *View) Do(items []model.DB, out io.Writer) error {
for i := range items {
v.renderDB(&items[i], out)
fmt.Fprintln(out)
for j := range items[i].Tables {
fmt.Fprintf(out, "### `%s`\n%s\n\n",
items[i].Tables[j].Name,
items[i].Tables[j].Comment,
)
v.renderTable(&items[i].Tables[j], out)
fmt.Fprintln(out)
}
}
return nil
}
func (v *View) renderDB(db *model.DB, out io.Writer) {
rows := [][]string{[]string{db.Name, db.CharSet, db.Collation}}
t := tablewriter.NewWriter(out)
t.SetHeader([]string{"Database", "Character Set", "Collation"})
t.SetBorders(tablewriter.Border{Left: true, Top: false, Right: true, Bottom: false})
t.SetCenterSeparator("|")
t.AppendBulk(rows)
t.Render()
}
func (v *View) renderTable(table *model.Table, out io.Writer) {
rows := make([][]string, 0, len(table.Columns))
for i := range table.Columns {
rows = append(rows, []string{
table.Columns[i].Name,
table.Columns[i].DataType,
table.Columns[i].Nullable,
table.Columns[i].Key,
table.Columns[i].Default,
table.Columns[i].CharSet,
table.Columns[i].Collation,
table.Columns[i].Comment,
})
}
t := tablewriter.NewWriter(out)
t.SetHeader([]string{"Column", "Data Type", "Nullable", "Key", "Default", "Character Set", "Collation", "Comment"})
t.SetBorders(tablewriter.Border{Left: true, Top: false, Right: true, Bottom: false})
t.SetCenterSeparator("|")
t.AppendBulk(rows)
t.Render()
}