talos/pkg/imager/post.go
Andrey Smirnov 4c0c626b78
feat: use zstd compression in place of xz
Initramfs and kernel are compressed with zstd.

Extensions are compressed with zstd for Talos 1.8+.

Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
2024-04-29 18:09:12 +04:00

79 lines
2.4 KiB
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package imager
import (
"fmt"
"os"
"path/filepath"
"github.com/siderolabs/go-cmd/pkg/cmd"
"github.com/siderolabs/talos/pkg/reporter"
)
func (i *Imager) postProcessTar(filename string, report *reporter.Reporter) (string, error) {
report.Report(reporter.Update{Message: "processing .tar.gz", Status: reporter.StatusRunning})
dir := filepath.Dir(filename)
src := "disk.raw"
if err := os.Rename(filename, filepath.Join(dir, src)); err != nil {
return "", err
}
outPath := filename + ".tar.gz"
if _, err := cmd.Run("tar", "-cvf", outPath, "-C", dir, "--sparse", "--use-compress-program=pigz -6", src); err != nil {
return "", err
}
if err := os.Remove(filepath.Join(dir, src)); err != nil {
return "", err
}
report.Report(reporter.Update{Message: fmt.Sprintf("archive is ready: %s", outPath), Status: reporter.StatusSucceeded})
return outPath, nil
}
func (i *Imager) postProcessGz(filename string, report *reporter.Reporter) (string, error) {
report.Report(reporter.Update{Message: "compressing .gz", Status: reporter.StatusRunning})
if _, err := cmd.Run("pigz", "-6", "-f", filename); err != nil {
return "", err
}
report.Report(reporter.Update{Message: fmt.Sprintf("compression done: %s.gz", filename), Status: reporter.StatusSucceeded})
return filename + ".gz", nil
}
func (i *Imager) postProcessXz(filename string, report *reporter.Reporter) (string, error) {
report.Report(reporter.Update{Message: "compressing .xz", Status: reporter.StatusRunning})
if _, err := cmd.Run("xz", "-0", "-f", "-T", "0", filename); err != nil {
return "", err
}
report.Report(reporter.Update{Message: fmt.Sprintf("compression done: %s.xz", filename), Status: reporter.StatusSucceeded})
return filename + ".xz", nil
}
func (i *Imager) postProcessZstd(filename string, report *reporter.Reporter) (string, error) {
report.Report(reporter.Update{Message: "compressing .zst", Status: reporter.StatusRunning})
out := filename + ".zst"
if _, err := cmd.Run("zstd", "-T0", "--rm", "-18", "--quiet", "--force", "-o", out, filename); err != nil {
return "", err
}
report.Report(reporter.Update{Message: fmt.Sprintf("compression done: %s", out), Status: reporter.StatusSucceeded})
return filename + ".zst", nil
}