naza/pkg/nazahttp/read_message.go

// Copyright 2020, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package nazahttp

import (
	"io"
	"strconv"
)

// e.g. bufio.Reader
type HTTPReader interface {
	LineReader
	io.Reader
}

type HTTPMsgCtx struct {
	ReqMethodOrRespVersion string
	ReqURIOrRespStatusCode string
	ReqVersionOrRespReason string
	Headers                map[string]string
	Body                   []byte
}

type HTTPReqMsgCtx struct {
	Method  string
	URI     string
	Version string
	Headers map[string]string
	Body    []byte
}

type HTTPRespMsgCtx struct {
	Version    string
	StatusCode string
	Reason     string
	Headers    map[string]string
	Body       []byte
}

func ReadHTTPRequestMessage(r HTTPReader) (ctx HTTPReqMsgCtx, err error) {
	msgCtx, err := ReadHTTPMessage(r)
	if err != nil {
		return
	}
	ctx.Method = msgCtx.ReqMethodOrRespVersion
	ctx.URI = msgCtx.ReqURIOrRespStatusCode
	ctx.Version = msgCtx.ReqVersionOrRespReason
	ctx.Headers = msgCtx.Headers
	ctx.Body = msgCtx.Body
	return
}

func ReadHTTPResponseMessage(r HTTPReader) (ctx HTTPRespMsgCtx, err error) {
	msgCtx, err := ReadHTTPMessage(r)
	if err != nil {
		return
	}
	ctx.Version = msgCtx.ReqMethodOrRespVersion
	ctx.StatusCode = msgCtx.ReqURIOrRespStatusCode
	ctx.Reason = msgCtx.ReqVersionOrRespReason
	ctx.Headers = msgCtx.Headers
	ctx.Body = msgCtx.Body
	return
}

// 注意，如果HTTP Header中不包含`Content-Length`，则不会读取HTTP Body，并且err返回值为nil
func ReadHTTPMessage(r HTTPReader) (ctx HTTPMsgCtx, err error) {
	var requestLine string
	requestLine, ctx.Headers, err = ReadHTTPHeader(r)
	if err != nil {
		return ctx, err
	}
	ctx.ReqMethodOrRespVersion, ctx.ReqURIOrRespStatusCode, ctx.ReqVersionOrRespReason, err = ParseHTTPRequestLine(requestLine)
	if err != nil {
		return ctx, err
	}

	contentLength, ok := ctx.Headers[HeaderFieldContentLength]
	if !ok {
		return ctx, nil
	}
	cl, err := strconv.Atoi(contentLength)
	if err != nil {
		return ctx, err
	}
	ctx.Body = make([]byte, cl)
	_, err = io.ReadFull(r, ctx.Body)

	return ctx, err
}