enable to use rocksdb for pwal sorting

CMakeLists.txt

@@ -41,6 +41,8 @@ if (ENABLE_GOOGLE_PERFTOOLS)
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DENABLE_GOOGLE_PERFTOOLS")
 endif()
 
+set(SORT_METHOD LEVELDB CACHE STRING "sort method at recovery process")
+
 # set(ENGINE "engine")
 
 find_package(Doxygen)

cmake/FindRocksDB.cmake

@@ -0,0 +1,24 @@
+if(TARGET rocksdb::rocksdb)
+    return()
+endif()
+
+find_library(rocksdb_LIBRARY_FILE NAMES rocksdb)
+find_path(rocksdb_INCLUDE_DIR NAMES rocksdb/db.h)
+
+include(FindPackageHandleStandardArgs)
+find_package_handle_standard_args(rocksdb DEFAULT_MSG
+    rocksdb_LIBRARY_FILE
+    rocksdb_INCLUDE_DIR)
+
+if(rocksdb_LIBRARY_FILE AND rocksdb_INCLUDE_DIR)
+    set(rocksdb_FOUND ON)
+    add_library(RocksDB::rocksdb SHARED IMPORTED)
+    set_target_properties(RocksDB::rocksdb PROPERTIES
+        IMPORTED_LOCATION "${rocksdb_LIBRARY_FILE}"
+        INTERFACE_INCLUDE_DIRECTORIES "${rocksdb_INCLUDE_DIR}")
+else()
+    set(rocksdb_FOUND OFF)
+endif()
+
+unset(rocksdb_LIBRARY_FILE CACHE)
+unset(rocksdb_INCLUDE_DIR CACHE)

src/CMakeLists.txt

@@ -10,14 +10,21 @@ set_target_properties(${package_name} PROPERTIES
         OUTPUT_NAME ${export_name}
         )
 
+if($CACHE{SORT_METHOD} STREQUAL "ROCKSDB")
+    set(sort_lib rocksdb)
+    target_compile_options(${package_name} PRIVATE -DSORT_METHOD_USE_ROCKSDB)
+else()
+    set(sort_lib leveldb)
+endif()
+
 target_link_libraries(${package_name}
         PUBLIC limestone-api
         PRIVATE Boost::boost
         PRIVATE Boost::filesystem
         PRIVATE Boost::thread
         PRIVATE Boost::container
         PRIVATE glog::glog
-        PRIVATE leveldb
+        PRIVATE ${sort_lib}
 )
 
 set_compile_options(${package_name})
@@ -38,5 +45,5 @@ target_link_libraries(limestone-impl
         INTERFACE Boost::thread
         INTERFACE Boost::container
         INTERFACE glog::glog
-        INTERFACE leveldb
+        INTERFACE ${sort_lib}
 )

src/limestone/leveldb_wrapper.h

@@ -15,7 +15,13 @@
  */
 // #include <boost/filesystem/operations.hpp>
 // #include <boost/foreach.hpp>
+
+#ifdef SORT_METHOD_USE_ROCKSDB
+#include <rocksdb/db.h>
+namespace leveldb = rocksdb;
+#else
 #include <leveldb/db.h>
+#endif
 
 #include <glog/logging.h>
 #include <limestone/logging.h>