Open-IM-Server/pkg/common/http/http_resp.go

package http

import (
	"Open_IM/pkg/common/constant"
	"fmt"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"

	//"Open_IM/pkg/cms_api_struct"
	"net/http"

	"github.com/gin-gonic/gin"
)

type BaseResp struct {
	Code   int32       `json:"code"`
	ErrMsg string      `json:"err_msg"`
	Data   interface{} `json:"data"`
}

func RespHttp200(ctx *gin.Context, err error, data interface{}) {
	var resp BaseResp
	switch e := err.(type) {
	case constant.ErrInfo:
		resp.Code = e.ErrCode
		resp.ErrMsg = e.ErrMsg
	default:
		s, ok := status.FromError(err)
		if !ok {
			fmt.Println("need grpc format error")
			return
		}
		resp.Code = int32(s.Code())
		resp.ErrMsg = s.Message()
	}
	resp.Data = data
	ctx.JSON(http.StatusOK, resp)
}

// warp error
func WrapError(err constant.ErrInfo, msg ...string) error {
	return status.Error(codes.Code(err.ErrCode), err.ErrMsg+msg[0])
}