135 lines
5.3 KiB
Rust
135 lines
5.3 KiB
Rust
// Copyright 2022 Zinc Labs Inc. and Contributors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use chrono::{DateTime, SecondsFormat, Utc};
|
|
use std::io::Result;
|
|
use std::process::Command;
|
|
|
|
fn main() -> Result<()> {
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=proto");
|
|
|
|
tonic_build::configure()
|
|
.type_attribute("FileList", "#[derive(Eq)]")
|
|
.type_attribute("FileList", "#[derive(serde::Serialize)]")
|
|
.type_attribute("FileKey", "#[derive(Eq)]")
|
|
.type_attribute("FileKey", "#[derive(serde::Serialize)]")
|
|
.type_attribute("FileDescriptor", "#[derive(Eq)]")
|
|
.type_attribute("FileDescriptor", "#[derive(serde::Serialize)]")
|
|
.type_attribute("FileMeta", "#[derive(Eq)]")
|
|
.type_attribute("FileMeta", "#[derive(serde::Serialize)]")
|
|
.type_attribute("Job", "#[derive(Eq)]")
|
|
.type_attribute("Job", "#[derive(serde::Serialize)]")
|
|
.type_attribute("Partition", "#[derive(Eq)]")
|
|
.type_attribute("SearchQuery", "#[derive(Eq)]")
|
|
.type_attribute("SearchRequest", "#[derive(Eq)]")
|
|
.type_attribute("SearchResponse", "#[derive(Eq)]")
|
|
.type_attribute("SearchResponse", "#[derive(serde::Serialize)]")
|
|
.type_attribute("SearchAggRequest", "#[derive(Eq)]")
|
|
.type_attribute("SearchAggRequest", "#[derive(serde::Serialize)]")
|
|
.type_attribute("SearchAggResponse", "#[derive(Eq)]")
|
|
.type_attribute("SearchAggResponse", "#[derive(serde::Serialize)]")
|
|
.type_attribute("Series", "#[derive(serde::Serialize)]")
|
|
.type_attribute("Label", "#[derive(serde::Serialize)]")
|
|
.type_attribute("Sample", "#[derive(serde::Serialize)]")
|
|
.type_attribute("MetricsQueryStmt", "#[derive(serde::Serialize)]")
|
|
.type_attribute("MetricsQueryRequest", "#[derive(serde::Serialize)]")
|
|
.type_attribute("MetricsQueryResponse", "#[derive(serde::Serialize)]")
|
|
.type_attribute("ScanStats", "#[derive(Eq)]")
|
|
.type_attribute("ScanStats", "#[derive(serde::Serialize)]")
|
|
.compile(
|
|
&[
|
|
"proto/cluster/common.proto",
|
|
"proto/cluster/event.proto",
|
|
"proto/cluster/metrics.proto",
|
|
"proto/cluster/search.proto",
|
|
],
|
|
&["proto"],
|
|
)
|
|
.unwrap();
|
|
|
|
let mut config = prost_build::Config::new();
|
|
config
|
|
.type_attribute(
|
|
"MetricMetadata",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute(
|
|
"MetricType",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute("Sample", "#[derive(serde::Deserialize,serde::Serialize)]")
|
|
.type_attribute("Exemplar", "#[derive(serde::Deserialize,serde::Serialize)]")
|
|
.type_attribute(
|
|
"Histogram",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute(
|
|
"BucketSpan",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute(
|
|
"TimeSeries",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute("Label", "#[derive(serde::Deserialize,serde::Serialize)]")
|
|
.type_attribute("Labels", "#[derive(serde::Deserialize,serde::Serialize)]")
|
|
.type_attribute(
|
|
"LabelMatcher",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute(
|
|
"ReadHints",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute("Chunk", "#[derive(serde::Deserialize,serde::Serialize)]")
|
|
.type_attribute(
|
|
"ChunkedSeries",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute(
|
|
"zero_count",
|
|
"#[derive(serde::Deserialize,serde::Serialize)]",
|
|
)
|
|
.type_attribute("count", "#[derive(serde::Deserialize,serde::Serialize)]");
|
|
config.compile_protos(
|
|
&[
|
|
"proto/prometheus/remote.proto",
|
|
"proto/prometheus/types.proto",
|
|
],
|
|
&["proto"],
|
|
)?;
|
|
|
|
// build information
|
|
let output = Command::new("git")
|
|
.args(["describe", "--tags", "--abbrev=0"])
|
|
.output()
|
|
.unwrap();
|
|
let git_tag = String::from_utf8(output.stdout).unwrap();
|
|
println!("cargo:rustc-env=GIT_VERSION={git_tag}");
|
|
|
|
let output = Command::new("git")
|
|
.args(["rev-parse", "HEAD"])
|
|
.output()
|
|
.unwrap();
|
|
let git_commit = String::from_utf8(output.stdout).unwrap();
|
|
println!("cargo:rustc-env=GIT_COMMIT_HASH={git_commit}");
|
|
|
|
let now: DateTime<Utc> = Utc::now();
|
|
let build_date = now.to_rfc3339_opts(SecondsFormat::Secs, true);
|
|
println!("cargo:rustc-env=GIT_BUILD_DATE={build_date}");
|
|
|
|
Ok(())
|
|
}
|