Fix fs_util not working due to workunit init (pantsbuild#17229)

TansyArron · Oct 15, 2022 · 698fa0c · 698fa0c
1 parent 89ef823
commit 698fa0c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/src/rust/engine/Cargo.lock b/src/rust/engine/Cargo.lock
diff --git a/src/rust/engine/fs/fs_util/Cargo.toml b/src/rust/engine/fs/fs_util/Cargo.toml
@@ -14,6 +14,7 @@ grpc_util = { path = "../../grpc_util" }
 fs = { path = ".." }
 futures = "0.3"
 hashing = { path = "../../hashing" }
+log = "0.4"
 parking_lot = "0.12"
 prost = "0.9"
 rand = "0.8"

diff --git a/src/rust/engine/fs/fs_util/src/main.rs b/src/rust/engine/fs/fs_util/src/main.rs
@@ -50,6 +50,7 @@ use std::collections::BTreeMap;
 use store::{
   Snapshot, SnapshotOps, Store, StoreError, StoreFileByDigest, SubsetParams, UploadSummary,
 };
+use workunit_store::WorkunitStore;
 
 #[derive(Debug)]
 enum ExitCode {
@@ -84,6 +85,8 @@ struct SummaryWithDigest {
 #[tokio::main]
 async fn main() {
   env_logger::init();
+  let workunit_store = WorkunitStore::new(false, log::Level::Debug);
+  workunit_store.init_thread_state(None);
 
   match execute(
     &Command::new("fs_util")