add update titan script (#359)

* add update titan script
Signed-off-by: Connor1996 <zbk602423539@gmail.com>

Cargo.toml

@@ -20,6 +20,7 @@ jemalloc = ["librocksdb_sys/jemalloc"]
 portable = ["librocksdb_sys/portable"]
 sse = ["librocksdb_sys/sse"]
 valgrind = []
+update_titan = ["librocksdb_sys/update_titan"]
 
 [dependencies]
 libc = "0.2.11"

Makefile

@@ -14,3 +14,12 @@ clean:
 	@cargo clean
 	@cd librocksdb_sys && cargo clean
 
+update_titan:
+	@if [ -n "${TITAN_REPO}" ]; then \
+		git config --file=.gitmodules submodule.titan.url https://github.com/${TITAN_REPO}/titan.git; \
+	fi
+	@if [ -n "${TITAN_BRANCH}" ]; then \
+		git config --file=.gitmodules submodule.titan.branch ${TITAN_BRANCH}; \
+	fi
+	@git submodule sync
+	@git submodule update --remote librocksdb_sys/libtitan_sys/titan
\ No newline at end of file

librocksdb_sys/Cargo.toml

@@ -19,6 +19,7 @@ jemalloc = ["jemalloc-sys"]
 # when not using with static-link right now in this crate.
 portable = ["libtitan_sys/portable"]
 sse = ["libtitan_sys/sse"]
+update_titan = ["libtitan_sys/update_titan"]
 
 [build-dependencies]
 cc = "1.0.3"

librocksdb_sys/libtitan_sys/Cargo.toml

@@ -13,6 +13,7 @@ default = []
 # when not using with static-link right now in this crate.
 portable = []
 sse = []
+update_titan = []
 
 [build-dependencies]
 cc = "1.0.3"

librocksdb_sys/libtitan_sys/build.rs

@@ -2,6 +2,7 @@ extern crate cc;
 extern crate cmake;
 
 use std::env;
+use std::process::Command;
 
 fn main() {
     // RocksDB cmake script expect libz.a being under ${DEP_Z_ROOT}/lib, but libz-sys crate put it
@@ -22,6 +23,13 @@ fn main() {
     if cfg!(feature = "sse") {
         cfg.define("FORCE_SSE42", "ON");
     }
+    if cfg!(feature = "update_titan") {
+        Command::new("make")
+            .current_dir("../..")
+            .args(&["update_titan"])
+            .status()
+            .expect("failed to update titan!");
+    }
     let dst = cfg
         .define("ROCKSDB_DIR", cur_dir.join("..").join("rocksdb"))
         .define("WITH_TITAN_TESTS", "OFF")