Хранение Golang в JSON в PostgreSQL

Я хочу сохранить определенную структуру в моей базе данных,в которой есть поле JSON.

type Comp struct {
    CompId               int64           `db:"comp_id" json:"comp_id"`
    StartDate            time.Time       `db:"start_date" json:"start_date"`
    EndDate              time.Time       `db:"end_date" json:"end_date"`
    WeeklySchedule       json.RawMessage `db:"weekly_schedule" json:"weekly_schedule"`
}

схема для таблицы:

CREATE TABLE IF NOT EXISTS Tr.Comp(
    comp_id                 SERIAL,
    start_date              timestamp NOT NULL,
    end_date                timestamp NOT NULL,
    weekly_schedule         json NOT NULL,
    PRIMARY KEY (comp_id)
);

Я использую драйвер sqlx и lib/pq в своем проекте, и следующее не будет выполняться. Вместо этого он паникует, говоря, что есть нулевой указатель. DB является глобальным *sqlx.DB структуры

    tx := DB.MustBegin()

    compFixture := Comp{
        StartDate:            time.Now(),
        EndDate:              time.Now().AddDate(1, 0, 0),
        WeeklySchedule:       json.RawMessage([]byte("{}")),
    }
    _, err = tx.NamedExec(
        `INSERT INTO 
            Tr.Comp(comp_id, 
                start_date, end_date, weekly_schedule) 
            VALUES (DEFAULT, 
                :start_date, :end_date, :weekly_schedule)  
            RETURNING comp_id;`, compFixture)
    if err != nil {
        t.Fatal("Error creating fixture.", err)
    }

когда я удалить weekly_schedule из схемы и прибор все работает нормально. Но по какой-то причине, когда это поле включено, программа паникует. Любая идея о том, как я должен определить weekly_schedule поле в моей схеме БД и Go struct?

10
go json postgresql sql struct

3 ответов

sqlx имеет тип JSONText на github.com/jmoiron/sqlx/types это сделает то, что вам нужно

doc для JSONText

8

Я не знаю, насколько это чистое решение, но я в конечном итоге сделал свой собственный тип данных JSONRaw. Драйвер DB видит его как []btye но его все еще можно рассматривать как json.RawMessage в коде Go.

type JSONRaw json.RawMessage

func (j JSONRaw) Value() (driver.Value, error) {
    byteArr := []byte(j)

    return driver.Value(byteArr), nil
}

func (j *JSONRaw) Scan(src interface{}) error {
    asBytes, ok := src.([]byte)
    if !ok {
        return error(errors.New("Scan source was not []bytes"))
    }
    err := json.Unmarshal(asBytes, &j)
    if err != nil {
        return error(errors.New("Scan could not unmarshal to []string"))
    }

    return nil
}

func (m *JSONRaw) MarshalJSON() ([]byte, error) {
    return *m, nil
}

func (m *JSONRaw) UnmarshalJSON(data []byte) error {
    if m == nil {
        return errors.New("json.RawMessage: UnmarshalJSON on nil pointer")
    }
    *m = append((*m)[0:0], data...)
    return nil
}

это копипаст перевыполнение MarshalJSON и UnmarshalJSON С encoding/json библиотека.

2

из документации go:

json.RawMessage is a raw encoded JSON object. It implements Marshaler and Unmarshaler and can be used to delay JSON decoding or precompute a JSON encoding.

Если вы войдете.Printf ("%# " , цвета) в примере, приведенном в пакете json json.RawMessage вы можете видеть, что после unmarshalling объект json "Point" - член не является unmarshalled, но остается в формате [] байт, пока цветовой формат не будет исправлен и "Point" не будет unmarshalled явно.

вы пробовали что-то вроде unmarshal WeeklySchedule перед тем, как поместить его в DB?

0