将 JSON.RawMessage 转换为 JSON

Question

我正在使用 gqlgen、sqlx 和 pgx。我正在尝试为 sqlx 的 types.JSONText.

使用自定义标量

我在项目 table 中有此属性 jsonb 字段。

-- migrations/001_up.sql

CREATE TABLE IF NOT EXISTS items (
    id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
    quantity INT NOT NULL,
    attributes JSONB
);

我有这些模型结构：

// graph/model/item.go

type Item struct {
    ID         string       `json:"id,omitempty" db:"id,omitempty"`
    Quantity   int          `json:"quantity" db:"quantity"`
    Attributes *Attributes  `json:"attributes,omitempty" db:"attributes,omitempty"`
}

type Attributes types.JSONText

我有这个 graphql 模式：

// graph/schema.graphql

type Item {
  id: ID!
  quantity: Int!
  attributes: Attributes
}
 
scalar Attributes

我可以成功插入数据库，但在检索时出错。

| id            | quantity | attributes                         |
|---------------|----------|------------------------------------|
| 031e1489-...  | 100      | {"size": "medium", "color": "red"} |

这是我从数据库查询得到的日志：

>> items.Db: &{
  031e1489-02c9-46d3-924d-6a2edf1ca3ba // id
  100                                  // quantity
  0xc000430600                         // attributes
}

我尝试编组属性标量：

// graph/model/item.go

...

func (a *Attributes) MarshalGQL(w io.Writer) {
    b, _ := json.Marshal(a)
    w.Write(b)
}

// Unmarshal here
...

在 gqlgen.yml 中添加自定义标量类型：

...

  Attributes:
    model:
      - github.com/my-api/graph/model.Attributes

但我得到的是字符串而不是 json:

{
  "data": {
    "item": {
      "id": "031e1489-02c9-46d3-924d-6a2edf1ca3ba",
      "quantity": 100,
      "attributes": "eyJjb2xvciI6ICJyZWQifQ==",
    }
  }
}

期望的输出是：

{
  "data": {
    "item": {
      "id": "031e1489-02c9-46d3-924d-6a2edf1ca3ba",
      "quantity": 100,
      "attributes": {
        "size": "medium",
        "color": "red",
      }
    }
  }
}

我做错了什么？

这是我的尝试：

如果我从 Item 结构中的属性中删除指针，gqlgen 会抛出错误：

go generate ./...
go: finding module for package github.com/my-api/graph/generated
generating core failed: type.gotpl: template: type.gotpl:49:28: executing "type.gotpl" at <$type.Elem.GO>: nil pointer evaluating *config.TypeReference.GOexit status 1
graph/resolver.go:3: running "go": exit status 1
make: *** [Makefile:2: gengql] Error 1

这是 returns 想要的结果，但我不知道如何将其用于实际数据：

func (a *Attributes) MarshalGQL(w io.Writer) {
    raw := json.RawMessage(`{"foo":"bar"}`)
    j, _ := json.Marshal(&raw)
    s := string(j)

    w.Write([]byte(s))
}

查询结果：

{
  "data": {
    "item": {
      "id": "031e1489-02c9-46d3-924d-6a2edf1ca3ba",
      "quantity": 100,
      "attributes": {
        "foo": "bar"
      }
    }
  }
}

Answer 1

Attributes 使用 types.JSONText which is defined using json.RawMessage 定义，使用 []byte 定义。这意味着underlying类型的所有4种类型，包括[]byte、是[]byte，这又意味着所有 4 种类型都可以转换为 []byte.

因此，这样做就足够了：

func (a *Attributes) MarshalGQL(w io.Writer) {
    w.Write([]byte(*a))
}

将 JSON.RawMessage 转换为 JSON

Convert JSON.RawMessage to JSON

postgresql

go

graphql

gqlgen