Add reqest properties for azure_kusto_data crate

.gitignore

@@ -20,3 +20,4 @@ __queuestorage__
 *.iml
 *.bat
 *.ps1
+.env

sdk/core/Cargo.toml

@@ -23,9 +23,11 @@ http = "0.2"
 hyper = { version = "0.14", optional = true }
 hyper-rustls = { version = "0.23", optional = true }
 log = "0.4"
-oauth2 = { version = "4.0", default-features=false }
+oauth2 = { version = "4.0", default-features = false }
 rand = "0.8"
-reqwest = { version = "0.11", features = ["stream"], default-features=false, optional = true }
+reqwest = { version = "0.11", features = [
+  "stream",
+], default-features = false, optional = true }
 serde = "1.0"
 serde_derive = "1.0"
 serde_json = "1.0"
@@ -48,6 +50,7 @@ tokio = { version = "1", features = ["default"] }
 [features]
 default = ["enable_reqwest"]
 enable_reqwest = ["reqwest/default-tls", "oauth2/native-tls"]
+enable_reqwest_gzip = ["reqwest/gzip"]
 enable_reqwest_rustls = ["reqwest/rustls-tls", "oauth2/rustls-tls"]
 enable_hyper = ["hyper", "hyper-rustls"]
 test_e2e = []

sdk/core/src/headers/mod.rs

@@ -211,6 +211,7 @@ pub fn add_mandatory_header<T: Header>(item: &T, builder: Builder) -> Builder {
 
 pub const ACCOUNT_KIND: &str = "x-ms-account-kind";
 pub const ACTIVITY_ID: &str = "x-ms-activity-id";
+pub const APP: &str = "x-ms-app";
 pub const APPEND_POSITION: &str = "x-ms-blob-condition-appendpos";
 pub const BLOB_ACCESS_TIER: &str = "x-ms-access-tier";
 pub const BLOB_CONTENT_LENGTH: &str = "x-ms-blob-content-length";
@@ -219,6 +220,7 @@ pub const BLOB_SEQUENCE_NUMBER: &str = "x-ms-blob-sequence-number";
 pub const BLOB_TYPE: &str = "x-ms-blob-type";
 pub const CACHE_CONTROL: &str = "x-ms-blob-cache-control";
 pub const CLIENT_REQUEST_ID: &str = "x-ms-client-request-id";
+pub const CLIENT_VERSION: &str = "x-ms-client-version";
 pub const CONTENT_DISPOSITION: &str = "x-ms-blob-content-disposition";
 pub const CONTINUATION: &str = "x-ms-continuation";
 pub const COPY_COMPLETION_TIME: &str = "x-ms-copy-completion-time";
@@ -262,4 +264,5 @@ pub const SOURCE_IF_MODIFIED_SINCE: &str = "x-ms-source-if-modified-since";
 pub const SOURCE_IF_NONE_MATCH: &str = "x-ms-source-if-none-match";
 pub const SOURCE_IF_UNMODIFIED_SINCE: &str = "x-ms-source-if-unmodified-since";
 pub const SOURCE_LEASE_ID: &str = "x-ms-source-lease-id";
+pub const USER: &str = "x-ms-user";
 pub const VERSION: &str = "x-ms-version";

sdk/core/src/request_options/accept.rs

@@ -0,0 +1,39 @@
+use crate::headers::{self, Header};
+
+/// Advertises which content types the client is able to understand.
+///
+/// The Accept request HTTP header advertises which content types, expressed
+/// as MIME types, the client is able to understand. Using content
+/// negotiation, the server then selects one of the proposals, uses it and
+/// informs the client of its choice with the Content-Type response header.
+/// Browsers set adequate values for this header depending of the context
+/// where the request is done: when fetching a CSS stylesheet a different
+/// value is set for the request than when fetching an image, video or a
+/// script.
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct Accept(String);
+
+impl Accept {
+    pub fn new(s: String) -> Self {
+        Self(s)
+    }
+}
+
+impl<S> From<S> for Accept
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for Accept {
+    fn name(&self) -> headers::HeaderName {
+        http::header::ACCEPT.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/core/src/request_options/accept_encoding.rs

@@ -0,0 +1,40 @@
+use crate::headers::{self, Header};
+
+/// Advertises which content encoding the client is able to understand.
+///
+/// The Accept-Encoding request HTTP header advertises which content
+/// encoding, usually a compression algorithm, the client is able to
+/// understand. Using content negotiation, the server selects one of the
+/// proposals, uses it and informs the client of its choice with the
+/// Content-Encoding response header.
+///
+/// Even if both the client and the server supports the same compression
+/// algorithms, the server may choose not to compress the body of a
+/// response, if the identity value is also acceptable.
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct AcceptEncoding(String);
+
+impl AcceptEncoding {
+    pub fn new(s: String) -> Self {
+        Self(s)
+    }
+}
+
+impl<S> From<S> for AcceptEncoding
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for AcceptEncoding {
+    fn name(&self) -> headers::HeaderName {
+        http::header::ACCEPT_ENCODING.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/core/src/request_options/activity_id.rs

@@ -1,15 +1,24 @@
 use crate::headers::{self, Header};
 
-#[derive(Debug, Clone, Copy)]
-pub struct ActivityId<'a>(&'a str);
+#[derive(Debug, Clone)]
+pub struct ActivityId(String);
 
-impl<'a> ActivityId<'a> {
-    pub fn new(id: &'a str) -> Self {
+impl ActivityId {
+    pub fn new(id: String) -> Self {
         Self(id)
     }
 }
 
-impl<'a> Header for ActivityId<'a> {
+impl<S> From<S> for ActivityId
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for ActivityId {
     fn name(&self) -> headers::HeaderName {
         headers::ACTIVITY_ID.into()
     }

sdk/core/src/request_options/app.rs

@@ -0,0 +1,30 @@
+use crate::headers::{self, Header};
+
+/// The (friendly) name of the application making the request
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct App(String);
+
+impl App {
+    pub fn new(s: String) -> Self {
+        Self(s)
+    }
+}
+
+impl<S> From<S> for App
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for App {
+    fn name(&self) -> headers::HeaderName {
+        headers::APP.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/core/src/request_options/client_request_id.rs

@@ -1,6 +1,7 @@
 use crate::headers;
 use crate::Header;
 
+/// A unique identifier for the request
 #[derive(Debug, Clone)]
 pub struct ClientRequestId(String);
 
@@ -10,15 +11,12 @@ impl ClientRequestId {
     }
 }
 
-impl From<String> for ClientRequestId {
-    fn from(client_request_id: String) -> Self {
-        Self::new(client_request_id)
-    }
-}
-
-impl From<&str> for ClientRequestId {
-    fn from(client_request_id: &str) -> Self {
-        Self::new(client_request_id.into())
+impl<S> From<S> for ClientRequestId
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
     }
 }
 

sdk/core/src/request_options/client_version.rs

@@ -0,0 +1,31 @@
+use crate::headers;
+use crate::Header;
+
+/// The (friendly) version identifier for the client making the request
+#[derive(Debug, Clone)]
+pub struct ClientVersion(String);
+
+impl ClientVersion {
+    pub fn new(client_request_id: String) -> Self {
+        Self(client_request_id)
+    }
+}
+
+impl<S> From<S> for ClientVersion
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for ClientVersion {
+    fn name(&self) -> headers::HeaderName {
+        headers::CLIENT_VERSION.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/core/src/request_options/mod.rs

@@ -1,5 +1,9 @@
+mod accept;
+mod accept_encoding;
 mod activity_id;
+mod app;
 mod client_request_id;
+mod client_version;
 mod content_disposition;
 mod content_encoding;
 mod content_language;
@@ -27,10 +31,16 @@ mod sequence_number;
 mod sequence_number_condition;
 mod source_lease_id;
 mod timeout;
+mod user;
 mod user_agent;
+mod version;
 
+pub use accept::Accept;
+pub use accept_encoding::AcceptEncoding;
 pub use activity_id::ActivityId;
+pub use app::App;
 pub use client_request_id::ClientRequestId;
+pub use client_version::ClientVersion;
 pub use content_disposition::ContentDisposition;
 pub use content_encoding::ContentEncoding;
 pub use content_language::ContentLanguage;
@@ -58,4 +68,6 @@ pub use sequence_number::SequenceNumber;
 pub use sequence_number_condition::SequenceNumberCondition;
 pub use source_lease_id::SourceLeaseId;
 pub use timeout::Timeout;
+pub use user::User;
 pub use user_agent::UserAgent;
+pub use version::Version;

sdk/core/src/request_options/user.rs

@@ -0,0 +1,30 @@
+use crate::headers::{self, Header};
+
+/// The (friendly) name of the user making the request
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct User(String);
+
+impl User {
+    pub fn new(s: String) -> Self {
+        Self(s)
+    }
+}
+
+impl<S> From<S> for User
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for User {
+    fn name(&self) -> headers::HeaderName {
+        headers::USER.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/core/src/request_options/version.rs

@@ -0,0 +1,29 @@
+use crate::headers::{self, Header};
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct Version(String);
+
+impl Version {
+    pub fn new(s: String) -> Self {
+        Self(s)
+    }
+}
+
+impl<S> From<S> for Version
+where
+    S: Into<String>,
+{
+    fn from(s: S) -> Self {
+        Self(s.into())
+    }
+}
+
+impl Header for Version {
+    fn name(&self) -> headers::HeaderName {
+        headers::VERSION.into()
+    }
+
+    fn value(&self) -> headers::HeaderValue {
+        self.0.to_owned().into()
+    }
+}

sdk/data_kusto/Cargo.toml

@@ -0,0 +1,40 @@
+[package]
+name = "azure_data_kusto"
+version = "0.1.0"
+description = "Rust wrappers around Microsoft Azure REST APIs - Azure Data Explorer"
+readme = "README.md"
+license = "MIT"
+edition = "2021"
+repository = "https://github.com/azure/azure-sdk-for-rust"
+homepage = "https://github.com/azure/azure-sdk-for-rust"
+documentation = "https://docs.rs/azure_data_kusto"
+keywords = ["sdk", "azure", "kusto", "azure-data-explorer"]
+categories = ["api-bindings"]
+
+[dependencies]
+arrow = { version = "9", optional = true }
+azure_core = { path = "../core", version = "0.1", features = [
+  "enable_reqwest",
+  "enable_reqwest_gzip",
+] }
+azure_identity = { path = "../identity", version = "0.1" }
+async-trait = "0.1"
+async-convert = "1"
+bytes = "1"
+futures = "0.3"
+http = "0.2"
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1"
+thiserror = "1"
+lazy_static = "1.4.0"
+hashbrown = "0.12.0"
+
+[dev-dependencies]
+env_logger = "0.9"
+tokio = { version = "1", features = ["macros"] }
+
+[features]
+default = ["arrow"]
+mock_transport_framework = ["azure_core/mock_transport_framework"]
+into_future = []
+test_e2e = []

sdk/data_kusto/README.md

@@ -0,0 +1,8 @@
+# Azure SDK for Rust - Azure Kusto crate
+
+## The Kusto crate.
+
+`azure-data-kusto` offers functionality needed to interact with Azure Data Explorer (Kusto) from Rust.
+As an abstraction over the [Azure Data Explorer REST API](https://docs.microsoft.com/en-us/azure/data-explorer/kusto/api/rest/)
+
+For usage have a look at the [examples](https://github.com/Azure/azure-sdk-for-rust/tree/main/sdk/data_kusto/examples).