idgenerator/Rust/source/src/lib.rs

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) {
    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<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;
}