zstd.go

package main

import (
    "os"
    "log"
    "bufio"
    "flag"
    "github.com/klauspost/compress/zstd"
)

func do_compress(input_file string, output_file string) {
    input, err := os.Open(input_file)
    reader := bufio.NewReader(input)
    if err != nil {
        log.Fatal(err)
    }
    defer input.Close()

    output, err := os.Create(output_file)
    writer := bufio.NewWriter(output)

    enc, _ := zstd.NewWriter(writer)
    _, err = enc.ReadFrom(reader)
    if err != nil {
        log.Fatal(err)
    }
    enc.Flush()
    writer.Flush()
    defer enc.Close()
    defer output.Close()
}

func do_decompress(input_file string, output_file string) {
    input, err := os.Open(input_file)
    reader := bufio.NewReader(input)
    if err != nil {
        log.Fatal(err)
    }
    defer input.Close()

    output, err := os.Create(output_file)
    writer := bufio.NewWriter(output)

    dec, _ := zstd.NewReader(reader)
    _, err = dec.WriteTo(writer)
    if err != nil {
        log.Fatal(err)
    }
    defer dec.Close()
    defer output.Close()
}

func main() {
    decompress := flag.Bool("d", false, "Decompress")
    input_file := flag.String("i", "", "Input file")
    output_file := flag.String("o", "", "Output file")
    flag.Parse()

    if *decompress {
        println("Decompressing")
        do_decompress(*input_file, *output_file)
    } else {
        println("Compressing")
        do_compress(*input_file, *output_file)
    }
}