rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-16 13:00:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: mailbox timeout (#2330)

Browse Source 
refactor: Optimize the timeout mechanism of the mailbox
This commit is contained in:
JeremyHi
2023-09-06 15:35:53 +08:00
committed by Ruihang Xia
parent e4de63625f
commit 56691ff03b
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/meta-srv/src/handler.rs
View File

@@ -15,7 +15,7 @@
use std::collections::{BTreeMap, HashSet};
use std::ops::Range;
use std::sync::Arc;
use std::time::Duration;
use std::time::{Duration, Instant};
use api::v1::meta::mailbox_message::Payload;
use api::v1::meta::{
@@ -262,7 +262,7 @@ pub struct HeartbeatMailbox {
pushers: Pushers,
sequence: Sequence,
senders: DashMap<MessageId, oneshot::Sender<Result<MailboxMessage>>>,
timeouts: DashMap<MessageId, Duration>,
timeouts: DashMap<MessageId, Instant>,
timeout_notify: Notify,
}
@@ -309,7 +309,7 @@ impl HeartbeatMailbox {
self.timeout_notify.notified().await;
}
let now = Duration::from_millis(common_time::util::current_time_millis() as u64);
let now = Instant::now();
let timeout_ids = self
.timeouts
.iter()
@@ -364,8 +364,7 @@ impl Mailbox for HeartbeatMailbox {
let (tx, rx) = oneshot::channel();
let _ = self.senders.insert(message_id, tx);
let deadline =
Duration::from_millis(common_time::util::current_time_millis() as u64) + timeout;
let deadline = Instant::now() + timeout;
let _ = self.timeouts.insert(message_id, deadline);
self.timeout_notify.notify_one();