com.aliyun.openservices.eas.predict.utils.ZstdUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of eas-sdk Show documentation

SDK for PAI-EAS online inference services

There is a newer version: 2.0.20

package com.aliyun.openservices.eas.predict.utils;

import com.github.luben.zstd.Zstd;
import com.github.luben.zstd.ZstdInputStream;
import com.github.luben.zstd.ZstdOutputStream;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

public abstract class ZstdUtils {

    public static byte[] compress(byte[] data) {
        if (data == null || data.length == 0) {
            return data;
        }
        byte[] compressed = Zstd.compress(data);
        return compressed;
    }

    public static void compress(byte[] data, OutputStream os) throws IOException {
        if (data == null || data.length == 0 || os == null) {
            return;
        }
        try (ZstdOutputStream zos = new ZstdOutputStream(os)) {
            zos.write(data);
            zos.flush();
        }
    }

    public static byte[] decompress(byte[] data) {
        if (data == null || data.length == 0) {
            return data;
        }
        long originalSize = Zstd.decompressedSize(data);
        if (originalSize == -1) {
            throw new IllegalArgumentException("Could not determine decompressed size");
        }
        byte[] decompressed = Zstd.decompress(data, (int) originalSize);
        return decompressed;
    }

    public static byte[] decompress(InputStream is) throws IOException {
        if (is == null) {
            return new byte[0];
        }
        try (ByteArrayOutputStream o = new ByteArrayOutputStream();
             ZstdInputStream zis = new ZstdInputStream(is)) {
            byte[] buf = new byte[1024];
            int len;
            while ((len = zis.read(buf)) > 0) {
                o.write(buf, 0, len);
            }
            return o.toByteArray();
        }
    }
}