Skip to content

Commit

Permalink
wtf
Browse files Browse the repository at this point in the history
  • Loading branch information
Xavrax committed Mar 15, 2024
1 parent dab433d commit 4cf6e2b
Show file tree
Hide file tree
Showing 2 changed files with 101 additions and 0 deletions.
79 changes: 79 additions & 0 deletions examples/benchmark.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,79 @@
use futures::stream::StreamExt;
use pubnub::{
subscribe::{EventEmitter, EventSubscriber, SubscriptionParams},
Keyset, PubNubClientBuilder,
};
use serde::Serialize;
use std::{
env,
sync::{Arc, Mutex, RwLock},
time::{Duration, Instant, SystemTime},
};
use tokio::time::sleep;

#[derive(Serialize)]
struct Message {
url: String,
description: String,
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn snafu::Error>> {
let publish_key = env::var("SDK_PUB_KEY")?;
let subscribe_key = env::var("SDK_SUB_KEY")?;

let client = PubNubClientBuilder::with_reqwest_transport()
.with_keyset(Keyset {
subscribe_key,
publish_key: Some(publish_key),
secret_key: None,
})
.with_user_id("user_id")
.build()?;

let sub = client.subscription(SubscriptionParams {
channels: Some(&["kekw"]),
channel_groups: None,
options: None,
});

sub.subscribe();

tokio::spawn(sub.messages_stream().for_each(|msg| async move {
println!(
"sub {:?}",
SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis()
);

println!("Received message: {:?}", msg);
}));

println!(
"pub {:?}",
SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis()
);

// publish simple string
client
.publish_message("chuj dupa cipa")
.channel("kekw")
.execute()
.await?;
println!(
"pu2 {:?}",
SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis()
);

sleep(Duration::from_secs(1)).await;

Ok(())
}
22 changes: 22 additions & 0 deletions src/transport/reqwest.rs
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
not(feature = "std")
))]
use crate::dx::pubnub_client::PubNubClientDeserializerBuilder;
use std::time::SystemTime;

#[cfg(all(any(feature = "subscribe", feature = "presence"), feature = "std"))]
use crate::dx::pubnub_client::PubNubClientRuntimeBuilder;
Expand Down Expand Up @@ -86,6 +87,7 @@ impl Transport for TransportReqwest {
#[cfg(feature = "std")]
let timeout = request.timeout;

let is_publish = request.path.contains("publish");
#[cfg(feature = "std")]
let mut builder = match request.method {
TransportMethod::Get => self.prepare_get_method(request, request_url),
Expand All @@ -105,6 +107,16 @@ impl Transport for TransportReqwest {
TransportMethod::Delete => self.prepare_delete_method(request, request_url),
}?;

if is_publish {
println!(
"p00 {:?}",
SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis()
);
}

let result = builder
.headers(headers)
.send()
Expand All @@ -114,6 +126,16 @@ impl Transport for TransportReqwest {
response: None,
})?;

if is_publish {
println!(
"p01 {:?}",
SystemTime::now()
.duration_since(SystemTime::UNIX_EPOCH)
.unwrap()
.as_millis()
);
}

let headers = result.headers().clone();
let status = result.status();
result
Expand Down

0 comments on commit 4cf6e2b

Please sign in to comment.