naza/pkg/nazahttp/body.go

// Copyright 2020, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package nazahttp

import (
	"encoding/json"
	"errors"
	"io/ioutil"
	"net/http"

	"github.com/q191201771/naza/pkg/nazajson"
)

var ErrParamMissing = errors.New("nazahttp: param missing")

// @param r            http请求对象
// @param info         输出参数，用于接收反序列化之后的数据
// @param keyFieldList 可选参数，可指定一个或多个json中必须存在的字段
func UnmarshalRequestJsonBody(r *http.Request, info interface{}, keyFieldList ...string) error {
	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		return err
	}

	j, err := nazajson.New(body)
	if err != nil {
		return err
	}
	for _, kf := range keyFieldList {
		if !j.Exist(kf) {
			return ErrParamMissing
		}
	}

	return json.Unmarshal(body, info)
}