Simple-pub-sub

A simple message broker implemented in rust.

The message frame looks like

So it's a 8 byte header followed by the topic and message.

Cli Usage

• Server:

  • Using Tcp socket:

  simple-pub-sub server tcp 0.0.0.0 6480 --log-level trace

  To use TLS:

  # Generate the server key and certificate
  openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem \
      -days 365 -nodes -subj "/CN=localhost"
  # generate the identity file.
  openssl pkcs12 -export -out identity.pfx -inkey key.pem -in cert.pem\
    -passout pass:password

  Start the server using these keys:

  simple-pub-sub server tcp -c identity.pfx -p password 0.0.0.0 6480 \
    --log-level trace

  • Using Unix socket:

  simple-pub-sub server unix /tmp/pubsub.sock --log-level trace

• Client:

  • Using Tcp socket:

  • subscribe:

    simple-pub-sub client subscribe the_topic tcp 0.0.0.0 6480 --log-level trace

    Using TLS:

    simple-pub-sub client subscribe the_topic tcp -c certs/cert.pem 0.0.0.0 6480\
        --log-level trace

  • publish:

    simple-pub-sub client publish the_topic the_message tcp 0.0.0.0 6480\
        --log-level info

    Using TLS:

    simple-pub-sub client publish the_topic the_message tcp -c certs/cert.pem\
     0.0.0.0 6480 --log-level trace

  • query:

    simple-pub-sub client query the_topic tcp 0.0.0.0 6480  --log-level trace

    Using TLS:

    simple-pub-sub client query the_topic tcp -c certs/cert.pem\
      0.0.0.0 6480  --log-level trace

  • Using Unix socket:

  • subscribe:

    simple-pub-sub client unix /tmp/pubsub.sock subscribe the_topic\
      --log-level trace

  • publish:

    simple-pub-sub client unix /tmp/pubsub.sock publish the_topic the_message\
      --log-level info

  • query:

    simple-pub-sub client unix /tmp/pubsub.sock query the_topic --log-level trace

API Usage

To subscribe

use simple-pub-sub

// define the on_message function (callback).
pub fn on_msg(topic: String, message: Vec<u8>) {
    println!("topic: {} message: {:?}", topic, message)
}
#[tokio::main]
async fn main() -> Result<(), String> {
    let client_type = simple_pub_sub::client::PubSubTcpClient {
        server: "localhost".to_string(),
        port: 6480,
    };
    // initialize the client.
    let mut client =
        simple_pub_sub::client::Client::new(simple_pub_sub::client::PubSubClient::Tcp(client_type));
    // set the callback function.
    client.on_message(on_msg);
    // connect the client.
    let _ = client.connect().await;
    // subscribe to the given topic.
    let _ = client.subscribe("abc".to_string()).await;
    Ok(())
}

To publish a message

use simple_pub_sub;
async fn main() -> Result<(), String> {
    let client_type = simple_pub_sub::client::PubSubTcpClient {
        server: "localhost".to_string(),
        port: 6480,
    };
    // initialize the client.
    let mut client =
        simple_pub_sub::client::Client::new(simple_pub_sub::client::PubSubClient::Tcp(client_type));
    // connect the client.
    let _ = client.connect().await;
    // subscribe to the given topic.
    let _ = client
        .publish(
            "abc".to_string(),
            "test message".to_string().into_bytes().to_vec(),
        )
        .await;
    Ok(())
}