google Snappy

源地址

#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>


#include <snappy.h>
#include <snappy-sinksource.h>


static const char* IN_NAME="./main.cc";
static const char* OUT_NAME="./main.cc.compress";


class FileSink:public snappy::Sink{
  public:
    FileSink(int fd):fd_(fd){
    }
    virtual ~FileSink(){}
    virtual char* GetAppendBuffer(size_t length,char* scratch){
        return scratch;
    }
    virtual void Append(const char* bytes,size_t n){
        write(fd_,bytes,n);
    }
  private:
    int fd_;
};


int main(){
    struct stat stbuf;
    stat(IN_NAME,&stbuf);
    size_t in_fsize=stbuf.st_size;
    size_t compress_fsize=0;


    // do comress
    char* in_buf=new char[in_fsize];
    {
        FILE* fin=fopen(IN_NAME,"rb");
        fread(in_buf,1,in_fsize,fin);
        fclose(fin);
        snappy::ByteArraySource source(in_buf,in_fsize);


        int fd=open(OUT_NAME,O_CREAT | O_WRONLY,0666);
        FileSink sink(fd);
        compress_fsize=snappy::Compress(&source,&sink);
        close(fd);
    }


    // do decompress
    stat(OUT_NAME,&stbuf);
    size_t out_fsize=stbuf.st_size;
    assert(out_fsize==compress_fsize);
    char* out_buf=new char[out_fsize];
    std::string out;
    {
        FILE* fin=fopen(OUT_NAME,"rb");
        fread(out_buf,1,out_fsize,fin);
        fclose(fin);
        assert(snappy::Uncompress(out_buf,out_fsize,&out)==true);
    }


    // do validate
    assert(out.size()==in_fsize);
    assert(memcmp(out.data(),in_buf,in_fsize)==0);
    delete [] in_buf;
    delete [] out_buf;
    return 0;
}