rust to cpp

Author: centos7

Cargo.toml

@@ -0,0 +1,18 @@
+[package]
+name = "client-cpp"
+version = "0.1.0"
+authors = ["venslu <1262612838@qq.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+[lib]
+name = "tikv_client_rust"
+crate-type = ["staticlib"]
+
+[dependencies]
+anyhow = "1.0"
+cxx = "1.0.18"
+futures = "0.3"
+tikv-client = "0.0.0"
+grpcio = { version = "0.6", features = ["openssl"] }
+#tikv-client = { git = "https://github.com/tikv/client-rust.git", branch = "master" }

Makefile

@@ -0,0 +1,40 @@
+pes_parent_dir:=$(shell pwd)/$(lastword $(MAKEFILE_LIST))
+cur_makefile_path := $(shell dirname $(pes_parent_dir))
+
+all: build-lib build-test
+
+build-test: target/tikv-test 
+
+build-lib: pre-build target/cxx/tikv_client_glue.o target/cxx/tikv_client_cpp.o target/cxx/libtikv_client.a 
+
+pre-build: target/cxx target/cxx/libtikv_client_rust.a target/cxx/tikv_client_glue.cc target/cxx/tikv_client_glue.h
+
+clean:
+	cargo clean
+
+run: target/tikv-test
+	RUST_BACKTRACE=1 $(cur_makefile_path)/target/tikv-test
+
+target/cxx:
+	mkdir -p $(cur_makefile_path)/target/cxx
+
+target/tikv-test: target/cxx/libtikv_client.a example/main.cpp
+	c++ $(cur_makefile_path)/example/main.cpp -o $(cur_makefile_path)/target/tikv-test -std=c++17 -g -I$(cur_makefile_path)/target/cxx -I$(cur_makefile_path)/include -L$(cur_makefile_path)/target/cxx -ltikv_client -lpthread -ldl -lssl -lcrypto
+
+target/cxx/libtikv_client.a: target/cxx/libtikv_client_rust.a target/cxx/tikv_client_glue.o target/cxx/tikv_client_cpp.o
+	cd $(cur_makefile_path)/target/cxx && cp libtikv_client_rust.a libtikv_client.a && ar cr libtikv_client.a tikv_client_cpp.o tikv_client_glue.o
+
+target/cxx/tikv_client_cpp.o: src/tikv_client.cpp
+	c++ -c $(cur_makefile_path)/src/tikv_client.cpp -o $(cur_makefile_path)/target/cxx/tikv_client_cpp.o -std=c++17 -g -I$(cur_makefile_path)/target/cxx -I$(cur_makefile_path)/include
+
+target/cxx/tikv_client_glue.o: target/cxx/tikv_client_glue.cc
+	c++ -c $(cur_makefile_path)/target/cxx/tikv_client_glue.cc -o $(cur_makefile_path)/target/cxx/tikv_client_glue.o -std=c++17 -I$(cur_makefile_path)/target/cxx
+
+target/cxx/libtikv_client_rust.a: src/lib.rs
+	cargo build && mv $(cur_makefile_path)/target/debug/libtikv_client_rust.a $(cur_makefile_path)/target/cxx
+
+target/cxx/tikv_client_glue.cc: src/lib.rs
+	cxxbridge $(cur_makefile_path)/src/lib.rs > $(cur_makefile_path)/target/cxx/tikv_client_glue.cc
+
+target/cxx/tikv_client_glue.h: src/lib.rs
+	cxxbridge $(cur_makefile_path)/src/lib.rs --header > $(cur_makefile_path)/target/cxx/tikv_client_glue.h

README.md

@@ -6,3 +6,22 @@ It's built on top of
 [TiKV Client in Rust](https://github.com/tikv/client-rust) via [cxx](https://github.com/dtolnay/cxx). 
 
 This client is still in the stage of prove-of-concept and under heavy development.
+
+# compilation process
+
+## all
+
+```bash
+make all
+```
+
+## build static lib
+
+```bash
+make build-lib
+```
+
+## output target
+
+- libtikv_client.a : static lib
+- tikv-test: execute file for test

example/main.cpp

@@ -0,0 +1,19 @@
+#include "tikv_client.h"
+#include <iostream>
+
+int main() {
+    auto client = tikv_client::TransactionClient({"127.0.0.1:2379"});
+    auto txn = client.begin();
+    //txn.put("k1", "v2");
+    auto val = txn.get("k1");
+    if (val) {
+        std::cout << "get key k1:" << *val << std::endl;
+    } else {
+        std::cout << "key not found" << std::endl;
+    }
+    auto kv_pairs = txn.scan("k1", Bound::Included, "", Bound::Unbounded, 10);
+    for (auto iter = kv_pairs.begin(); iter != kv_pairs.end(); ++iter) {
+        std::cout << "scan:" << iter->key << ": " << iter->value << std::endl;
+    }
+    return 0;
+}

include/tikv_client.h

@@ -0,0 +1,47 @@
+#ifndef _TIKV_CLIENT_H_
+#define _TIKV_CLIENT_H_ 
+
+#include "tikv_client_glue.h"
+#include <iostream>
+#include <optional>
+
+namespace tikv_client {
+
+struct KvPair final {
+    std::string key;
+    std::string value;
+
+    KvPair(std::string &&key, std::string &&value);
+};
+
+class Transaction {
+public:
+    Transaction(::rust::cxxbridge1::Box<tikv_client_glue::Transaction> txn);
+    std::optional<std::string> get(const std::string &key);
+    std::optional<std::string> get_for_update(const std::string &key);
+    std::vector<KvPair> batch_get(const std::vector<std::string> &keys);
+    std::vector<KvPair> batch_get_for_update(const std::vector<std::string> &keys);
+    std::vector<KvPair> scan(const std::string &start, Bound start_bound, const std::string &end, Bound end_bound, std::uint32_t limit);
+    std::vector<std::string> scan_keys(const std::string &start, Bound start_bound, const std::string &end, Bound end_bound, std::uint32_t limit);
+    bool key_may_exist(const std::string &key); 
+    void put(const std::string &key, const std::string &value);
+    void batch_put(const std::vector<KvPair> &kvs);
+    bool merge(const std::string &key, const std::string &value);
+    void remove(const std::string &key);
+    void commit();
+private:
+    ::rust::cxxbridge1::Box<tikv_client_glue::Transaction> _txn;
+};
+
+class TransactionClient {
+public:
+    TransactionClient(const std::vector<std::string> &pd_endpoints);
+    Transaction begin();
+    Transaction begin_pessimistic();
+private:
+    ::rust::cxxbridge1::Box<tikv_client_glue::TransactionClient> _client;
+};
+
+}
+
+#endif //_TIKV_CLIENT_H_