Golang:将Sql结果集存储在嵌套结构中,如果为空则省略

时间:2015-11-01 11:04:03

标签: mysql go

我正在尝试将SQL查询数据集存储在结构中并以JSON格式显示。我差不多完成了。现在的问题是,如果嵌套结构的结果集为空,那么我不想显示它。

同样的问题specified here但是在扫描时使用指针以恐慌结束可能是因为我正在使用&user.Profile.Firstname

  

2015/11/01 16:42:16恐慌恢复 - >运行时错误:无效的内存地址或无指针解除引用

如果我删除指针,那么一切正常,只剩下空字段。我很困惑如何实现这一点。

package main

import (
    "database/sql"
    "github.com/gin-gonic/gin"
    _ "github.com/go-sql-driver/mysql"
    "log"
)

type User struct {
    Id       int64
    Username string
    Email    string
    Profile  Profile `json:",omitempty"`
}

type Profile struct {
    Id        int64  `json:",omitempty"`
    UserId    int64  `json:",omitempty"`
    Firstname *string `json:",omitempty"`
    Lastname  *string `json:",omitempty"`
}

var DB *sql.DB

func checkErr(err error, msg string) {
    if err != nil {
        log.Fatal(msg, err)
    }
}

func main() {
    DB, _ = sql.Open("mysql", "username:secrect@/database")
    defer DB.Close()

    r := gin.Default()
    v1 := r.Group("api/v1/")
    {
        v1.GET("users", GetUsers)
    }
    r.Run(":8080")
}


func GetUsers(c *gin.Context) {
    stmt, err := DB.Query("Select users.id, username , email , firstname , lastname from users left join profiles on users.id = profiles.user_id ")
    if err != nil {
        panic(err.Error())
    }
    defer stmt.Close()

    users := []User{}

    for stmt.Next() {
        var user User
        err := stmt.Scan(&user.Id, &user.Username, &user.Email, &user.Profile.Firstname, &user.Profile.Lastname)
        if err != nil {
            panic(err.Error())
        }

        users = append(users, user)
    }

    c.JSON(200, &users)
}

输出:

{
  "Id": 1,
  "Username": "test1",
  "Email": "test1@example.com",
  "Profile": {
    "Firstname": "John",
    "Lastname": "Doe"
  }
},
{
  "Id": 2,
  "Username": "test2",
  "Email": "test2@example.com",
  "Profile": {
  }
},

1 个答案:

答案 0 :(得分:2)

假设gin使用了encoding/json,如果你想省略空Profile,我相信你需要它作为User中的指针:

type User struct {
    ....
    Profile *Profile `json:",omitempty"`
}

请注意documentation

  

" omitempty" option指定如果字段具有空值(定义为false,0,nil指针,nil接口值以及任何空数组,切片,映射或字符串),则应从编码中省略该字段。

它没有说会检测到结构的零值。

这会让您在从数据库中扫描数据时遇到一些问题。您可以声明Profile,扫描它,与零个人资料进行比较,如果有资料,则将其分配给User

var user User
var profile Profile
_ = stmt.Scan(&user.Id, &user.Username, &user.Email, &profile.Firstname, &profile.Lastname)
if profile != Profile{} {
    user.Profile = &profile
}

我怀疑这不是最优雅的解决方案,但我希望它是正确的方向。