Martin Blanchard pushed to branch mablanch/93-disk-storage-path-scheme at BuildGrid / buildgrid
Commits:
-
9401c87d
by Finn at 2018-10-02T16:33:03Z
-
51f661fd
by Jürg Billeter at 2018-10-03T07:50:34Z
-
febe2ced
by Martin Blanchard at 2018-10-03T08:08:56Z
-
e80b6503
by Martin Blanchard at 2018-10-03T08:08:56Z
5 changed files:
- buildgrid/client/cas.py
- buildgrid/server/cas/storage/disk.py
- docs/source/using_bazel.rst
- docs/source/using_buildstream.rst
- setup.py
Changes:
| ... | ... | @@ -283,7 +283,7 @@ class Downloader: |
| 283 | 283 |
try:
|
| 284 | 284 |
batch_response = self.__cas_stub.BatchReadBlobs(batch_request)
|
| 285 | 285 |
for response in batch_response.responses:
|
| 286 |
- assert response.digest.hash in digests
|
|
| 286 |
+ assert response.digest in digests
|
|
| 287 | 287 |
|
| 288 | 288 |
read_blobs.append(response.data)
|
| 289 | 289 |
|
| ... | ... | @@ -21,7 +21,6 @@ A CAS storage provider that stores files as blobs on disk. |
| 21 | 21 |
"""
|
| 22 | 22 |
|
| 23 | 23 |
import os
|
| 24 |
-import pathlib
|
|
| 25 | 24 |
import tempfile
|
| 26 | 25 |
|
| 27 | 26 |
from .storage_abc import StorageABC
|
| ... | ... | @@ -30,28 +29,41 @@ from .storage_abc import StorageABC |
| 30 | 29 |
class DiskStorage(StorageABC):
|
| 31 | 30 |
|
| 32 | 31 |
def __init__(self, path):
|
| 33 |
- self._path = pathlib.Path(path)
|
|
| 34 |
- os.makedirs(str(self._path / "temp"), exist_ok=True)
|
|
| 32 |
+ if not os.path.isabs(path):
|
|
| 33 |
+ self.__root_path = os.path.abspath(path)
|
|
| 34 |
+ else:
|
|
| 35 |
+ self.__root_path = path
|
|
| 36 |
+ self.__cas_path = os.path.join(self.__root_path, 'cas')
|
|
| 37 |
+ |
|
| 38 |
+ self.objects_path = os.path.join(self.__cas_path, 'objects')
|
|
| 39 |
+ self.temp_path = os.path.join(self.__root_path, 'tmp')
|
|
| 40 |
+ |
|
| 41 |
+ os.makedirs(self.objects_path, exist_ok=True)
|
|
| 42 |
+ os.makedirs(self.temp_path, exist_ok=True)
|
|
| 35 | 43 |
|
| 36 | 44 |
def has_blob(self, digest):
|
| 37 |
- return (self._path / (digest.hash + "_" + str(digest.size_bytes))).exists()
|
|
| 45 |
+ return os.path.exists(self._get_object_path(digest))
|
|
| 38 | 46 |
|
| 39 | 47 |
def get_blob(self, digest):
|
| 40 | 48 |
try:
|
| 41 |
- return (self._path / (digest.hash + "_" + str(digest.size_bytes))).open('rb')
|
|
| 49 |
+ return open(self._get_object_path(digest), 'rb')
|
|
| 42 | 50 |
except FileNotFoundError:
|
| 43 | 51 |
return None
|
| 44 | 52 |
|
| 45 |
- def begin_write(self, _digest):
|
|
| 46 |
- return tempfile.NamedTemporaryFile("wb", dir=str(self._path / "temp"))
|
|
| 53 |
+ def begin_write(self, digest):
|
|
| 54 |
+ return tempfile.NamedTemporaryFile("wb", dir=self.temp_path)
|
|
| 47 | 55 |
|
| 48 | 56 |
def commit_write(self, digest, write_session):
|
| 49 |
- # Atomically move the temporary file into place.
|
|
| 50 |
- path = self._path / (digest.hash + "_" + str(digest.size_bytes))
|
|
| 51 |
- os.replace(write_session.name, str(path))
|
|
| 57 |
+ object_path = self._get_object_path(digest)
|
|
| 58 |
+ |
|
| 52 | 59 |
try:
|
| 53 |
- write_session.close()
|
|
| 54 |
- except FileNotFoundError:
|
|
| 55 |
- # We moved the temporary file to a new location, so when Python
|
|
| 56 |
- # tries to delete its old location, it'll fail.
|
|
| 60 |
+ os.makedirs(os.path.dirname(object_path), exist_ok=True)
|
|
| 61 |
+ os.link(write_session.name, object_path)
|
|
| 62 |
+ except FileExistsError:
|
|
| 63 |
+ # Object is already there!
|
|
| 57 | 64 |
pass
|
| 65 |
+ |
|
| 66 |
+ write_session.close()
|
|
| 67 |
+ |
|
| 68 |
+ def _get_object_path(self, digest):
|
|
| 69 |
+ return os.path.join(self.objects_path, digest.hash[:2], digest.hash[2:])
|
| ... | ... | @@ -98,7 +98,7 @@ has ``gcc`` installed, run: |
| 98 | 98 |
|
| 99 | 99 |
The ``--remote`` option is used to specify the server location (running on the
|
| 100 | 100 |
same machine here, and listening to port 50051). The ``--parent`` option is used
|
| 101 |
-to specify the server instance you except the bot to be attached to. Refer to
|
|
| 101 |
+to specify the server instance you expect the bot to be attached to. Refer to
|
|
| 102 | 102 |
the :ref:`CLI reference section <invoking-bgd-bot-temp-directory>` for command
|
| 103 | 103 |
line interface details.
|
| 104 | 104 |
|
| ... | ... | @@ -120,7 +120,7 @@ You can finally have Bazel to build the example workspace by running: |
| 120 | 120 |
|
| 121 | 121 |
bazel build //main:hello-world
|
| 122 | 122 |
|
| 123 |
-You can verify that the example has been successfully build by running the
|
|
| 123 |
+You can verify that the example has been successfully built by running the
|
|
| 124 | 124 |
generated executable. Simply invoke:
|
| 125 | 125 |
|
| 126 | 126 |
.. code-block:: sh
|
| ... | ... | @@ -149,7 +149,7 @@ that the ``buildbox`` tool is functional on your machine (refer to |
| 149 | 149 |
|
| 150 | 150 |
The ``--remote`` option is used to specify the server location (running on the
|
| 151 | 151 |
same machine here, and listening to port 50051). The ``--parent`` option is
|
| 152 |
-used to specify the server instance you except the bot to be attached to (empty
|
|
| 152 |
+used to specify the server instance you expect the bot to be attached to (empty
|
|
| 153 | 153 |
here). ``--fuse-dir`` and ``--local-cas`` are specific to the ``buildbox`` bot
|
| 154 | 154 |
and respectively specify the sandbox mount point and local CAS cache locations.
|
| 155 | 155 |
Refer to the :ref:`CLI reference section <invoking-bgd-bot-buildbox>` for
|
| ... | ... | @@ -178,7 +178,7 @@ You can finally have BuildStream to build the example project by running: |
| 178 | 178 |
|
| 179 | 179 |
bst build hello.bst
|
| 180 | 180 |
|
| 181 |
-You can verify that the example has been successfully build by running the
|
|
| 181 |
+You can verify that the example has been successfully built by running the
|
|
| 182 | 182 |
generated executable. Simply invoke:
|
| 183 | 183 |
|
| 184 | 184 |
.. code-block:: sh
|
| ... | ... | @@ -97,7 +97,8 @@ tests_require = [ |
| 97 | 97 |
]
|
| 98 | 98 |
|
| 99 | 99 |
docs_require = [
|
| 100 |
- 'sphinx',
|
|
| 100 |
+ # rtd-theme broken in Sphinx >= 1.8, this breaks search functionality.
|
|
| 101 |
+ 'sphinx == 1.7.8',
|
|
| 101 | 102 |
'sphinx-click',
|
| 102 | 103 |
'sphinx-rtd-theme',
|
| 103 | 104 |
'sphinxcontrib-apidoc',
|
