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 = vec!(0); // static ref MAP: HashMap = HashMap::new(); } // #[export_name = "SetIdGenerator"] #[no_mangle] pub extern "C" fn SetIdGenerator(options: IdGeneratorOptions) { YitIdHelper::SetIdGenerator(options); } #[no_mangle] pub extern "C" fn SetWorkerId(workerId: u32) { YitIdHelper::SetWorkerId(workerId); } #[no_mangle] 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 { // 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; }