package response

import (
	"github.com/google/uuid"
	"net/http"
)

// Error 失败数据处理
func Error(code int, err error, msg string) Response {
	var res Response
	if err != nil {
		res.SetMsg(err.Error())
	}
	if msg != "" {
		res.SetMsg(msg)
	}
	res.SetTraceID(uuid.New().String())
	res.SetCode(int32(code))
	res.SetSuccess(false)
	return res
}

// OK 通常成功数据处理
func OK(data interface{}, msg string) Response {
	var res Response
	res.SetSuccess(true)
	if msg != "" {
		res.SetMsg(msg)
	}
	res.SetTraceID(uuid.New().String())
	res.SetCode(http.StatusOK)
	res.Data = data
	return res
}

// PageOK 分页数据处理
func PageOK(result interface{}, count int, pageIndex int, pageSize int, msg string) Page {
	var res Page
	res.SetSuccess(true)
	if msg != "" {
		res.SetMsg(msg)
	}
	res.SetTraceID(uuid.New().String())
	res.SetCode(http.StatusOK)
	res.Data.List = result
	res.Data.Count = count
	res.Data.Page = pageIndex
	res.Data.PageSize = pageSize
	return res
}