file_io.h

/*
 * Copyright (C) 2011-2017 Redis Labs Ltd.
 *
 * This file is part of memtier_benchmark.
 *
 * memtier_benchmark is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 2.
 *
 * memtier_benchmark is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with memtier_benchmark. If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef _FILE_IO_H
#define_FILE_IO_H

#include<stdio.h>
#include"item.h"

/** Provides a mechanism to read a CSV-like memcache_dump file and extract memcache
 * items from it.
*/
classfile_reader {
protected:
constchar *m_filename; /** name of file */
 FILE* m_file; /** handle of open file */

unsignedint m_line; /** current line being read */

char* read_string(unsignedint len, unsignedint alloc_len, unsignedint* actual_len);
public:
file_reader(constchar *filename);
file_reader(const file_reader& from);
~file_reader();

boolopen_file(void);
boolis_eof(void);
 memcache_item* read_item(void);
};


/** Provides a mechanism to write memcache items into a CSV-like memcache_dump file.
*/
classfile_writer {
protected:
constchar *m_filename; /** name of file */
 FILE *m_file; /** handle of open file */

char* get_quoted_str(char* str, int str_len, int* new_str_len);

public:
file_writer(constchar *filename);
~file_writer();

boolopen_file(void);
boolwrite_item(memcache_item *item);
};

#endif/* _FILE_IO_H */