扫描 Postgres array_agg

你在学习Golang相关的知识吗？本文《扫描 Postgres array_agg》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

我在 postgres 中有一对多关系（event 有许多 eventuser），并且想要扫描并存储到结构 event 中。

// eventuser struct
type eventuser struct {
    id              int64
    checkedin       bool
    paidamount      float32
}

// event struct
type event struct {
    id         int64  `json:"id"`
    name       string `json:"name"`
    starttime  string `json:"starttime"`
    eventusers string
}

这是查询：

select events.id, events.name, events."starttime", e."eventusers" as "eventusers"
from "events" as events
left join (
        select events.id as id, array_to_json(array_agg(eu.*)) as "eventusers"
        from "eventusers" as eu
        join "events" as "events" on events.id = eu."eventid"
        where eu.status = 'reserved'
        group by events.id
    ) as e using (id)
where events.status = 'completed'

查询返回：

{
  id: 2,
  name: "2 events are 48 days from now",
  starttime: 1590471343345,
  eventusers: [
    {
      id: 2,
      checkedin: false,
      paidamount: 8
    },
    {
      id: 3,
      checkedin: false, 
      paidamount: 8,
    },
  ],
};

这就是我想做的，直接将 eventusers 下的每个项目扫描到结构中，并存储在 event 的结构中。

got := []Event{}

for rows.Next() {
    var r Event
    err = rows.Scan(&r.ID, &r.Name, &r.StartTime, &r.EventUsers)
    if err != nil {
        panic(err)
    }
}

我认为我可以通过将数组存储为字符串然后 unmarshal 来实现此目的。

我想要的是类似于 sql.nullstring 的东西。

解决方案

您可以定义一个实现 sql.Scanner 接口的切片类型。请注意，当您将实现 scanner 的类型的实例传递给 (*sql.rows).scan 或 (*sql.row).scan 调用时，将自动调用执行器的 scan 方法。

type eventuserlist []*eventuser

func (list *eventuserlist) scan(src interface{}) error {
    if data, ok := src.([]byte); ok && len(data) > 0 {
        if err := json.unmarshal(data, list); err != nil {
            return err
        }
    }
    return nil
}

然后，假设 select 查询中的 e."eventusers" 是一个 json 数组，您可以这样使用它：

// EventUser struct
type EventUser struct {
    ID         int64
    CheckedIn  bool
    PaidAmount float32
}

// Event struct
type Event struct {
    ID         int64         `json:"id"`
    Name       string        `json:"name"`
    StartTime  string        `json:"startTime"`
    EventUsers EventUserList `json:"eventUsers"`
}

// ...

var events []*Event
for rows.Next() {
    e := new(Event)
    if err := rows.Scan(&e.ID, &e.Name, &e.StartTime, &e.EventUsers); err != nil {
        return err
    }
    events = append(events, e)
}

本篇关于《扫描 Postgres array_agg》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！