Преобразование JSON.RawMessage в JSON

Я использую gqlgen, sqlx и pgx. И я пытаюсь использовать собственный скаляр для sqlxс types.JSONText.

У меня есть эти атрибуты 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",
      }
    }
  }
}

Что я делаю не так?


Вот мои попытки:

Если я удалил указатель из атрибутов в структуре элемента, 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

Это возвращает желаемый результат, но я не знаю, как его использовать с реальными данными:

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"
      }
    }
  }
}

1 ответ

Решение

Attributes определяется с использованием types.JSONText который определяется с помощью json.RawMessage который определяется с помощью []byte. Это означает, что базовый тип всех 4 типов, включая []byte, это []byte, что, в свою очередь, означает, что все 4 типа могут быть преобразованы в []byte.

Следовательно, для этого должно быть достаточно:

func (a *Attributes) MarshalGQL(w io.Writer) {
    w.Write([]byte(*a))
}
Другие вопросы по тегам