26 lines
673 B
Python
26 lines
673 B
Python
|
import os
|
||
|
|
import logging
|
||
|
|
from ..object import FileBlobStorage
|
||
|
|
from .chunk import ChunkID
|
||
|
|
|
||
|
|
|
||
|
|
class ChunkStore:
|
||
|
|
def __init__(self, root_dir: str):
|
||
|
|
logging.info(f"will init chunk store, root_dir={root_dir}")
|
||
|
|
|
||
|
|
if not os.path.exists(root_dir):
|
||
|
|
os.makedirs(root_dir)
|
||
|
|
|
||
|
|
self.root = root_dir
|
||
|
|
self.blob = FileBlobStorage(root_dir)
|
||
|
|
|
||
|
|
def put_chunk(self, chunk_id: ChunkID, contents: bytes):
|
||
|
|
self.blob.put(chunk_id, contents)
|
||
|
|
|
||
|
|
def get_chunk(self, chunk_id: ChunkID) -> bytes:
|
||
|
|
return self.blob.get(chunk_id)
|
||
|
|
|
||
|
|
def delete_chunk(self, chunk_id: ChunkID):
|
||
|
|
self.blob.delete(chunk_id)
|
||
|
|
|
||
|
|
|