Как исправить проблему с функциями UnmarshalJSON и MarshalJSON?

Question

[nurzhannogerbek]

Здравствуйте, товарищи! Помогите пожалуйста исправить проблему.

В базе данных PostgreSQL есть таблица.

CREATE TABLE SURVEYS(
  SURVEY_ID UUID PRIMARY KEY NOT NULL DEFAULT uuid_generate_v4(),
  SURVEY_NAME VARCHAR NOT NULL,
  SURVEY_DESCRIPTION TEXT,
  START_PERIOD TIMESTAMP,
  END_PERIOD TIMESTAMP
);

Как вы можете заметить, последние 3 поля не обязательны для заполнения и могут хранить NULL.

На основе этой таблицы создал структуру в Golang:

type Survey struct {
	ID string `json:"survey_id"`
	Name string `json:"survey_name"`
	Description utils.NullString `json:"survey_description"`
	StartPeriod utils.NullTime `json:"start_period"`
	EndPeriod utils.NullTime `json:"end_period"`
}

В Golang приложении делаю GET запрос к определенной записи:

http://localhost:8000/api/survey/0cf1cf18-d5fd-474e-a8be-754fbdc89720

Получаю странный ответ:

{
    "survey_id": "0cf1cf18-d5fd-474e-a8be-754fbdc89720",
    "survey_name": "NAME",
    "survey_description": {
        "String": "DESCRIPTION",
        "Valid": true
    },
    "start_period": {
        "Time": "2019-01-01T00:00:00Z",
        "Valid": true
    },
    "end_period": {
        "Time": "0001-01-01T00:00:00Z",
        "Valid": false
    }
}

Необходимо получить такой ответ:

{
    "survey_id": "0cf1cf18-d5fd-474e-a8be-754fbdc89720",
    "survey_name": "NAME",
    "survey_description": "DESCRIPTION",
    "start_period": "2019-01-01 00:00:00",
    "end_period": null
}

Как правильно настроить MarshalJSON и MarshalJSON функции?

utils.go:

const layout = "2006-01-02 15:04:05"

type NullTime struct {
	pq.NullTime
}

func (nt *NullTime) Scan(value interface{}) error {
	nt.Time, nt.Valid = value.(time.Time)
	return nil
}

func (nt NullTime) Value() (driver.Value, error) {
	if !nt.Valid {
		return nil, nil
	}
	return nt.Time, nil
}

func (nt *NullTime) MarshalJSON() ([]byte, error) {
	if !nt.Valid {
		return []byte("null"), nil
	}
	val := fmt.Sprintf("\"%s\"", nt.Time.Format(layout))
	return []byte(val), nil
}

func (nt *NullTime) UnmarshalJSON(b []byte) error {
	err := json.Unmarshal(b, &nt.Time)
	nt.Valid = err == nil
	return err
}


type NullString struct {
	sql.NullString
}

func (ns *NullString) MarshalJSON() ([]byte, error) {
	if !ns.Valid {
		return []byte("null"), nil
	}
	return json.Marshal(ns.String)
}

func (ns *NullString) UnmarshalJSON(b []byte) error {
	err := json.Unmarshal(b, &ns.String)
	ns.Valid = err == nil
	return err
}

Answer 1

[Леонид Николаев]

самый просто вариант тупо сделать второй тип, куда конвертните полученный. в нем уже сделать логику.

Comments

[nurzhannogerbek]

Здравствуйте, Леонид! Не совсем уловил вашу мысль. Чтобы беседа была предметной, могли бы вы пожалуйста показать пример?!

[Леонид Николаев]

type SurveyShow struct {
  ID string `json:"survey_id"`
  Name string `json:"survey_name"`
  Description *string `json:"survey_description"`
  StartPeriod *string `json:"start_period"`
  EndPeriod *string `json:"end_period"`
}

func prepareShow(in *Survery) *SurveryShow {
   var ret = SurveryShow{ ID: in.IN, Name: in.Name}
   if in.Description.Valid {
        ret.Description = &in.Description.Value
   }
...

  return &ret
}

[nurzhannogerbek]

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

var GetSurvey = func(responseWriter http.ResponseWriter, request *http.Request) {
	// Initialize variables.
	survey := models.Survey{}
	var err error

	// Gorilla Mux Vars.
	vars := mux.Vars(request)

	// Execute SQL statement.
	err = database.DB.QueryRow("SELECT * FROM surveys WHERE survey_id = $1;", vars["survey_id"]).Scan(&survey.ID, &survey.Name, &survey.Description, &survey.StartPeriod, &survey.EndPeriod)

	// Shape the response depending on the result of the previous command.
	if err != nil {
		log.Println(err)
		switch err {
		case sql.ErrNoRows:
			utils.ResponseWithError(responseWriter, http.StatusNotFound, "The entry not found.")
		default:
			utils.ResponseWithError(responseWriter, http.StatusInternalServerError, err.Error())
		}
		return
	}
	utils.Response(responseWriter, http.StatusOK, survey)
}

Я предполагал, что нужно лишь поменять функцию Scan. Не могу понять как безболезненно добавить вашу функцию.

[Леонид Николаев]

utils.Response(responseWriter, http.StatusOK, prepareShow (&survey))

например так

но, я сразу говорю, что это просто самый простой способ.

[nurzhannogerbek]

Леонид Николаев, Ругается на эту строчку вашего кода:
ret.Description = &in.Description.Value

Если быть точнее на &in.Description.Value.

Описание:

Cannot use '&in.Description.Value' (type *func() (driver.Value, error)) as type utils.NullString in assignment less...
Inspection info: Reports incompatible types in binary and unary expressions.

[Леонид Николаев]

ну да, там нужно String писать, но это уже сами отладите.

Answer 2

[falconandy]

У вас поля структуры - это структуры utils.NullString и utils.NullTime, а MarshalJSON реализован для указателей на эти структуры. Просто уберите * в сигнатуре MarshalJS.

https://stackoverflow.com/questions/39164471/marsh...