open-vault/vendor/github.com/mholt/archiver/xz.go
Calvin Leung Huang d2dbb8c963
Vault Debug (#7375)
* cli: initial work on debug; server-status target

* debug: add metrics capture target (#7376)

* check against DR secondary

* debug: add compression

* refactor check into preflight func

* debug: set short test time on tests, fix exit code bug

* debug: use temp dir for output on tests

* debug: use mholt/archiver for compression

* first pass on adding pprof

* use logger for output

* refactor polling target capture logic

* debug: poll and collect replication status

* debug: poll and collect host-info; rename output files and collection refactor

* fix comments

* add archive test; fix bugs found

* rename flag name to singular target

* add target output test; scaffold other tests cases

* debug/test: add pprof and index file tests

* debug/test: add min timing check tests

* debug: fix index gen race and collection goroutine race

* debug: extend archive tests, handle race between program exit and polling goroutines

* update docstring

* debug: correctly add to pollingWg

* debug: add config target support

* debug: don't wait on interrupt shutdown; add file exists unit tests

* move pprof bits into its goroutine

* debug: skip empty metrics and some pprof file creation if permission denied, add matching unit test

* address comments and feedback

* Vault debug using run.Group (#7658)

* debug: switch to use oklog/run.Group

* debug: use context to cancel requests and interrupt rungroups.

* debug: trigger the first interval properly

* debug: metrics collection should use metrics interval

* debug: add missing continue on metrics error

* debug: remove the use of buffered chan to trigger first interval

* debug: don't shadow BaseCommand's client, properly block on interval capture failures

* debug: actually use c.cachedClient everywhere

* go mod vendor

* debug: run all pprof in goroutines; bump pprof timings in tests to reduce flakiness

* debug: update help text
2019-10-15 15:39:19 -07:00

59 lines
1.2 KiB
Go

package archiver
import (
"fmt"
"io"
"path/filepath"
"github.com/ulikunitz/xz"
fastxz "github.com/xi2/xz"
)
// Xz facilitates XZ compression.
type Xz struct{}
// Compress reads in, compresses it, and writes it to out.
func (x *Xz) Compress(in io.Reader, out io.Writer) error {
w, err := xz.NewWriter(out)
if err != nil {
return err
}
defer w.Close()
_, err = io.Copy(w, in)
return err
}
// Decompress reads in, decompresses it, and writes it to out.
func (x *Xz) Decompress(in io.Reader, out io.Writer) error {
r, err := fastxz.NewReader(in, 0)
if err != nil {
return err
}
_, err = io.Copy(out, r)
return err
}
// CheckExt ensures the file extension matches the format.
func (x *Xz) CheckExt(filename string) error {
if filepath.Ext(filename) != ".xz" {
return fmt.Errorf("filename must have a .xz extension")
}
return nil
}
func (x *Xz) String() string { return "xz" }
// NewXz returns a new, default instance ready to be customized and used.
func NewXz() *Xz {
return new(Xz)
}
// Compile-time checks to ensure type implements desired interfaces.
var (
_ = Compressor(new(Xz))
_ = Decompressor(new(Xz))
)
// DefaultXz is a default instance that is conveniently ready to use.
var DefaultXz = NewXz()