baize/framework/gateway/builder_request.go

package gateway

import (
	"git.sxidc.com/go-framework/baize/framework/core/api"
	"git.sxidc.com/go-tools/utils/http_client"
)

type Request interface {
	Request(c *api.Context, request *http_client.Request) (Request, error)
	Response() *http_client.Response
}

type PostRequest struct {
	Url  string
	Body any

	headers  map[string]string
	response *http_client.Response
}

func (req *PostRequest) Request(c *api.Context, request *http_client.Request) (Request, error) {
	preparedRequest, err := req.prepare(c)
	if err != nil {
		return nil, err
	}

	response, err := request.Post(preparedRequest.Url, preparedRequest.Body,
		http_client.WithRequestHeaders(preparedRequest.headers))
	if err != nil {
		return nil, err
	}

	preparedRequest.response = response

	return preparedRequest, nil
}

func (req *PostRequest) Response() *http_client.Response {
	return req.response
}

func (req *PostRequest) prepare(c *api.Context) (*PostRequest, error) {
	preparedRequest := &PostRequest{
		Url:  req.Url,
		Body: req.Body,
	}

	// 赋值默认body
	if preparedRequest.Body == nil {
		cacheBody, err := c.GetBytesBody()
		if err != nil {
			return nil, err
		}

		preparedRequest.Body = cacheBody.Bytes()
	}

	// 赋值默认headers
	preparedRequest.headers = c.GetHeader().Map()

	return preparedRequest, nil
}

type DeleteRequest struct {
	Url         string
	QueryParams map[string]string
	PathParams  map[string]string

	headers  map[string]string
	response *http_client.Response
}

func (req *DeleteRequest) Request(c *api.Context, request *http_client.Request) (Request, error) {
	preparedRequest, err := req.prepare(c)
	if err != nil {
		return nil, err
	}

	response, err := request.Delete(preparedRequest.Url,
		http_client.WithRequestQueryParams(preparedRequest.QueryParams),
		http_client.WithRequestPathParams(preparedRequest.PathParams),
		http_client.WithRequestHeaders(preparedRequest.headers))
	if err != nil {
		return nil, err
	}

	preparedRequest.response = response

	return preparedRequest, nil
}

func (req *DeleteRequest) Response() *http_client.Response {
	return req.response
}

func (req *DeleteRequest) prepare(c *api.Context) (*DeleteRequest, error) {
	preparedRequest := &DeleteRequest{
		Url:         req.Url,
		QueryParams: req.QueryParams,
		PathParams:  req.PathParams,
	}

	// 赋值默认查询参数
	if preparedRequest.QueryParams == nil || len(preparedRequest.QueryParams) == 0 {
		preparedRequest.QueryParams = c.GetQueryParams().Map()
	}

	// 赋值默认路径参数
	if preparedRequest.PathParams == nil || len(preparedRequest.PathParams) == 0 {
		preparedRequest.PathParams = c.GetPathParams().Map()
	}

	// 赋值默认headers
	preparedRequest.headers = c.GetHeader().Map()

	return preparedRequest, nil
}

type PutRequest struct {
	Url  string
	Body any

	headers  map[string]string
	response *http_client.Response
}

func (req *PutRequest) Request(c *api.Context, request *http_client.Request) (Request, error) {
	preparedRequest, err := req.prepare(c)
	if err != nil {
		return nil, err
	}

	response, err := request.Put(preparedRequest.Url, preparedRequest.Body,
		http_client.WithRequestHeaders(preparedRequest.headers))
	if err != nil {
		return nil, err
	}

	preparedRequest.response = response

	return preparedRequest, nil
}

func (req *PutRequest) Response() *http_client.Response {
	return req.response
}

func (req *PutRequest) prepare(c *api.Context) (*PutRequest, error) {
	preparedRequest := &PutRequest{
		Url:  req.Url,
		Body: req.Body,
	}

	// 赋值默认body
	if preparedRequest.Body == nil {
		cacheBody, err := c.GetBytesBody()
		if err != nil {
			return nil, err
		}

		preparedRequest.Body = cacheBody.Bytes()
	}

	// 赋值默认headers
	preparedRequest.headers = c.GetHeader().Map()

	return preparedRequest, nil
}

type GetRequest struct {
	Url         string
	QueryParams map[string]string
	PathParams  map[string]string

	headers  map[string]string
	response *http_client.Response
}

func (req *GetRequest) Request(c *api.Context, request *http_client.Request) (Request, error) {
	preparedRequest, err := req.prepare(c)
	if err != nil {
		return nil, err
	}

	response, err := request.Get(preparedRequest.Url,
		http_client.WithRequestQueryParams(preparedRequest.QueryParams),
		http_client.WithRequestPathParams(preparedRequest.PathParams),
		http_client.WithRequestHeaders(preparedRequest.headers))
	if err != nil {
		return nil, err
	}

	preparedRequest.response = response

	return preparedRequest, nil
}

func (req *GetRequest) Response() *http_client.Response {
	return req.response
}

func (req *GetRequest) prepare(c *api.Context) (*GetRequest, error) {
	preparedRequest := &GetRequest{
		Url:         req.Url,
		QueryParams: req.QueryParams,
		PathParams:  req.PathParams,
	}

	// 赋值默认查询参数
	if preparedRequest.QueryParams == nil || len(preparedRequest.QueryParams) == 0 {
		preparedRequest.QueryParams = c.GetQueryParams().Map()
	}

	// 赋值默认路径参数
	if preparedRequest.PathParams == nil || len(preparedRequest.PathParams) == 0 {
		preparedRequest.PathParams = c.GetPathParams().Map()
	}

	// 赋值默认headers
	preparedRequest.headers = c.GetHeader().Map()

	return preparedRequest, nil
}