rename the new pub type (requires moving the internal DTrace type with the same name)

Author: davepacheco

dropshot/src/dtrace.rs

@@ -0,0 +1,50 @@
+// Copyright 2023 Oxide Computer Company
+//! DTrace probes and support
+
+// The `usdt` crate may require nightly, enabled if our consumer is enabling
+// DTrace probes.
+#![cfg_attr(all(feature = "usdt-probes", usdt_need_asm), feature(asm))]
+#![cfg_attr(
+    all(feature = "usdt-probes", target_os = "macos", usdt_need_asm_sym),
+    feature(asm_sym)
+)]
+
+#[derive(Debug, Clone, serde::Serialize)]
+pub(crate) struct RequestInfo {
+    id: String,
+    local_addr: std::net::SocketAddr,
+    remote_addr: std::net::SocketAddr,
+    method: String,
+    path: String,
+    query: Option<String>,
+}
+
+#[derive(Debug, Clone, serde::Serialize)]
+pub(crate) struct ResponseInfo {
+    id: String,
+    local_addr: std::net::SocketAddr,
+    remote_addr: std::net::SocketAddr,
+    status_code: u16,
+    message: String,
+}
+
+#[cfg(feature = "usdt-probes")]
+#[usdt::provider(provider = "dropshot")]
+mod probes {
+    use super::{RequestInfo, ResponseInfo};
+    fn request__start(_: &RequestInfo) {}
+    fn request__done(_: &ResponseInfo) {}
+}
+
+/// The result of registering a server's DTrace USDT probes.
+#[derive(Debug, Clone, PartialEq)]
+pub enum ProbeRegistration {
+    /// The probes are explicitly disabled at compile time.
+    Disabled,
+
+    /// Probes were successfully registered.
+    Succeeded,
+
+    /// Registration failed, with an error message explaining the cause.
+    Failed(String),
+}

dropshot/src/extractor/mod.rs

@@ -31,7 +31,7 @@ use std::fmt::Debug;
 
 mod common;
 
-use crate::RequestHeader;
+use crate::RequestInfo;
 pub use common::ExclusiveExtractor;
 pub use common::ExtractorMetadata;
 pub use common::RequestExtractor;
@@ -58,7 +58,7 @@ impl<QueryType: DeserializeOwned + JsonSchema + Send + Sync> Query<QueryType> {
 /// Given an HTTP request, pull out the query string and attempt to deserialize
 /// it as an instance of `QueryType`.
 fn http_request_load_query<QueryType>(
-    request: &RequestHeader,
+    request: &RequestInfo,
 ) -> Result<Query<QueryType>, HttpError>
 where
     QueryType: DeserializeOwned + JsonSchema + Send + Sync,

dropshot/src/handler.rs

@@ -85,23 +85,22 @@ pub struct RequestContext<Context: ServerContext> {
     pub log: Logger,
 
     /// basic request information (method, URI, etc.)
-    pub request: RequestHeader,
+    pub request: RequestInfo,
 }
 
 // This is deliberately as close to compatible with `hyper::Request` as
 // reasonable.
-// XXX-dap TODO This could use a better name.
 #[derive(Debug)]
-pub struct RequestHeader {
+pub struct RequestInfo {
     method: http::Method,
     uri: http::Uri,
     version: http::Version,
     headers: http::HeaderMap<http::HeaderValue>,
 }
 
-impl<B> From<&hyper::Request<B>> for RequestHeader {
+impl<B> From<&hyper::Request<B>> for RequestInfo {
     fn from(request: &hyper::Request<B>) -> Self {
-        RequestHeader {
+        RequestInfo {
             method: request.method().clone(),
             uri: request.uri().clone(),
             version: request.version().clone(),
@@ -110,7 +109,7 @@ impl<B> From<&hyper::Request<B>> for RequestHeader {
     }
 }
 
-impl RequestHeader {
+impl RequestInfo {
     pub fn method(&self) -> &http::Method {
         &self.method
     }

dropshot/src/lib.rs

@@ -504,8 +504,8 @@
 //! Dropshot optionally exposes two DTrace probes, `request_start` and
 //! `request_finish`. These provide detailed information about each request,
 //! such as their ID, the local and remote IPs, and the response information.
-//! See the [`RequestInfo`] and [`ResponseInfo`] types for a complete listing
-//! of what's available.
+//! See the dropshot::dtrace::RequestInfo` and `dropshot::dtrae::ResponseInfo`
+//! types for a complete listing of what's available.
 //!
 //! These probes are implemented via the [`usdt`] crate. They may require a
 //! nightly toolchain if built on macOS prior to Rust version 1.66. Otherwise a
@@ -543,56 +543,10 @@
 // Clippy's style advice is definitely valuable, but not worth the trouble for
 // automated enforcement.
 #![allow(clippy::style)]
-// The `usdt` crate may require nightly, enabled if our consumer is enabling
-// DTrace probes.
-#![cfg_attr(all(feature = "usdt-probes", usdt_need_asm), feature(asm))]
-#![cfg_attr(
-    all(feature = "usdt-probes", target_os = "macos", usdt_need_asm_sym),
-    feature(asm_sym)
-)]
-
-#[derive(Debug, Clone, serde::Serialize)]
-pub(crate) struct RequestInfo {
-    id: String,
-    local_addr: std::net::SocketAddr,
-    remote_addr: std::net::SocketAddr,
-    method: String,
-    path: String,
-    query: Option<String>,
-}
-
-#[derive(Debug, Clone, serde::Serialize)]
-pub(crate) struct ResponseInfo {
-    id: String,
-    local_addr: std::net::SocketAddr,
-    remote_addr: std::net::SocketAddr,
-    status_code: u16,
-    message: String,
-}
-
-#[cfg(feature = "usdt-probes")]
-#[usdt::provider(provider = "dropshot")]
-mod probes {
-    use crate::{RequestInfo, ResponseInfo};
-    fn request__start(_: &RequestInfo) {}
-    fn request__done(_: &ResponseInfo) {}
-}
-
-/// The result of registering a server's DTrace USDT probes.
-#[derive(Debug, Clone, PartialEq)]
-pub enum ProbeRegistration {
-    /// The probes are explicitly disabled at compile time.
-    Disabled,
-
-    /// Probes were successfully registered.
-    Succeeded,
-
-    /// Registration failed, with an error message explaining the cause.
-    Failed(String),
-}
 
 mod api_description;
 mod config;
+mod dtrace;
 mod error;
 mod extractor;
 mod from_map;
@@ -626,6 +580,7 @@ pub use api_description::TagDetails;
 pub use api_description::TagExternalDocs;
 pub use config::ConfigDropshot;
 pub use config::ConfigTls;
+pub use dtrace::ProbeRegistration;
 pub use error::HttpError;
 pub use error::HttpErrorResponseBody;
 pub use extractor::ExclusiveExtractor;
@@ -653,7 +608,7 @@ pub use handler::HttpResponseTemporaryRedirect;
 pub use handler::HttpResponseUpdatedNoContent;
 pub use handler::NoHeaders;
 pub use handler::RequestContext;
-pub use handler::RequestHeader;
+pub use handler::RequestInfo;
 pub use http_util::CONTENT_TYPE_JSON;
 pub use http_util::CONTENT_TYPE_NDJSON;
 pub use http_util::CONTENT_TYPE_OCTET_STREAM;

dropshot/src/server.rs

@@ -1,4 +1,4 @@
-// Copyright 2020 Oxide Computer Company
+// Copyright 2023 Oxide Computer Company
 //! Generic server-wide state and facilities
 
 use super::api_description::ApiDescription;
@@ -38,7 +38,7 @@ use tokio::net::{TcpListener, TcpStream};
 use tokio_rustls::{server::TlsStream, TlsAcceptor};
 use uuid::Uuid;
 
-use crate::RequestHeader;
+use crate::RequestInfo;
 use slog::Logger;
 
 // TODO Replace this with something else?
@@ -679,7 +679,7 @@ async fn http_request_handle_wrap<C: ServerContext>(
     #[cfg(feature = "usdt-probes")]
     probes::request__start!(|| {
         let uri = request.uri();
-        crate::RequestInfo {
+        crate::dtrace::RequestInfo {
             id: request_id.clone(),
             local_addr: server.local_addr,
             remote_addr,
@@ -710,7 +710,7 @@ async fn http_request_handle_wrap<C: ServerContext>(
 
             #[cfg(feature = "usdt-probes")]
             probes::request__done!(|| {
-                crate::ResponseInfo {
+                crate::dtrace::ResponseInfo {
                     id: request_id.clone(),
                     local_addr,
                     remote_addr,
@@ -771,7 +771,7 @@ async fn http_request_handle<C: ServerContext>(
         server.router.lookup_route(&method, uri.path().into())?;
     let rqctx = RequestContext {
         server: Arc::clone(&server),
-        request: RequestHeader::from(&request),
+        request: RequestInfo::from(&request),
         path_variables: lookup_result.variables,
         body_content_type: lookup_result.body_content_type,
         request_id: request_id.to_string(),

dropshot/src/websocket.rs

@@ -297,7 +297,7 @@ mod tests {
     use crate::router::HttpRouter;
     use crate::server::{DropshotState, ServerConfig};
     use crate::{
-        ExclusiveExtractor, HttpError, RequestContext, RequestHeader,
+        ExclusiveExtractor, HttpError, RequestContext, RequestInfo,
         WebsocketUpgrade,
     };
     use http::Request;
@@ -332,7 +332,7 @@ mod tests {
                 ),
                 tls_acceptor: None,
             }),
-            request: RequestHeader::from(&request),
+            request: RequestInfo::from(&request),
             path_variables: Default::default(),
             body_content_type: Default::default(),
             request_id: "".to_string(),