分析分层数据结构的方法

哈喽！今天心血来潮给大家带来了《分析分层数据结构的方法》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

在oracle中，我有这样的表：

| organization_id | ogranization_name | parent_id | level |
|-----------------|-------------------|-----------|-------|
| 1               | facebook          |           | 0     |
| 2               | instagram         | 1         | 1     |
| 3               | whatsapp          | 1         | 1     |
| 4               | dynamic           | 2         | 2     |
| 5               | google            |           | 0     |

此表显示组织的层次结构。我需要在我的 go 应用程序中创建一个树状嵌套 json 响应：

[
    {
        "organization_id": 1
        "ogranization_name": "facebook",
        "childs": [
            {
                "organization_id": 2,
                "ogranization_name": "instagram",
                "childs": null
            },
            {
                "organization_id": 3,
                "ogranization_name": "whatsapp",
                "childs": [
                    {
                        "organization_id": 4,
                        "ogranization_name": "dynamic",
                        "childs": null
                    }
                ]
            }
        ]
    },
    {
        "organization_id": 5
        "ogranization_name": "google",
        "childs": null
    }
]

我向 oracle 数据库进行查询。我还创建了结构体。我的问题是如何知道一个组织是否有孩子？

ma​​in.go：

type Organisation struct {
    ID       int            `json:"organization_id"`
    Name     string         `json:"ogranization_name"`
    Children []Organisation `json:"childs"`
}

rows,err := db.Query("select * from ORG_TABLE")

if err != nil {
    fmt.Println(err)
    return
}

var organizations []Organization

for rows.Next() {
    var organization Organization

    err = rows.Scan(&organization.ID, &organization.Name, ???)

    if err != nil {
        fmt.Println(err)
        return
    }
}

organizations = append(organizations, organization)

解决方案

最后，我找到了解决方案：

package main

import (
    "database/sql"
    "encoding/json"
    "fmt"
    "log"
    "sort"

    _ "github.com/mattn/go-sqlite3"
)

type Organization struct {
    ID       int             `json:"organization_id"`
    Name     string          `json:"organization_name"`
    Rang     int             `json:"organization_rang"`
    Children []*Organization `json:"children"`
}

func main() {
    db, err := sql.Open("sqlite3", "./database")
    if err != nil {
        log.Fatal(err)
    }

    defer db.Close()

    rows, err := db.Query("select ORGANIZATION_ID,ORGANIZATION_NAME,ORGANIZATION_RANG,PARENT_ORGANIZATION_ID from ORG")
    if err != nil {
        log.Fatal(err)
    }

    defer rows.Close()

    orgs := map[int]*Organization{}

    for rows.Next() {
        organization := &Organization{}
        var parentID sql.NullInt64

        if err = rows.Scan(&organization.ID, &organization.Name, &organization.Rang, &parentID); err != nil {
            log.Fatal(err)
        }

        if parentID.Valid {
            if parentOrg, ok := orgs[int(parentID.Int64)]; ok {
                parentOrg.Children = append(parentOrg.Children, organization)
            } else {
                orgs[int(parentID.Int64)] = &Organization{ID: int(parentID.Int64)}
                orgs[int(parentID.Int64)].Children = append(orgs[int(parentID.Int64)].Children, organization)
            }
        }

        if _, ok := orgs[organization.ID]; ok {
            orgs[organization.ID].Name = organization.Name
            orgs[organization.ID].Rang = organization.Rang
            continue
        }

        orgs[organization.ID] = organization
    }

    IDs := []int{}
    for k := range orgs {
        IDs = append(IDs, k)
    }
    sort.Ints(IDs)

    organizations := []Organization{}
    for _, ID := range IDs {
        if len(orgs[ID].Children) > 0 && orgs[ID].Rang == 1 {
            organizations = append(organizations, *orgs[ID])
        }
    }

    content, err := json.MarshalIndent(organizations, "", "  ")
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(string(content))
}

以上就是《分析分层数据结构的方法》的详细内容，更多关于的资料请关注golang学习网公众号！