diff --git a/src/cmd/build.rs b/src/cmd/build.rs index 15d858e847..8547bbd1d9 100644 --- a/src/cmd/build.rs +++ b/src/cmd/build.rs @@ -12,8 +12,18 @@ // See the License for the specific language governing permissions and // limitations under the License. +const DEFAULT_VALUE: &str = "unknown"; fn main() { - build_data::set_GIT_BRANCH(); - build_data::set_GIT_COMMIT(); - build_data::set_GIT_DIRTY(); + println!( + "cargo:rustc-env=GIT_COMMIT={}", + build_data::get_git_commit().unwrap_or_else(|_| DEFAULT_VALUE.to_string()) + ); + println!( + "cargo:rustc-env=GIT_BRANCH={}", + build_data::get_git_branch().unwrap_or_else(|_| DEFAULT_VALUE.to_string()) + ); + println!( + "cargo:rustc-env=GIT_DIRTY={}", + build_data::get_git_dirty().map_or(DEFAULT_VALUE.to_string(), |v| v.to_string()) + ); } diff --git a/src/cmd/src/bin/greptime.rs b/src/cmd/src/bin/greptime.rs index 578bee7e3b..5174591142 100644 --- a/src/cmd/src/bin/greptime.rs +++ b/src/cmd/src/bin/greptime.rs @@ -77,7 +77,9 @@ fn print_version() -> &'static str { "\ncommit: ", env!("GIT_COMMIT"), "\ndirty: ", - env!("GIT_DIRTY") + env!("GIT_DIRTY"), + "\nversion: ", + env!("CARGO_PKG_VERSION") ) }