X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/d59645f3e566e691ba757f74bba503c13773dbe8..138fef8ee97f3cbd335434ad6acd26771fd0b762:/sdk/cwl/arvados_cwl/fsaccess.py diff --git a/sdk/cwl/arvados_cwl/fsaccess.py b/sdk/cwl/arvados_cwl/fsaccess.py index 935fe65e1f..0b577b06a2 100644 --- a/sdk/cwl/arvados_cwl/fsaccess.py +++ b/sdk/cwl/arvados_cwl/fsaccess.py @@ -27,14 +27,17 @@ from schema_salad.ref_resolver import DefaultFetcher logger = logging.getLogger('arvados.cwl-runner') class CollectionCache(object): - def __init__(self, api_client, keep_client, num_retries): + def __init__(self, api_client, keep_client, num_retries, + cap=256*1024*1024, + min_entries=2): self.api_client = api_client self.keep_client = keep_client + self.num_retries = num_retries self.collections = OrderedDict() self.lock = threading.Lock() self.total = 0 - self.cap = 256*1024*1024 - self.min_entries = 4 + self.cap = cap + self.min_entries = min_entries def cap_cache(self): if self.total > self.cap: @@ -52,7 +55,8 @@ class CollectionCache(object): if pdh not in self.collections: logger.debug("Creating collection reader for %s", pdh) cr = arvados.collection.CollectionReader(pdh, api_client=self.api_client, - keep_client=self.keep_client) + keep_client=self.keep_client, + num_retries=self.num_retries) sz = len(cr.manifest_text()) * 128 self.collections[pdh] = (cr, sz) self.total += sz @@ -108,7 +112,7 @@ class CollectionFsAccess(cwltool.stdfsaccess.StdFsAccess): if collection is not None and not rest: return [pattern] patternsegments = rest.split("/") - return self._match(collection, patternsegments, "keep:" + collection.manifest_locator()) + return sorted(self._match(collection, patternsegments, "keep:" + collection.manifest_locator())) def open(self, fn, mode): collection, rest = self.get_collection(fn) @@ -177,16 +181,13 @@ class CollectionFsAccess(cwltool.stdfsaccess.StdFsAccess): return os.path.realpath(path) class CollectionFetcher(DefaultFetcher): - def __init__(self, cache, session, api_client=None, fs_access=None, num_retries=4, overrides=None): + def __init__(self, cache, session, api_client=None, fs_access=None, num_retries=4): super(CollectionFetcher, self).__init__(cache, session) self.api_client = api_client self.fsaccess = fs_access self.num_retries = num_retries - self.overrides = overrides if overrides else {} def fetch_text(self, url): - if url in self.overrides: - return self.overrides[url] if url.startswith("keep:"): with self.fsaccess.open(url, "r") as f: return f.read() @@ -197,8 +198,6 @@ class CollectionFetcher(DefaultFetcher): return super(CollectionFetcher, self).fetch_text(url) def check_exists(self, url): - if url in self.overrides: - return True try: if url.startswith("http://arvados.org/cwl"): return True