#Centipede Add WorkDir::ShardedFileInfo::IsShardPath() query

PiperOrigin-RevId: 578685041
google · Nov 2, 2023 · f375329 · f375329
1 parent ed5459a
commit f375329
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/centipede/workdir.cc b/centipede/workdir.cc
@@ -73,6 +73,13 @@ std::string WorkDir::ShardedFileInfo::AllShardsGlob() const {
   return absl::StrCat(prefix_, "*");
 }
 
+bool WorkDir::ShardedFileInfo::IsShardPath(std::string_view path) const {
+  // TODO(ussuri): This is as barebones as it can be right now. Possible
+  //  improvements: 1. Make `path` & `prefix_` absolute before comparing (or in
+  //  ctor for `prefix_`). 2. Add option to require the actual file's existence.
+  return absl::StartsWith(path, prefix_);
+}
+
 //------------------------------------------------------------------------------
 //                                 WorkDir
 

diff --git a/centipede/workdir.h b/centipede/workdir.h
@@ -42,6 +42,11 @@ class WorkDir {
     std::string MyShardPath() const;
     // Returns a glob matching all the shard files.
     std::string AllShardsGlob() const;
+    // Returns true if `path` looks like a shard file path from this set.
+    // Matching is purely lexicographical: the actual file doesn't have to exist
+    // on disk, but `path` must have the exact `base_dir`/`rel_prefix` prefix,
+    // including any relative "." and ".." path elements.
+    bool IsShardPath(std::string_view path) const;
 
    private:
     friend class WorkDir;