--- /dev/null
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2015 Haomai Wang <haomaiwang@gmail.com>
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation. See file COPYING.
+ *
+ */
+
+
+#include <time.h>
+#include <boost/random/mersenne_twister.hpp>
+#include <boost/random/uniform_int.hpp>
+#include <boost/random/binomial_distribution.hpp>
+#include <gtest/gtest.h>
+#include "common/ceph_argparse.h"
+#include "compressor/AsyncCompressor.h"
+#include "global/global_init.h"
+
+typedef boost::mt11213b gen_type;
+
+class AsyncCompressorTest : public ::testing::Test {
+ public:
+ AsyncCompressor *async_compressor;
+ void SetUp() override {
+ cerr << __func__ << " start set up " << std::endl;
+ async_compressor = new AsyncCompressor(g_ceph_context);
+ async_compressor->init();
+ }
+ void TearDown() override {
+ async_compressor->terminate();
+ delete async_compressor;
+ }
+
+ void generate_random_data(bufferlist &bl, uint64_t len = 0) {
+ static const char *base= "znvm,x12399zasdfjkl1209zxcvjlkasjdfljwqelrjzx,cvn,m123#*(@)";
+ if (!len) {
+ boost::uniform_int<> kb(16, 4096);
+ gen_type rng(time(NULL));
+ len = kb(rng) * 1024;
+ }
+
+ while (bl.length() < len)
+ bl.append(base, sizeof(base)-1);
+ }
+};
+
+TEST_F(AsyncCompressorTest, SimpleTest) {
+ bufferlist compress_data, decompress_data, rawdata;
+ generate_random_data(rawdata, 1<<22);
+ bool finished;
+ uint64_t id = async_compressor->async_compress(rawdata);
+ ASSERT_EQ(0, async_compressor->get_compress_data(id, compress_data, true, &finished));
+ ASSERT_TRUE(finished == true);
+ id = async_compressor->async_decompress(compress_data);
+ do {
+ ASSERT_EQ(0, async_compressor->get_decompress_data(id, decompress_data, false, &finished));
+ } while (!finished);
+ ASSERT_TRUE(finished == true);
+ ASSERT_TRUE(rawdata.contents_equal(decompress_data));
+ ASSERT_EQ(-ENOENT, async_compressor->get_decompress_data(id, decompress_data, true, &finished));
+}
+
+TEST_F(AsyncCompressorTest, GrubWaitTest) {
+ async_compressor->terminate();
+ bufferlist compress_data, decompress_data, rawdata;
+ generate_random_data(rawdata, 1<<22);
+ bool finished;
+ uint64_t id = async_compressor->async_compress(rawdata);
+ ASSERT_EQ(0, async_compressor->get_compress_data(id, compress_data, true, &finished));
+ ASSERT_TRUE(finished == true);
+ id = async_compressor->async_decompress(compress_data);
+ ASSERT_EQ(0, async_compressor->get_decompress_data(id, decompress_data, true, &finished));
+ ASSERT_TRUE(finished == true);
+ ASSERT_TRUE(rawdata.contents_equal(decompress_data));
+ async_compressor->init();
+}
+
+TEST_F(AsyncCompressorTest, DecompressInjectTest) {
+ bufferlist compress_data, decompress_data, rawdata;
+ generate_random_data(rawdata, 1<<22);
+ bool finished;
+ uint64_t id = async_compressor->async_compress(rawdata);
+ ASSERT_EQ(0, async_compressor->get_compress_data(id, compress_data, true, &finished));
+ ASSERT_TRUE(finished == true);
+ char error[] = "asjdfkwejrljqwaelrj";
+ memcpy(compress_data.c_str()+1024, error, sizeof(error)-1);
+ id = async_compressor->async_decompress(compress_data);
+ ASSERT_EQ(-EIO, async_compressor->get_decompress_data(id, decompress_data, true, &finished));
+}
+
+class SyntheticWorkload {
+ set<pair<uint64_t, uint64_t> > compress_jobs, decompress_jobs;
+ AsyncCompressor *async_compressor;
+ vector<bufferlist> rand_data, compress_data;
+ gen_type rng;
+ static const uint64_t MAX_INFLIGHT = 128;
+
+ public:
+ explicit SyntheticWorkload(AsyncCompressor *ac): async_compressor(ac), rng(time(NULL)) {
+ for (int i = 0; i < 100; i++) {
+ bufferlist bl;
+ boost::uniform_int<> u(4096, 1<<24);
+ uint64_t value_len = u(rng);
+ bufferptr bp(value_len);
+ bp.zero();
+ for (uint64_t j = 0; j < value_len-sizeof(i); ) {
+ memcpy(bp.c_str()+j, &i, sizeof(i));
+ j += 4096;
+ }
+
+ bl.append(bp);
+ rand_data.push_back(bl);
+ compress_jobs.insert(make_pair(async_compressor->async_compress(rand_data[i]), i));
+ if (!(i % 10)) cerr << "seeding compress data " << i << std::endl;
+ }
+ compress_data.resize(100);
+ reap(true);
+ }
+ void do_compress() {
+ boost::uniform_int<> u(0, rand_data.size()-1);
+ uint64_t index = u(rng);
+ compress_jobs.insert(make_pair(async_compressor->async_compress(rand_data[index]), index));
+ }
+ void do_decompress() {
+ boost::uniform_int<> u(0, compress_data.size()-1);
+ uint64_t index = u(rng);
+ if (compress_data[index].length())
+ decompress_jobs.insert(make_pair(async_compressor->async_decompress(compress_data[index]), index));
+ }
+ void reap(bool blocking) {
+ bufferlist data;
+ bool finished;
+ set<pair<uint64_t, uint64_t> >::iterator prev;
+ uint64_t c_reap = 0, d_reap = 0;
+ do {
+ for (set<pair<uint64_t, uint64_t> >::iterator it = compress_jobs.begin();
+ it != compress_jobs.end();) {
+ prev = it;
+ ++it;
+ ASSERT_EQ(0, async_compressor->get_compress_data(prev->first, data, blocking, &finished));
+ if (finished) {
+ c_reap++;
+ if (compress_data[prev->second].length())
+ ASSERT_TRUE(compress_data[prev->second].contents_equal(data));
+ else
+ compress_data[prev->second].swap(data);
+ compress_jobs.erase(prev);
+ }
+ }
+
+ for (set<pair<uint64_t, uint64_t> >::iterator it = decompress_jobs.begin();
+ it != decompress_jobs.end();) {
+ prev = it;
+ ++it;
+ ASSERT_EQ(0, async_compressor->get_decompress_data(prev->first, data, blocking, &finished));
+ if (finished) {
+ d_reap++;
+ ASSERT_TRUE(rand_data[prev->second].contents_equal(data));
+ decompress_jobs.erase(prev);
+ }
+ }
+ usleep(1000 * 500);
+ } while (compress_jobs.size() + decompress_jobs.size() > MAX_INFLIGHT);
+ cerr << " reap compress jobs " << c_reap << " decompress jobs " << d_reap << std::endl;
+ }
+ void print_internal_state() {
+ cerr << "inlfight compress jobs: " << compress_jobs.size()
+ << " inflight decompress jobs: " << decompress_jobs.size() << std::endl;
+ }
+ bool empty() const { return compress_jobs.empty() && decompress_jobs.empty(); }
+};
+
+TEST_F(AsyncCompressorTest, SyntheticTest) {
+ SyntheticWorkload test_ac(async_compressor);
+ gen_type rng(time(NULL));
+ boost::uniform_int<> true_false(0, 99);
+ int val;
+ for (int i = 0; i < 3000; ++i) {
+ if (!(i % 10)) {
+ cerr << "Op " << i << ": ";
+ test_ac.print_internal_state();
+ }
+ val = true_false(rng);
+ if (val < 45) {
+ test_ac.do_compress();
+ } else if (val < 95) {
+ test_ac.do_decompress();
+ } else {
+ test_ac.reap(false);
+ }
+ }
+ while (!test_ac.empty()) {
+ test_ac.reap(false);
+ test_ac.print_internal_state();
+ usleep(1000*500);
+ }
+}
+
+
+int main(int argc, char **argv) {
+ vector<const char*> args;
+ argv_to_vec(argc, (const char **)argv, args);
+
+ auto cct = global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT, CODE_ENVIRONMENT_UTILITY, 0);
+ common_init_finish(g_ceph_context);
+
+ const char* env = getenv("CEPH_LIB");
+ string directory(env ? env : ".libs");
+ g_conf->set_val("plugin_dir", directory, false);
+
+ ::testing::InitGoogleTest(&argc, argv);
+ return RUN_ALL_TESTS();
+}
+
+/*
+ * Local Variables:
+ * compile-command: "cd ../.. ; make -j4 unittest_async_compressor && valgrind --tool=memcheck ./unittest_async_compressor"
+ * End:
+ */