auto commit

Rust/source/src/lib.rs

@@ -3,6 +3,23 @@ mod yitgen;
 use yitgen::gen::YitIdHelper;
 use yitgen::contract::*;
 
+#[macro_use]
+extern crate lazy_static;
+extern crate redis;
+extern crate libc;
+
+use redis::Commands;
+
+use libc::{c_char, uint32_t};
+use std::ffi::{CStr, CString};
+use std::str;
+
+lazy_static! {
+        //static ref TestValue: Vec<i32> = vec!(0);
+        // static ref MAP: HashMap<u32, String> = HashMap::new();
+ }
+
+
 // #[export_name = "SetIdGenerator"]
 #[no_mangle]
 pub extern "C" fn SetIdGenerator(options: IdGeneratorOptions) {
@@ -19,3 +36,64 @@ pub extern "C" fn NextId() -> i64 {
     YitIdHelper::NextId()
 }
 
+static mut TestValue: i32 = 0;
+
+#[no_mangle]
+pub extern "C" fn Test() -> i32 {
+    unsafe {
+        TestValue += 1;
+        return TestValue;
+    }
+}
+
+#[no_mangle]
+pub extern "C" fn GetWorkerId(ip: *const c_char, port: i32) -> redis::RedisResult<isize> {
+    // let c_str = unsafe {
+    //     assert!(!ip.is_null());
+    //     CStr::from_ptr(ip)
+    // };
+    //
+    // let r_str = c_str.to_str();
+
+    // connect to redis
+    // let client = redis::Client::open(format!("redis://{}:{}/", String::from(r_str).to_string(), port))?;
+    let client = redis::Client::open(format!("redis://localhost:{}/", port))?;
+
+    let mut con = client.get_connection()?;
+    // throw away the result, just make sure it does not fail
+    unsafe {
+        let _: () = con.set("my_key111", TestValue.clone())?;
+    }
+    con.get("my_key")
+// read back the key and return it.  Because the return value
+// from the function is a result for integer this will automatically
+// convert into one.
+//
+
+// match simple_redis::create(&format!("redis://{}:{}/", ip, port)) {
+//     Ok(mut client) => {
+//         println!("Created Redis Client");
+//
+//         let valueString = TestValue.to_string();
+//         let valueString2 = (*TestValue).to_string();
+//
+//         match client.set("my_key", valueString) {
+//             Err(error) => println!("Unable to set value in Redis: {}", error),
+//             _ => println!("Value set in Redis")
+//         };
+//
+//         match client.set("my_key2", valueString2) {
+//             Err(error) => println!("Unable to set value in Redis: {}", error),
+//             _ => println!("Value set in Redis")
+//         };
+//
+//         match client.quit() {
+//             Err(error) => println!("Error: {}", error),
+//             _ => println!("Connection Closed.")
+//         }
+//     }
+//     Err(error) => println!("Unable to create Redis client: {}", error)
+// }
+
+//return 1;
+}