ref(consumer): Add batch write timeout to Rust consumer

ref(consumer): Add batch write timeout to Rust consumer #16973