cpp/src/compress/uncompressed_compressor.h

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * License); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

#ifndef COMPRESS_UNCOMPRESSED_COMPRESSOR_H
#define COMPRESS_UNCOMPRESSED_COMPRESSOR_H

#include "compressor.h"

namespace storage {

class UncompressedCompressor : public Compressor {
   public:
    UncompressedCompressor() : uncompressed_buf_(nullptr) {}
    virtual ~UncompressedCompressor() {
        if (uncompressed_buf_ != nullptr) {
            common::mem_free(uncompressed_buf_);
            uncompressed_buf_ = nullptr;
        }
    }
    int reset(bool for_compress) {
        UNUSED(for_compress);
        if (uncompressed_buf_ != nullptr) {
            common::mem_free(uncompressed_buf_);
            uncompressed_buf_ = nullptr;
        }
        return common::E_OK;
    }
    void destroy() {
        if (uncompressed_buf_ != nullptr) {
            common::mem_free(uncompressed_buf_);
            uncompressed_buf_ = nullptr;
        }
    }
    int compress(char* uncompressed_buf, uint32_t uncompressed_buf_len,
                 char*& compressed_buf, uint32_t& compressed_buf_len) {
        compressed_buf = uncompressed_buf;
        compressed_buf_len = uncompressed_buf_len;
        return common::E_OK;
    }
    void after_compress(char* compressed_buf) { UNUSED(compressed_buf); }

    int uncompress(char* compressed_buf, uint32_t compressed_buf_len,
                   char*& uncompressed_buf, uint32_t& uncompressed_buf_len) {
        char* buf = static_cast<char*>(
            common::mem_alloc(compressed_buf_len, common::MOD_COMPRESSOR_OBJ));
        if (buf == nullptr) {
            return common::E_OOM;
        }
        memcpy(buf, compressed_buf, compressed_buf_len);
        uncompressed_buf = buf;
        uncompressed_buf_ = buf;
        uncompressed_buf_len = compressed_buf_len;
        return common::E_OK;
    }
    void after_uncompress(char* uncompressed_buf) {
        if (uncompressed_buf != nullptr) {
            common::mem_free(uncompressed_buf_);
            uncompressed_buf_ = nullptr;
        }
    }

   private:
    char* uncompressed_buf_;
};

}  // end namespace storage
#endif  // COMPRESS_UNCOMPRESSED_COMPRESSOR_H