mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-19 06:54:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Change receive code to use new streaming decomp infra

Browse Source
This commit is contained in:
Kovid Goyal
2023-07-24 22:15:00 +05:30
parent b4d8eaa0fe
commit 42c9f30618
2 changed files with 74 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tools/utils/stream_decompressor.go
View File

@@ -13,10 +13,10 @@ type StreamDecompressor = func(chunk []byte, is_last bool) error
// Wrap Go's awful decompressor routines to allow feeding them
// data in chunks. For example:
// sd, err := NewStreamDecompressor(zlib.NewReader)
// sd(chunk, false, output_callback)
// sd, err := NewStreamDecompressor(zlib.NewReader, output)
// sd(chunk, false)
// ...
// sd(last_chunk, true, output_callback)
// sd(last_chunk, true)
// after this call calling sd() further will just return io.EOF
func NewStreamDecompressor(constructor func(io.Reader) (io.ReadCloser, error), output io.Writer) StreamDecompressor {
if constructor == nil { // identity decompressor
@@ -25,7 +25,15 @@ func NewStreamDecompressor(constructor func(io.Reader) (io.ReadCloser, error), o
if err != nil {
return err
}
_, err = output.Write(chunk)
if len(chunk) > 0 {
_, err = output.Write(chunk)
}
if is_last {
if err == nil {
err = io.EOF
return nil
}
}
return err
}
}