load tags into tree, display them as tree with depth 2, ordered

This commit is contained in:
Dan Ballard 2015-05-08 00:07:30 -07:00
parent 605fc93409
commit 9ca8deed23
2 changed files with 23 additions and 24 deletions

View File

@ -4,47 +4,43 @@ import (
"database/sql" "database/sql"
_ "github.com/lib/pq" _ "github.com/lib/pq"
"fmt" "fmt"
"strings"
"strconv"
) )
type Tag struct { type Tag struct {
Id int
Name string Name string
Parent sql.NullInt64 Parent sql.NullInt64
Children []int Children []*Tag
} }
var Tags map[int]*Tag var Tags []*Tag
func LoadTags(db *sql.DB) { func LoadTags(db *sql.DB) {
Tags = make(map[int]*Tag) Tags = []*Tag{}
rows, err := db.Query("select tags.tag_id, tags.name, tags.parent_id, array_agg(distinct(parent_of.tag_id)) as children from tags left join tags as parent_of on parent_of.parent_id=tags.tag_id group by tags.tag_id") //rows, err := db.Query("select tags.tag_id, tags.name, tags.parent_id, array_agg(distinct(parent_of.tag_id)) as children from tags left join tags as parent_of on parent_of.parent_id=tags.tag_id group by tags.tag_id")
rows, err := db.Query("select tags.tag_id, tags.name, tags.parent_id from tags order by tag_id,tags.parent_id desc")
if err != nil { if err != nil {
fmt.Println("DB Error loading tags:", err) fmt.Println("DB Error loading tags:", err)
return return
} }
var flatMap map[int]*Tag = make(map[int]*Tag)
for rows.Next() { for rows.Next() {
tag := &Tag{Children: nil} tag := &Tag{Children: []*Tag{}}
var children sql.NullString err = rows.Scan(&tag.Id, &tag.Name, &tag.Parent)
var tag_id int
err = rows.Scan(&tag_id, &tag.Name, &tag.Parent, &children)
if err != nil { if err != nil {
fmt.Println("tags DB Error: ", err) fmt.Println("tags DB Error: ", err)
} }
if children.Valid { flatMap[tag.Id] = tag
var childrenStr = children.String[1 : len(children.String)-1]
if childrenStr != "NULL" { if tag.Parent.Valid {
pids := strings.Split(childrenStr, ",") pid := int(tag.Parent.Int64)
for _, spid := range pids { flatMap[pid].Children = append(flatMap[pid].Children, tag)
pid, _ := strconv.Atoi(spid) } else {
tag.Children = append(tag.Children, pid) Tags = append(Tags, tag)
} }
}
}
Tags[tag_id] = tag
} }
} }

View File

@ -21,5 +21,8 @@
{{define "option-tag"}} {{define "option-tag"}}
<option>{{.Name}}</option> <option value="{{.Id}}">{{if .Parent.Valid}}-{{end}}{{.Name}}</option>
{{range $child := .Children}}
{{template "option-tag" $child}}
{{end}}
{{end}} {{end}}