#![cfg(feature = "invocation")]
use std::{
sync::{Arc, Barrier},
thread::spawn,
};
use jni::{objects::AutoLocal, objects::JValue, 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 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()
)
);
}
}