|
|
@@ -20,9 +20,17 @@ lazy_static! { |
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// #[export_name = "SetIdGenerator"]
|
|
|
|
// // #[export_name = "SetIdGenerator"]
|
|
|
|
// #[no_mangle]
|
|
|
|
// pub extern "C" fn SetIdGenerator(options: IdGeneratorOptions) {
|
|
|
|
// YitIdHelper::SetIdGenerator(options);
|
|
|
|
// }
|
|
|
|
|
|
|
|
#[no_mangle]
|
|
|
|
pub extern "C" fn SetIdGenerator(options: IdGeneratorOptions) {
|
|
|
|
pub extern "C" fn SetOptions(workerId: u32, workerIdBitLength: u8, seqBitLength: u8) {
|
|
|
|
let mut options = IdGeneratorOptions::New(1);
|
|
|
|
options.WorkerIdBitLength = workerIdBitLength;
|
|
|
|
options.SeqBitLength = seqBitLength;
|
|
|
|
YitIdHelper::SetIdGenerator(options);
|
|
|
|
}
|
|
|
|
|
|
|
@@ -36,64 +44,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.
|
|
|
|
//
|
|
|
|
// static mut TestValue: i32 = 0;
|
|
|
|
// #[no_mangle]
|
|
|
|
// pub extern "C" fn Test() -> i32 {
|
|
|
|
// unsafe {
|
|
|
|
// TestValue += 1;
|
|
|
|
// return TestValue;
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
|
|
|
|
// match simple_redis::create(&format!("redis://{}:{}/", ip, port)) {
|
|
|
|
// Ok(mut client) => {
|
|
|
|
// println!("Created Redis Client");
|
|
|
|
// #[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();
|
|
|
|
//
|
|
|
|
// let valueString = TestValue.to_string();
|
|
|
|
// let valueString2 = (*TestValue).to_string();
|
|
|
|
// // 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))?;
|
|
|
|
//
|
|
|
|
// match client.set("my_key", valueString) {
|
|
|
|
// Err(error) => println!("Unable to set value in Redis: {}", error),
|
|
|
|
// _ => println!("Value set in Redis")
|
|
|
|
// };
|
|
|
|
// 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 client.set("my_key2", valueString2) {
|
|
|
|
// Err(error) => println!("Unable to set value in Redis: {}", error),
|
|
|
|
// _ => println!("Value set in Redis")
|
|
|
|
// };
|
|
|
|
// // 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)
|
|
|
|
// // }
|
|
|
|
//
|
|
|
|
// match client.quit() {
|
|
|
|
// Err(error) => println!("Error: {}", error),
|
|
|
|
// _ => println!("Connection Closed.")
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
// Err(error) => println!("Unable to create Redis client: {}", error)
|
|
|
|
// //return 1;
|
|
|
|
// }
|
|
|
|
|
|
|
|
//return 1;
|
|
|
|
} |