#![cfg(feature = "invocation")]
extern crate error_chain;
extern crate jni;
use std::sync::{Arc, Barrier};
use std::thread::spawn;
use jni::objects::AutoLocal;
use jni::objects::JValue;
use jni::sys::jint;
mod util;
use util::{attach_current_thread, unwrap};
#[test]
pub fn global_ref_works_in_other_threads() {
const ITERS_PER_THREAD: usize = 10_000;
let env = attach_current_thread();
let mut join_handlers = Vec::new();
let atomic_integer = {
let local_ref = AutoLocal::new(&env, unwrap(&env, env.new_object(
"java/util/concurrent/atomic/AtomicInteger",
"(I)V",
&[JValue::from(0)]
)));
unwrap(&env, env.new_global_ref(local_ref.as_obj()))
};
for thread_num in 2..9 {
let barrier = Arc::new(Barrier::new(thread_num));
for _ in 0..thread_num {
let barrier = barrier.clone();
let mut atomic_integer = atomic_integer.clone();
let jh = spawn(move || {
let env = attach_current_thread();
barrier.wait();
for _ in 0..ITERS_PER_THREAD {
unwrap(&env, unwrap(&env, env.call_method(
atomic_integer.as_obj(), "incrementAndGet", "()I", &[])).i());
}
});
join_handlers.push(jh);
};
for jh in join_handlers.drain(..) {
jh.join().unwrap();
}
let expected = (ITERS_PER_THREAD * thread_num) as jint;
assert_eq!(expected, unwrap(&env, unwrap(&env, env.call_method(
atomic_integer.as_obj(), "getAndSet", "(I)I", &[JValue::from(0)])).i()));
}
}