Jonas Schäfer created
tokio-xmpp/Cargo.toml | 2
tokio-xmpp/examples/keep_connection.rs | 83 ++++++++++++++++++++++++++++
2 files changed, 84 insertions(+), 1 deletion(-)
@@ -40,7 +40,7 @@ ktls = { version = "6", optional = true }
[dev-dependencies]
env_logger = { version = "0.11", default-features = false, features = ["auto-color", "humantime"] }
# this is needed for echo-component example
-tokio = { version = "1", features = ["test-util"] }
+tokio = { version = "1", features = ["signal", "test-util"] }
tokio-xmpp = { path = ".", features = ["insecure-tcp"]}
[features]
@@ -0,0 +1,83 @@
+// Copyright (c) 2024 Jonas Schäfer <jonas@zombofant.net>
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+//! Keep a connection alive
+//!
+//! This example demonstrates that tokio_xmpp will keep a connection alive
+//! as good as it can, transparently reconnecting on interruptions of the TCP
+//! stream.
+
+use std::env::args;
+use std::process::exit;
+use std::str::FromStr;
+use std::time::Duration;
+
+use rand::{thread_rng, Rng};
+
+use futures::StreamExt;
+
+use tokio_xmpp::{
+ connect::{DnsConfig, StartTlsServerConnector},
+ parsers::{
+ iq::Iq,
+ jid::{BareJid, Jid},
+ ping,
+ },
+ stanzastream::StanzaStream,
+ xmlstream::Timeouts,
+};
+
+#[tokio::main]
+async fn main() {
+ env_logger::init();
+
+ let args: Vec<String> = args().collect();
+ if args.len() != 3 {
+ println!("Usage: {} <jid> <password>", args[0]);
+ exit(1);
+ }
+ let jid = BareJid::from_str(&args[1]).expect(&format!("Invalid JID: {}", &args[1]));
+ let password = &args[2];
+
+ let mut timeouts = Timeouts::tight();
+ timeouts.read_timeout = Duration::new(5, 0);
+
+ let mut stream = StanzaStream::new_c2s(
+ StartTlsServerConnector::from(DnsConfig::UseSrv {
+ host: jid.domain().as_str().to_owned(),
+ srv: "_xmpp-client._tcp".to_owned(),
+ fallback_port: 5222,
+ }),
+ jid.clone().into(),
+ password.clone(),
+ timeouts,
+ 16,
+ );
+ let domain: Jid = jid.domain().to_owned().into();
+ let mut ping_timer = tokio::time::interval(Duration::new(5, 0));
+ let mut ping_ctr: u64 = thread_rng().gen();
+ let signal = tokio::signal::ctrl_c();
+ tokio::pin!(signal);
+ loop {
+ tokio::select! {
+ _ = &mut signal => {
+ log::info!("Ctrl+C pressed, shutting down cleanly.");
+ break;
+ }
+ _ = ping_timer.tick() => {
+ log::info!("sending ping for fun & profit");
+ ping_ctr = ping_ctr.wrapping_add(1);
+ let mut iq = Iq::from_get(format!("ping-{}", ping_ctr), ping::Ping);
+ iq.to = Some(domain.clone());
+ stream.send(Box::new(iq.into())).await;
+ }
+ ev = stream.next() => {
+ log::info!("{:?}", ev);
+ }
+ }
+ }
+ stream.close().await;
+}