Testing

getdozer · Mar 6, 2024 · 570dd9a · 570dd9a
1 parent 1e945ed
commit 570dd9a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/dozer-ingestion/aerospike/src/connector.rs b/dozer-ingestion/aerospike/src/connector.rs
@@ -166,6 +166,15 @@ impl AerospikeConnector {
 
         Ok(HttpServer::new(move || {
             App::new()
+                .app_data(web::JsonConfig::default().error_handler(|err, _req| {
+                    actix_web::error::InternalError::from_response(
+                        "",
+                        HttpResponse::BadRequest()
+                            .content_type("application/json")
+                            .body(format!(r#"{{"error":"{}"}}"#, err)),
+                    )
+                    .into()
+                }))
                 .app_data(web::Data::new(server_state.clone()))
                 .service(healthcheck)
                 .service(healthcheck_batch)

diff --git a/dozer-sink-oracle/src/lib.rs b/dozer-sink-oracle/src/lib.rs
@@ -565,6 +565,7 @@ enum OpKind {
 impl OracleSink {
     fn exec_batch(&mut self) -> oracle::Result<()> {
         debug!("Executing batch of size {}", self.batch_params.len());
+        let started = std::time::Instant::now();
         let mut batch = self
             .conn
             .batch(&self.merge_statement, self.batch_params.len())
@@ -587,6 +588,7 @@ impl OracleSink {
             batch.append_row(&[])?;
         }
         batch.execute()?;
+        info!("Execution took {:?}", started.elapsed());
         Ok(())
     }