Go语言中如何将数据库查询结果转换为自定义分组结构体？-Golang-PHP中文网

Go语言中如何将数据库查询结果转换为自定义分组结构体？

霞舞

发布： 2025-03-04 08:38:00

原创

544人浏览过

go语言数据库查询结果转换为自定义分组结构体详解

本文将详细讲解如何将MySQL数据库查询结果转换为Go语言中预定义的自定义分组结构体。核心在于高效处理数据分组和结构转换。

假设MySQL数据库查询结果如下：

{id:1,username:'xx',age:xx,frist_name:"a"},
{id:2,username:'xx',age:xx,frist_name:"b"},
{id:3,username:'xx',age:xx,frist_name:"a"},

登录后复制

目标是将其转换为以下Go结构体：

type myLists struct {
    FristName string   `json:"frist_name"`
    Lists     []userLists `json:"lists"`
}

type userLists struct {
    Id       string `json:"id"`
    Username string `json:"username"`
    Age      string `json:"age"`
}

登录后复制

期望的JSON输出格式：

立即学习“go语言免费学习笔记（深入）”；

[
  {
    "frist_name": "a",
    "lists": [
      {
        "id": "1",
        "username": "xxx",
        "age": "xxx"
      },
      {
        "id": "3",
        "username": "xxx",
        "age": "xxx"
      }
    ]
  },
  {
    "frist_name": "b",
    "lists": [
      {
        "id": "2",
        "username": "xxx",
        "age": "xxx"
      }
    ]
  }
]

登录后复制

实现步骤：

定义中间结构体: 方便解析JSON数据。

type r struct {
    Id        uint64 `json:"id"`
    Username  string `json:"username"`
    Age       string `json:"age"`
    FristName string `json:"frist_name"`
}

登录后复制

编写数据转换函数:

package main

import (
    "encoding/json"
    "fmt"
)

// ... (myLists and userLists definitions as above) ...

func transformData(data []r) ([]myLists, error) {
    result := make(map[string]myLists)
    for _, item := range data {
        key := item.FristName
        if _, ok := result[key]; !ok {
            result[key] = myLists{FristName: key}
        }
        result[key].Lists = append(result[key].Lists, userLists{Id: fmt.Sprint(item.Id), Username: item.Username, Age: item.Age})
    }

    var finalResult []myLists
    for _, v := range result {
        finalResult = append(finalResult, v)
    }
    return finalResult, nil
}

func main() {
    rss := []string{
        `{"id":1,"username":"xx","age":"xx","frist_name":"a"}`,
        `{"id":2,"username":"xx","age":"xx","frist_name":"b"}`,
        `{"id":3,"username":"xx","age":"xx","frist_name":"a"}`,
    }

    var data []r
    for _, rs := range rss {
        var t r
        err := json.Unmarshal([]byte(rs), &t)
        if err != nil {
            fmt.Println("Unmarshal error:", err)
            return
        }
        data = append(data, t)
    }

    transformedData, err := transformData(data)
    if err != nil {
        fmt.Println("Transform error:", err)
        return
    }

    jsonData, err := json.MarshalIndent(transformedData, "", "  ")
    if err != nil {
        fmt.Println("Marshal error:", err)
        return
    }
    fmt.Println(string(jsonData))
}

登录后复制

这段代码首先使用map进行分组，避免了原代码中冗余的循环查找。 transformData 函数将中间结构体 r 的 slice 转换为 myLists 的 slice。main 函数模拟数据库查询结果，并调用 transformData 函数进行转换，最后将结果转换为 JSON 格式输出。请将 rss 替换为实际的数据库查询代码。记住处理潜在的错误，例如数据库连接错误和JSON解析错误。 fmt.Sprint(item.Id) 将 uint64 类型的 ID 转换为字符串，与目标 JSON 结构匹配。

Go语言中如何将数据库查询结果转换为自定义分组结构体？