mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-11-30 00:34:26 +01:00
makes the analytics available for all the routes
This commit is contained in:
parent
664d09e86a
commit
d72c887422
@ -151,6 +151,7 @@ mod segment {
|
|||||||
impl super::Analytics for SegmentAnalytics {
|
impl super::Analytics for SegmentAnalytics {
|
||||||
fn publish(&'static self, event_name: String, send: Value) {
|
fn publish(&'static self, event_name: String, send: Value) {
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
|
println!("ANALYTICS pushing {} in the batcher", event_name);
|
||||||
let _ = self
|
let _ = self
|
||||||
.batcher
|
.batcher
|
||||||
.lock()
|
.lock()
|
||||||
@ -162,6 +163,7 @@ mod segment {
|
|||||||
..Default::default()
|
..Default::default()
|
||||||
})
|
})
|
||||||
.await;
|
.await;
|
||||||
|
println!("ANALYTICS {} pushed", event_name);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -205,6 +207,6 @@ impl Display for MockAnalytics {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[async_trait::async_trait]
|
#[async_trait::async_trait]
|
||||||
pub trait Analytics: Display {
|
pub trait Analytics: Display + Sync + Send {
|
||||||
fn publish(&'static self, event_name: String, send: Value);
|
fn publish(&'static self, event_name: String, send: Value);
|
||||||
}
|
}
|
||||||
|
@ -12,6 +12,7 @@ use std::time::Duration;
|
|||||||
use crate::error::MeilisearchHttpError;
|
use crate::error::MeilisearchHttpError;
|
||||||
use crate::extractors::authentication::AuthConfig;
|
use crate::extractors::authentication::AuthConfig;
|
||||||
use actix_web::error::JsonPayloadError;
|
use actix_web::error::JsonPayloadError;
|
||||||
|
use analytics::Analytics;
|
||||||
use error::PayloadError;
|
use error::PayloadError;
|
||||||
use http::header::CONTENT_TYPE;
|
use http::header::CONTENT_TYPE;
|
||||||
pub use option::Opt;
|
pub use option::Opt;
|
||||||
@ -73,10 +74,16 @@ pub fn setup_meilisearch(opt: &Opt) -> anyhow::Result<MeiliSearch> {
|
|||||||
meilisearch.build(opt.db_path.clone(), opt.indexer_options.clone())
|
meilisearch.build(opt.db_path.clone(), opt.indexer_options.clone())
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn configure_data(config: &mut web::ServiceConfig, data: MeiliSearch, opt: &Opt) {
|
pub fn configure_data(
|
||||||
|
config: &mut web::ServiceConfig,
|
||||||
|
data: MeiliSearch,
|
||||||
|
opt: &Opt,
|
||||||
|
analytics: &'static dyn Analytics,
|
||||||
|
) {
|
||||||
let http_payload_size_limit = opt.http_payload_size_limit.get_bytes() as usize;
|
let http_payload_size_limit = opt.http_payload_size_limit.get_bytes() as usize;
|
||||||
config
|
config
|
||||||
.app_data(data)
|
.app_data(data)
|
||||||
|
.app_data(web::Data::new(analytics))
|
||||||
.app_data(
|
.app_data(
|
||||||
web::JsonConfig::default()
|
web::JsonConfig::default()
|
||||||
.content_type(|mime| mime == mime::APPLICATION_JSON)
|
.content_type(|mime| mime == mime::APPLICATION_JSON)
|
||||||
@ -167,7 +174,7 @@ pub fn dashboard(config: &mut web::ServiceConfig, _enable_frontend: bool) {
|
|||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! create_app {
|
macro_rules! create_app {
|
||||||
($data:expr, $enable_frontend:expr, $opt:expr) => {{
|
($data:expr, $enable_frontend:expr, $opt:expr, $analytics:expr) => {{
|
||||||
use actix_cors::Cors;
|
use actix_cors::Cors;
|
||||||
use actix_web::middleware::TrailingSlash;
|
use actix_web::middleware::TrailingSlash;
|
||||||
use actix_web::App;
|
use actix_web::App;
|
||||||
@ -177,7 +184,7 @@ macro_rules! create_app {
|
|||||||
use meilisearch_http::{configure_auth, configure_data, dashboard};
|
use meilisearch_http::{configure_auth, configure_data, dashboard};
|
||||||
|
|
||||||
App::new()
|
App::new()
|
||||||
.configure(|s| configure_data(s, $data.clone(), &$opt))
|
.configure(|s| configure_data(s, $data.clone(), &$opt, $analytics))
|
||||||
.configure(|s| configure_auth(s, &$opt))
|
.configure(|s| configure_auth(s, &$opt))
|
||||||
.configure(routes::configure)
|
.configure(routes::configure)
|
||||||
.configure(|s| dashboard(s, $enable_frontend))
|
.configure(|s| dashboard(s, $enable_frontend))
|
||||||
|
@ -56,15 +56,20 @@ async fn main() -> anyhow::Result<()> {
|
|||||||
|
|
||||||
print_launch_resume(&opt, analytics);
|
print_launch_resume(&opt, analytics);
|
||||||
|
|
||||||
run_http(meilisearch, opt).await?;
|
run_http(meilisearch, opt, analytics).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn run_http(data: MeiliSearch, opt: Opt) -> anyhow::Result<()> {
|
async fn run_http(
|
||||||
|
data: MeiliSearch,
|
||||||
|
opt: Opt,
|
||||||
|
analytics: &'static dyn Analytics,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
let _enable_dashboard = &opt.env == "development";
|
let _enable_dashboard = &opt.env == "development";
|
||||||
let opt_clone = opt.clone();
|
let opt_clone = opt.clone();
|
||||||
let http_server = HttpServer::new(move || create_app!(data, _enable_dashboard, opt_clone))
|
let http_server =
|
||||||
|
HttpServer::new(move || create_app!(data, _enable_dashboard, opt_clone, analytics))
|
||||||
// Disable signals allows the server to terminate immediately when a user enter CTRL-C
|
// Disable signals allows the server to terminate immediately when a user enter CTRL-C
|
||||||
.disable_signals();
|
.disable_signals();
|
||||||
|
|
||||||
|
@ -2,7 +2,7 @@ use actix_web::{http::StatusCode, test};
|
|||||||
use meilisearch_lib::MeiliSearch;
|
use meilisearch_lib::MeiliSearch;
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
|
|
||||||
use meilisearch_http::{create_app, Opt};
|
use meilisearch_http::{analytics, create_app, Opt};
|
||||||
|
|
||||||
pub struct Service {
|
pub struct Service {
|
||||||
pub meilisearch: MeiliSearch,
|
pub meilisearch: MeiliSearch,
|
||||||
@ -11,7 +11,13 @@ pub struct Service {
|
|||||||
|
|
||||||
impl Service {
|
impl Service {
|
||||||
pub async fn post(&self, url: impl AsRef<str>, body: Value) -> (Value, StatusCode) {
|
pub async fn post(&self, url: impl AsRef<str>, body: Value) -> (Value, StatusCode) {
|
||||||
let app = test::init_service(create_app!(&self.meilisearch, true, &self.options)).await;
|
let app = test::init_service(create_app!(
|
||||||
|
&self.meilisearch,
|
||||||
|
true,
|
||||||
|
&self.options,
|
||||||
|
analytics::MockAnalytics::new(&self.options)
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::post()
|
let req = test::TestRequest::post()
|
||||||
.uri(url.as_ref())
|
.uri(url.as_ref())
|
||||||
@ -31,7 +37,13 @@ impl Service {
|
|||||||
url: impl AsRef<str>,
|
url: impl AsRef<str>,
|
||||||
body: impl AsRef<str>,
|
body: impl AsRef<str>,
|
||||||
) -> (Value, StatusCode) {
|
) -> (Value, StatusCode) {
|
||||||
let app = test::init_service(create_app!(&self.meilisearch, true, &self.options)).await;
|
let app = test::init_service(create_app!(
|
||||||
|
&self.meilisearch,
|
||||||
|
true,
|
||||||
|
&self.options,
|
||||||
|
analytics::MockAnalytics::new(&self.options)
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::post()
|
let req = test::TestRequest::post()
|
||||||
.uri(url.as_ref())
|
.uri(url.as_ref())
|
||||||
@ -47,7 +59,13 @@ impl Service {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn get(&self, url: impl AsRef<str>) -> (Value, StatusCode) {
|
pub async fn get(&self, url: impl AsRef<str>) -> (Value, StatusCode) {
|
||||||
let app = test::init_service(create_app!(&self.meilisearch, true, &self.options)).await;
|
let app = test::init_service(create_app!(
|
||||||
|
&self.meilisearch,
|
||||||
|
true,
|
||||||
|
&self.options,
|
||||||
|
analytics::MockAnalytics::new(&self.options)
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::get().uri(url.as_ref()).to_request();
|
let req = test::TestRequest::get().uri(url.as_ref()).to_request();
|
||||||
let res = test::call_service(&app, req).await;
|
let res = test::call_service(&app, req).await;
|
||||||
@ -59,7 +77,13 @@ impl Service {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn put(&self, url: impl AsRef<str>, body: Value) -> (Value, StatusCode) {
|
pub async fn put(&self, url: impl AsRef<str>, body: Value) -> (Value, StatusCode) {
|
||||||
let app = test::init_service(create_app!(&self.meilisearch, true, &self.options)).await;
|
let app = test::init_service(create_app!(
|
||||||
|
&self.meilisearch,
|
||||||
|
true,
|
||||||
|
&self.options,
|
||||||
|
analytics::MockAnalytics::new(&self.options)
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::put()
|
let req = test::TestRequest::put()
|
||||||
.uri(url.as_ref())
|
.uri(url.as_ref())
|
||||||
@ -74,7 +98,13 @@ impl Service {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn delete(&self, url: impl AsRef<str>) -> (Value, StatusCode) {
|
pub async fn delete(&self, url: impl AsRef<str>) -> (Value, StatusCode) {
|
||||||
let app = test::init_service(create_app!(&self.meilisearch, true, &self.options)).await;
|
let app = test::init_service(create_app!(
|
||||||
|
&self.meilisearch,
|
||||||
|
true,
|
||||||
|
&self.options,
|
||||||
|
analytics::MockAnalytics::new(&self.options)
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
|
||||||
let req = test::TestRequest::delete().uri(url.as_ref()).to_request();
|
let req = test::TestRequest::delete().uri(url.as_ref()).to_request();
|
||||||
let res = test::call_service(&app, req).await;
|
let res = test::call_service(&app, req).await;
|
||||||
|
@ -4,7 +4,7 @@ mod common;
|
|||||||
|
|
||||||
use crate::common::Server;
|
use crate::common::Server;
|
||||||
use actix_web::test;
|
use actix_web::test;
|
||||||
use meilisearch_http::create_app;
|
use meilisearch_http::{analytics, create_app};
|
||||||
use serde_json::{json, Value};
|
use serde_json::{json, Value};
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
@ -40,7 +40,8 @@ async fn error_json_bad_content_type() {
|
|||||||
let app = test::init_service(create_app!(
|
let app = test::init_service(create_app!(
|
||||||
&server.service.meilisearch,
|
&server.service.meilisearch,
|
||||||
true,
|
true,
|
||||||
&server.service.options
|
&server.service.options,
|
||||||
|
analytics::MockAnalytics::new(&server.service.options)
|
||||||
))
|
))
|
||||||
.await;
|
.await;
|
||||||
for route in routes {
|
for route in routes {
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
use crate::common::{GetAllDocumentsOptions, Server};
|
use crate::common::{GetAllDocumentsOptions, Server};
|
||||||
use actix_web::test;
|
use actix_web::test;
|
||||||
use chrono::DateTime;
|
use chrono::DateTime;
|
||||||
use meilisearch_http::create_app;
|
use meilisearch_http::{analytics, create_app};
|
||||||
use serde_json::{json, Value};
|
use serde_json::{json, Value};
|
||||||
|
|
||||||
/// This is the basic usage of our API and every other tests uses the content-type application/json
|
/// This is the basic usage of our API and every other tests uses the content-type application/json
|
||||||
@ -19,7 +19,8 @@ async fn add_documents_test_json_content_types() {
|
|||||||
let app = test::init_service(create_app!(
|
let app = test::init_service(create_app!(
|
||||||
&server.service.meilisearch,
|
&server.service.meilisearch,
|
||||||
true,
|
true,
|
||||||
&server.service.options
|
&server.service.options,
|
||||||
|
analytics::MockAnalytics::new(&server.service.options)
|
||||||
))
|
))
|
||||||
.await;
|
.await;
|
||||||
// post
|
// post
|
||||||
@ -63,7 +64,8 @@ async fn error_add_documents_test_bad_content_types() {
|
|||||||
let app = test::init_service(create_app!(
|
let app = test::init_service(create_app!(
|
||||||
&server.service.meilisearch,
|
&server.service.meilisearch,
|
||||||
true,
|
true,
|
||||||
&server.service.options
|
&server.service.options,
|
||||||
|
analytics::MockAnalytics::new(&server.service.options)
|
||||||
))
|
))
|
||||||
.await;
|
.await;
|
||||||
// post
|
// post
|
||||||
@ -129,7 +131,8 @@ async fn error_add_documents_test_no_content_type() {
|
|||||||
let app = test::init_service(create_app!(
|
let app = test::init_service(create_app!(
|
||||||
&server.service.meilisearch,
|
&server.service.meilisearch,
|
||||||
true,
|
true,
|
||||||
&server.service.options
|
&server.service.options,
|
||||||
|
analytics::MockAnalytics::new(&server.service.options)
|
||||||
))
|
))
|
||||||
.await;
|
.await;
|
||||||
// post
|
// post
|
||||||
|
Loading…
Reference in New Issue
Block a user