Merge pull request #18 from k1LoW/fix-style-additional-relation

Change style of additional relation edges / can set relation def
k1LoW · May 30, 2018 · f00e1ae · f00e1ae
2 parents d5aa24b + 16e9238
commit f00e1ae
Show file tree

Hide file tree

Showing 30 changed files with 24 additions and 16 deletions.
diff --git a/output/dot/templates.go b/output/dot/templates.go
@@ -6,24 +6,24 @@ import (
 	"github.com/jessevdk/go-assets"
 )
 
-var _Assets21532ae17ad95976ac467eeaeab81f2bb1d537e4 = "digraph {{ .Schema.Name }} {\n  // Config\n  graph [rankdir=TB, layout=dot, fontname=\"Arial\"];\n  node [shape=record, fontsize=14, margin=0.6, fontname=\"Arial\"];\n  edge [fontsize=10, labelfloat=false, splines=none, fontname=\"Arial\"];\n\n  // Tables\n  {{- range $i, $t := .Schema.Tables }}\n  {{ $t.Name }} [shape=none, label=<<table border=\"0\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ $t.Name }}</font> <font color=\"#666666\">[{{ $t.Type }}]</font></td></tr>\n                 {{- range $ii, $c := $t.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- end }}\n\n  // Relations\n  {{- range $j, $r := .Schema.Relations }}\n  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, taillabel=<<table cellpadding=\"5\" border=\"0\" cellborder=\"0\"><tr><td>{{ $r.Def }}</td></tr></table>>];\n  {{- end }}\n}\n"
-var _Assets5bd148e6149bb9adcdddfcf8cc46d6e3047dbe26 = "digraph {{ .Table.Name }} {\n  // Config\n  graph [rankdir=TB, layout=dot, fontname=\"Arial\"];\n  node [shape=record, fontsize=14, margin=0.6, fontname=\"Arial\"];\n  edge [fontsize=10, labelfloat=false, splines=none, fontname=\"Arial\"];\n\n  // Tables\n  {{ .Table.Name }} [shape=none, label=<<table border=\"3\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ .Table.Name }}</font> <font color=\"#666666\">[{{ .Table.Type }}]</font></td></tr>\n                 {{- range $ii, $c := .Table.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- range $i, $t := .Tables }}\n  {{ $t.Name }} [shape=none, label=<<table border=\"0\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ $t.Name }}</font> <font color=\"#666666\">[{{ $t.Type }}]</font></td></tr>\n                 {{- range $ii, $c := $t.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- end }}\n\n  // Relations\n  {{- range $i, $r := .Relations }}\n  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, taillabel=<<table cellpadding=\"5\" border=\"0\" cellborder=\"0\"><tr><td>{{ $r.Def }}</td></tr></table>>];\n  {{- end }}\n}\n"
+var _Assets21532ae17ad95976ac467eeaeab81f2bb1d537e4 = "digraph {{ .Schema.Name }} {\n  // Config\n  graph [rankdir=TB, layout=dot, fontname=\"Arial\"];\n  node [shape=record, fontsize=14, margin=0.6, fontname=\"Arial\"];\n  edge [fontsize=10, labelfloat=false, splines=none, fontname=\"Arial\"];\n\n  // Tables\n  {{- range $i, $t := .Schema.Tables }}\n  {{ $t.Name }} [shape=none, label=<<table border=\"0\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ $t.Name }}</font> <font color=\"#666666\">[{{ $t.Type }}]</font></td></tr>\n                 {{- range $ii, $c := $t.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- end }}\n\n  // Relations\n  {{- range $j, $r := .Schema.Relations }}\n  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, {{ if $r.IsAdditional }}style=\"dashed\",{{ end }} taillabel=<<table cellpadding=\"5\" border=\"0\" cellborder=\"0\"><tr><td>{{ $r.Def }}</td></tr></table>>];\n  {{- end }}\n}\n"
+var _Assets5bd148e6149bb9adcdddfcf8cc46d6e3047dbe26 = "digraph {{ .Table.Name }} {\n  // Config\n  graph [rankdir=TB, layout=dot, fontname=\"Arial\"];\n  node [shape=record, fontsize=14, margin=0.6, fontname=\"Arial\"];\n  edge [fontsize=10, labelfloat=false, splines=none, fontname=\"Arial\"];\n\n  // Tables\n  {{ .Table.Name }} [shape=none, label=<<table border=\"3\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ .Table.Name }}</font> <font color=\"#666666\">[{{ .Table.Type }}]</font></td></tr>\n                 {{- range $ii, $c := .Table.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- range $i, $t := .Tables }}\n  {{ $t.Name }} [shape=none, label=<<table border=\"0\" cellborder=\"1\" cellspacing=\"0\" cellpadding=\"6\">\n                 <tr><td bgcolor=\"#EFEFEF\"><font face=\"Arial Bold\" point-size=\"18\">{{ $t.Name }}</font> <font color=\"#666666\">[{{ $t.Type }}]</font></td></tr>\n                 {{- range $ii, $c := $t.Columns }}\n                 <tr><td port=\"{{ $c.Name }}\" align=\"left\">{{ $c.Name }} <font color=\"#666666\">[{{ $c.Type }}]</font></td></tr>\n                 {{- end }}\n              </table>>];\n  {{- end }}\n\n  // Relations\n  {{- range $i, $r := .Relations }}\n  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, {{ if $r.IsAdditional }}style =\"dashed\",{{ end }} taillabel=<<table cellpadding=\"5\" border=\"0\" cellborder=\"0\"><tr><td>{{ $r.Def }}</td></tr></table>>];\n  {{- end }}\n}\n"
 
 // Assets returns go-assets FileSystem
 var Assets = assets.NewFileSystem(map[string][]string{"/": []string{"schema.dot.tmpl", "table.dot.tmpl"}}, map[string]*assets.File{
-	"/table.dot.tmpl": &assets.File{
-		Path:     "/table.dot.tmpl",
-		FileMode: 0x1a4,
-		Mtime:    time.Unix(1527595732, 1527595732000000000),
-		Data:     []byte(_Assets5bd148e6149bb9adcdddfcf8cc46d6e3047dbe26),
-	}, "/": &assets.File{
+	"/": &assets.File{
 		Path:     "/",
 		FileMode: 0x800001ed,
-		Mtime:    time.Unix(1527595732, 1527595732000000000),
+		Mtime:    time.Unix(1527686637, 1527686637000000000),
 		Data:     nil,
 	}, "/schema.dot.tmpl": &assets.File{
 		Path:     "/schema.dot.tmpl",
 		FileMode: 0x1a4,
-		Mtime:    time.Unix(1527512634, 1527512634000000000),
+		Mtime:    time.Unix(1527686637, 1527686637000000000),
 		Data:     []byte(_Assets21532ae17ad95976ac467eeaeab81f2bb1d537e4),
+	}, "/table.dot.tmpl": &assets.File{
+		Path:     "/table.dot.tmpl",
+		FileMode: 0x1a4,
+		Mtime:    time.Unix(1527685972, 1527685972000000000),
+		Data:     []byte(_Assets5bd148e6149bb9adcdddfcf8cc46d6e3047dbe26),
 	}}, "")
diff --git a/output/dot/templates/schema.dot.tmpl b/output/dot/templates/schema.dot.tmpl
@@ -16,6 +16,6 @@ digraph {{ .Schema.Name }} {
 
   // Relations
   {{- range $j, $r := .Schema.Relations }}
-  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, taillabel=<<table cellpadding="5" border="0" cellborder="0"><tr><td>{{ $r.Def }}</td></tr></table>>];
+  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, {{ if $r.IsAdditional }}style="dashed",{{ end }} taillabel=<<table cellpadding="5" border="0" cellborder="0"><tr><td>{{ $r.Def }}</td></tr></table>>];
   {{- end }}
 }
diff --git a/output/dot/templates/table.dot.tmpl b/output/dot/templates/table.dot.tmpl
@@ -22,6 +22,6 @@ digraph {{ .Table.Name }} {
 
   // Relations
   {{- range $i, $r := .Relations }}
-  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, taillabel=<<table cellpadding="5" border="0" cellborder="0"><tr><td>{{ $r.Def }}</td></tr></table>>];
+  {{ $r.Table.Name }}:{{ $c := index $r.Columns 0 }}{{ $c.Name }} -> {{ $r.ParentTable.Name }}:{{ $pc := index $r.ParentColumns 0 }}{{ $pc.Name }} [dir=back, arrowtail=crow, {{ if $r.IsAdditional }}style ="dashed",{{ end }} taillabel=<<table cellpadding="5" border="0" cellborder="0"><tr><td>{{ $r.Def }}</td></tr></table>>];
   {{- end }}
 }
diff --git a/output/md/templates.go b/output/md/templates.go
@@ -14,16 +14,16 @@ var Assets = assets.NewFileSystem(map[string][]string{"/": []string{"index.md.tm
 	"/": &assets.File{
 		Path:     "/",
 		FileMode: 0x800001ed,
-		Mtime:    time.Unix(1527597737, 1527597737000000000),
+		Mtime:    time.Unix(1527684406, 1527684406000000000),
 		Data:     nil,
 	}, "/index.md.tmpl": &assets.File{
 		Path:     "/index.md.tmpl",
 		FileMode: 0x1a4,
-		Mtime:    time.Unix(1527597737, 1527597737000000000),
+		Mtime:    time.Unix(1527684406, 1527684406000000000),
 		Data:     []byte(_Assets43889384df1c6f74d764c29d91b9d5637eb46061),
 	}, "/table.md.tmpl": &assets.File{
 		Path:     "/table.md.tmpl",
 		FileMode: 0x1a4,
-		Mtime:    time.Unix(1527597735, 1527597735000000000),
+		Mtime:    time.Unix(1527684406, 1527684406000000000),
 		Data:     []byte(_Assetsac44302fb6150a621aa9d04a0350aac972bf7e18),
 	}}, "")
diff --git a/sample/mysql/CamelizeTable.png b/sample/mysql/CamelizeTable.png
diff --git a/sample/mysql/comment_stars.png b/sample/mysql/comment_stars.png
diff --git a/sample/mysql/comments.png b/sample/mysql/comments.png
diff --git a/sample/mysql/logs.png b/sample/mysql/logs.png
diff --git a/sample/mysql/post_comments.png b/sample/mysql/post_comments.png
diff --git a/sample/mysql/posts.png b/sample/mysql/posts.png
diff --git a/sample/mysql/schema.png b/sample/mysql/schema.png
diff --git a/sample/mysql/users.png b/sample/mysql/users.png
diff --git a/sample/mysql8/CamelizeTable.png b/sample/mysql8/CamelizeTable.png
diff --git a/sample/mysql8/comment_stars.png b/sample/mysql8/comment_stars.png
diff --git a/sample/mysql8/comments.png b/sample/mysql8/comments.png
diff --git a/sample/mysql8/logs.png b/sample/mysql8/logs.png
diff --git a/sample/mysql8/post_comments.png b/sample/mysql8/post_comments.png
diff --git a/sample/mysql8/posts.png b/sample/mysql8/posts.png
diff --git a/sample/mysql8/schema.png b/sample/mysql8/schema.png
diff --git a/sample/mysql8/users.png b/sample/mysql8/users.png
diff --git a/sample/postgres/CamelizeTable.png b/sample/postgres/CamelizeTable.png
diff --git a/sample/postgres/comment_stars.png b/sample/postgres/comment_stars.png
diff --git a/sample/postgres/comments.png b/sample/postgres/comments.png
diff --git a/sample/postgres/logs.png b/sample/postgres/logs.png
diff --git a/sample/postgres/post_comments.png b/sample/postgres/post_comments.png
diff --git a/sample/postgres/posts.png b/sample/postgres/posts.png
diff --git a/sample/postgres/schema.png b/sample/postgres/schema.png
diff --git a/sample/postgres/users.png b/sample/postgres/users.png
diff --git a/schema/schema.go b/schema/schema.go
@@ -50,6 +50,7 @@ type Relation struct {
 	ParentTable   *Table
 	ParentColumns []*Column
 	Def           string
+	IsAdditional  bool
 }
 
 // Schema is the struct for database schema
@@ -70,6 +71,7 @@ type AdditionalRelation struct {
 	Columns       []string `yaml:"columns"`
 	ParentTable   string   `yaml:"parentTable"`
 	ParentColumns []string `yaml:"parentColumns"`
+	Def           string   `yaml:"def"`
 }
 
 // FindTableByName find table by table name
@@ -142,7 +144,12 @@ func (s *Schema) LoadAdditionalRelations(path string) error {
 
 	for _, r := range data.Relations {
 		relation := &Relation{
-			Def: "Additional Relation",
+			IsAdditional: true,
+		}
+		if r.Def != "" {
+			relation.Def = r.Def
+		} else {
+			relation.Def = "Additional Relation"
 		}
 		relation.Table, err = s.FindTableByName(r.Table)
 		if err != nil {

diff --git a/test/relations.yml b/test/relations.yml
@@ -7,6 +7,7 @@ relations:
     parentTable: users
     parentColumns:
       - id
+    def: logs to users
   -
     table: logs
     columns: