send.go

package response

import (
	"net/http"

	"github.com/yoas0bi/micro-toolkit/exception"
	"github.com/yoas0bi/micro-toolkit/http/response"
	"github.com/yoas0bi/micro-toolkit/logger/zap"
)

// Failed use to response error messge
func Failed(w *restful.Response, err error, opts ...response.Option) {
	var (
		errCode  int
		httpCode int
		ns       string
		reason   string
		data     interface{}
		meta     interface{}
	)

	switch t := err.(type) {
	case exception.APIException:
		errCode = t.ErrorCode()
		reason = t.GetReason()
		data = t.GetData()
		meta = t.GetMeta()
		ns = t.GetNamespace()
		httpCode = t.GetHttpCode()
	default:
		errCode = exception.UnKnownException
	}

	if httpCode == 0 {
		httpCode = http.StatusInternalServerError
	}

	resp := response.Data{
		Code:      &errCode,
		Namespace: ns,
		Reason:    reason,
		Message:   err.Error(),
		Data:      data,
		Meta:      meta,
	}

	for _, opt := range opts {
		opt.Apply(&resp)
	}

	err = w.WriteHeaderAndEntity(httpCode, resp)
	if err != nil {
		zap.L().Errorf("send failed response error, %s", err)
	}
}

// Success use to response success data
func Success(w *restful.Response, data interface{}, opts ...response.Option) {
	err := w.WriteEntity(data)

	if err != nil {
		zap.L().Errorf("send success response error, %s", err)
	}
}