Skip to content
This repository has been archived by the owner on Aug 30, 2022. It is now read-only.

Commit

Permalink
replace unbounded channel with bounded channel
Browse files Browse the repository at this point in the history
  • Loading branch information
Robert-Steiner committed Aug 7, 2020
1 parent 3eb86aa commit e8cfb8e
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions rust/src/metrics/mod.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use influxdb::{Client, WriteQuery};
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};
use tokio::sync::mpsc::{channel, Receiver, Sender};

mod models;

Expand Down Expand Up @@ -202,17 +202,17 @@ pub async fn run_metric_service(mut metics_service: MetricsService) {
}
}

pub struct MetricsSender(UnboundedSender<WriteQuery>);
pub struct MetricsSender(Sender<WriteQuery>);

impl MetricsSender {
pub fn send(&self, query: WriteQuery) {
let _ = self.0.send(query).map_err(|e| error!("{}", e));
pub fn send(&mut self, query: WriteQuery) {
let _ = self.0.try_send(query).map_err(|e| error!("{}", e));
}
}

pub struct MetricsService {
client: Client,
receiver: UnboundedReceiver<WriteQuery>,
receiver: Receiver<WriteQuery>,
}

impl MetricsService {
Expand All @@ -232,7 +232,7 @@ impl MetricsService {
}

fn new_metrics_service(client: Client) -> (MetricsService, MetricsSender) {
let (sender, receiver) = unbounded_channel();
let (sender, receiver) = channel(4096);
(MetricsService { client, receiver }, MetricsSender(sender))
}
}
Expand Down

0 comments on commit e8cfb8e

Please sign in to comment.