Timothy Messier
fe870fc989
A ChunkDecoder can be used to BSR Chunks. Each call to Decode will read the next chunk from the decoder's io.Reader. It handles decompressing any compressed chunk data, and verifying the CRCs of each chunk.
67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package bsr
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
)
|
|
|
|
const (
|
|
compressionSize = 1
|
|
)
|
|
|
|
// Compression is used to identify the compression used for the data in chunks.
|
|
type Compression uint8
|
|
|
|
// Supported compression methods.
|
|
const (
|
|
NoCompression Compression = iota
|
|
GzipCompression
|
|
)
|
|
|
|
func (c Compression) String() string {
|
|
switch c {
|
|
case NoCompression:
|
|
return "no compression"
|
|
case GzipCompression:
|
|
return "gzip"
|
|
default:
|
|
return "unknown compression"
|
|
}
|
|
}
|
|
|
|
// ValidCompression checks if a given Compression is valid.
|
|
func ValidCompression(c Compression) bool {
|
|
switch c {
|
|
case NoCompression, GzipCompression:
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
type nullCompressionWriter struct {
|
|
*bytes.Buffer
|
|
}
|
|
|
|
func (w *nullCompressionWriter) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func newNullCompressionWriter(b *bytes.Buffer) io.WriteCloser {
|
|
return &nullCompressionWriter{Buffer: b}
|
|
}
|
|
|
|
type nullCompressionReader struct {
|
|
*bytes.Buffer
|
|
}
|
|
|
|
func (w *nullCompressionReader) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func newNullCompressionReader(b *bytes.Buffer) io.ReadCloser {
|
|
return &nullCompressionReader{Buffer: b}
|
|
}
|