Function decodeOTLPWriteRequest

pkg/util/push/otlp.go:125–182

(ctx context.Context, r *http.Request, maxSize int)

}

func decodeOTLPWriteRequest(ctx context.Context, r *http.Request, maxSize int) (pmetricotlp.ExportRequest, error) {
	expectedSize := int(r.ContentLength)
	if expectedSize > maxSize {
		return pmetricotlp.NewExportRequest(), fmt.Errorf("received message larger than max (%d vs %d)", expectedSize, maxSize)
	}

	contentType := r.Header.Get("Content-Type")
	contentEncoding := r.Header.Get("Content-Encoding")

	var compressionType util.CompressionType
	switch contentEncoding {
	case "gzip":
		compressionType = util.Gzip
	case "":
		compressionType = util.NoCompression
	default:
		return pmetricotlp.NewExportRequest(), fmt.Errorf("unsupported compression: %s, Supported compression types are \"gzip\" or '' (no compression)", contentEncoding)
	}

	var decoderFunc func(reader io.Reader) (pmetricotlp.ExportRequest, error)
	switch contentType {
	case pbContentType:
		decoderFunc = func(reader io.Reader) (pmetricotlp.ExportRequest, error) {
			req := pmetricotlp.NewExportRequest()
			otlpReqProto := otlpProtoMessage{req: &req}
			return req, util.ParseProtoReader(ctx, reader, expectedSize, maxSize, otlpReqProto, compressionType)
		}
	case jsonContentType:
		decoderFunc = func(reader io.Reader) (pmetricotlp.ExportRequest, error) {
			req := pmetricotlp.NewExportRequest()

			reader = io.LimitReader(reader, int64(maxSize)+1)
			if compressionType == util.Gzip {
				var err error
				reader, err = gzip.NewReader(reader)
				if err != nil {
					return req, err
				}
				reader = io.LimitReader(reader, int64(maxSize)+1)
			}

			var buf bytes.Buffer
			if expectedSize > 0 {
				buf.Grow(expectedSize + bytes.MinRead) // extra space guarantees no reallocation
			}
			_, err := buf.ReadFrom(reader)
			if err != nil {
				return req, err
			}

			return req, req.UnmarshalJSON(buf.Bytes())
		}
	default:
		return pmetricotlp.NewExportRequest(), fmt.Errorf("unsupported content type: %s, supported: [%s, %s]", contentType, jsonContentType, pbContentType)
	}

	return decoderFunc(r.Body)
}