baize/framework/core/api/context.go

package api

import (
	"bytes"
	"git.sxidc.com/go-framework/baize/framework/core/infrastructure/logger"
	"github.com/gin-gonic/gin"
	"io"
	"mime/multipart"
	"strings"
)

type Context struct {
	*gin.Context
}

// GetFileHeaderBytes 获取传递的文件名和文件内容
func (c *Context) GetFileHeaderBytes(fileHeader *multipart.FileHeader) (string, []byte, error) {
	file, err := fileHeader.Open()
	if err != nil {
		return "", nil, err
	}

	defer func(file multipart.File) {
		err := file.Close()
		if err != nil {
			logger.GetInstance().Error(err)
			return
		}
	}(file)

	contentBytes, err := io.ReadAll(file)
	if err != nil {
		return "", nil, err
	}

	return fileHeader.Filename, contentBytes, nil
}

func (c *Context) GetHeaders() map[string]string {
	headers := make(map[string]string, 0)

	for key, values := range c.Request.Header {
		headers[key] = strings.Join(values, ",")
	}

	return headers
}

func (c *Context) ReadBody() ([]byte, error) {
	if c.Request.Body == nil {
		return make([]byte, 0), nil
	}

	body, err := io.ReadAll(c.Request.Body)
	if err != nil {
		return nil, err
	}

	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			logger.GetInstance().Error(err)
			return
		}
	}(c.Request.Body)

	c.Request.Body = io.NopCloser(bytes.NewBuffer(body))

	return body, nil
}

func (c *Context) ReplaceBody(body []byte) error {
	if c.Request.Body != nil {
		err := c.Request.Body.Close()
		if err != nil {
			return err
		}
	}

	c.Request.Body = io.NopCloser(bytes.NewBuffer(body))

	return nil
}

func (c *Context) GetAllQueryParams() map[string]string {
	queryParams := make(map[string]string, 0)

	for key, values := range c.Request.URL.Query() {
		queryParams[key] = strings.Join(values, ",")
	}

	return queryParams
}

func (c *Context) GetAllPathParams() map[string]string {
	pathParams := make(map[string]string, 0)

	for _, params := range c.Params {
		pathParams[params.Key] = params.Value
	}

	return pathParams
}