Merge branch 'main' into duncan-harvey/datadog-trace-utils-bump-hyper…

…-rustls-version

.github/CODEOWNERS

@@ -21,3 +21,4 @@ serverless              @Datadog/serverless
 sidecar                 @Datadog/libdatadog-php @Datadog/libdatadog-apm
 sidecar-ffi             @Datadog/libdatadog-php @Datadog/libdatadog-apm
 data-pipeline*/         @Datadog/libdatadog-apm
+ddsketch		@Datadog/libdatadog-apm @Datadog/libdatadog-telemetry

Cargo.toml

@@ -28,7 +28,8 @@ members = [
   "serverless",
   "bin_tests",
   "data-pipeline",
-  "data-pipeline-ffi"
+  "data-pipeline-ffi",
+  "ddsketch",
 ]
 # https://doc.rust-lang.org/cargo/reference/resolver.html#feature-resolver-version-2
 resolver = "2"

ddsketch/Cargo.toml

@@ -0,0 +1,19 @@
+[package]
+name = "datadog-ddsketch"
+description = "Minimal implementation of Datadog's DDSketch"
+edition.workspace = true
+version.workspace = true
+rust-version.workspace = true
+license.workspace = true
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+prost = "0.11.6"
+
+[build-dependencies]
+prost-build = { version = "0.11.9", optional = true  }
+protoc-bin-vendored = { version = "3.0.0", optional = true }
+
+[features]
+generate-protobuf = ["dep:prost-build", "dep:protoc-bin-vendored"]

ddsketch/build.rs

@@ -0,0 +1,52 @@
+// Copyright 2023-Present Datadog, Inc. https://www.datadoghq.com/
+// SPDX-License-Identifier: Apache-2.0
+
+use std::error::Error;
+
+#[cfg(feature = "generate-protobuf")]
+use std::{
+    env,
+    fs::File,
+    io::{Read, Write},
+    path::Path,
+};
+
+#[cfg(feature = "generate-protobuf")]
+const HEADER: &str = "// Copyright 2023-Present Datadog, Inc. https://www.datadoghq.com/
+// SPDX-License-Identifier: Apache-2.0
+
+";
+
+fn main() -> Result<(), Box<dyn Error>> {
+    #[cfg(feature = "generate-protobuf")]
+    {
+        // protoc is required to compile proto files. This uses protobuf_src to compile protoc
+        // from the source, setting the env var to tell prost_build where to find it.
+        std::env::set_var("PROTOC", protoc_bin_vendored::protoc_bin_path().unwrap());
+
+        let mut config = prost_build::Config::new();
+
+        let cur_working_dir = env::var("CARGO_MANIFEST_DIR")?;
+        let output_path = Path::new(&cur_working_dir).join("src");
+
+        config.out_dir(output_path.clone());
+
+        println!("cargo:rerun-if-changed=src/pb/DDSketch.proto");
+        config.compile_protos(&["src/pb/DDSketch.proto"], &["src/"])?;
+
+        prepend_to_file(HEADER.as_bytes(), &output_path.join("pb.rs"))?;
+    }
+
+    Ok(())
+}
+
+#[cfg(feature = "generate-protobuf")]
+fn prepend_to_file(data: &[u8], file_path: &Path) -> Result<(), Box<dyn Error>> {
+    let mut f = File::open(file_path)?;
+    let mut content = data.to_owned();
+    f.read_to_end(&mut content)?;
+
+    let mut f = File::create(file_path)?;
+    f.write_all(content.as_slice())?;
+    Ok(())
+}