-
-
Notifications
You must be signed in to change notification settings - Fork 198
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #132 from vaptu/master
feat: abstract streamhub message notifications
- Loading branch information
Showing
5 changed files
with
192 additions
and
121 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,122 @@ | ||
| use crate::notify::Notifier; | ||
| use reqwest::Client; | ||
| use async_trait::async_trait; | ||
| use crate::define::{StreamHubEventMessage}; | ||
|
|
||
| macro_rules! serialize_event { | ||
| ($message:expr) => {{ | ||
| let event_serialize_str = match serde_json::to_string(&$message) { | ||
| Ok(data) => { | ||
| log::info!("event data: {}", data); | ||
| data | ||
| } | ||
| Err(_) => String::from("empty body"), | ||
| }; | ||
| event_serialize_str | ||
| }}; | ||
| } | ||
|
|
||
|
|
||
| pub struct HttpNotifier { | ||
| request_client: Client, | ||
| on_publish_url: Option<String>, | ||
| on_unpublish_url: Option<String>, | ||
| on_play_url: Option<String>, | ||
| on_stop_url: Option<String>, | ||
| } | ||
|
|
||
| impl HttpNotifier { | ||
| pub fn new( | ||
| on_publish_url: Option<String>, | ||
| on_unpublish_url: Option<String>, | ||
| on_play_url: Option<String>, | ||
| on_stop_url: Option<String>, | ||
| ) -> Self { | ||
| Self { | ||
| request_client: reqwest::Client::new(), | ||
| on_publish_url, | ||
| on_unpublish_url, | ||
| on_play_url, | ||
| on_stop_url, | ||
| } | ||
| } | ||
| } | ||
|
|
||
| #[async_trait] | ||
| impl Notifier for HttpNotifier { | ||
| async fn on_publish_notify(&self, event: &StreamHubEventMessage) { | ||
| if let Some(on_publish_url) = &self.on_publish_url { | ||
| match self | ||
| .request_client | ||
| .post(on_publish_url) | ||
| .body(serialize_event!(event)) | ||
| .send() | ||
| .await | ||
| { | ||
| Err(err) => { | ||
| log::error!("on_publish error: {}", err); | ||
| } | ||
| Ok(response) => { | ||
| log::info!("on_publish success: {:?}", response); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| async fn on_unpublish_notify(&self, event: &StreamHubEventMessage) { | ||
| if let Some(on_unpublish_url) = &self.on_unpublish_url { | ||
| match self | ||
| .request_client | ||
| .post(on_unpublish_url) | ||
| .body(serialize_event!(event)) | ||
| .send() | ||
| .await | ||
| { | ||
| Err(err) => { | ||
| log::error!("on_unpublish error: {}", err); | ||
| } | ||
| Ok(response) => { | ||
| log::info!("on_unpublish success: {:?}", response); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| async fn on_play_notify(&self, event: &StreamHubEventMessage) { | ||
| if let Some(on_play_url) = &self.on_play_url { | ||
| match self | ||
| .request_client | ||
| .post(on_play_url) | ||
| .body(serialize_event!(event)) | ||
| .send() | ||
| .await | ||
| { | ||
| Err(err) => { | ||
| log::error!("on_play error: {}", err); | ||
| } | ||
| Ok(response) => { | ||
| log::info!("on_play success: {:?}", response); | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| async fn on_stop_notify(&self, event: &StreamHubEventMessage) { | ||
| if let Some(on_stop_url) = &self.on_stop_url { | ||
| match self | ||
| .request_client | ||
| .post(on_stop_url) | ||
| .body(serialize_event!(event)) | ||
| .send() | ||
| .await | ||
| { | ||
| Err(err) => { | ||
| log::error!("on_stop error: {}", err); | ||
| } | ||
| Ok(response) => { | ||
| log::info!("on_stop success: {:?}", response); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.