rawfile-localpv/remote.py

58 lines
1.4 KiB
Python
Raw Normal View History

2020-04-24 11:48:09 +00:00
from util import remote_fn
@remote_fn
def scrub(volume_id):
2020-04-24 15:38:18 +00:00
import time
import rawfile_util
rawfile_util.patch_metadata(volume_id, {"deleted_at": time.time()})
2020-04-24 11:48:09 +00:00
@remote_fn
def init_rawfile(volume_id, size, fs_type):
2020-04-24 15:38:18 +00:00
import time
import rawfile_util
from volume_schema import LATEST_SCHEMA_VERSION
2020-06-13 23:47:32 +00:00
from pathlib import Path
2020-04-24 11:48:09 +00:00
from util import run
2020-04-24 15:38:18 +00:00
img_dir = rawfile_util.img_dir(volume_id)
2020-06-13 23:47:32 +00:00
img_dir.mkdir(exist_ok=True)
img_file = Path(f"{img_dir}/disk.img")
if img_file.exists():
return
2020-04-24 15:38:18 +00:00
rawfile_util.patch_metadata(
volume_id,
{
"schema_version": LATEST_SCHEMA_VERSION,
2020-04-24 15:38:18 +00:00
"volume_id": volume_id,
"created_at": time.time(),
2020-06-13 23:47:32 +00:00
"img_file": img_file.as_posix(),
2020-04-24 15:38:18 +00:00
"size": size,
"fs_type": fs_type,
2020-04-24 15:38:18 +00:00
},
)
2020-04-24 11:48:09 +00:00
run(f"truncate -s {size} {img_file}")
if fs_type == "ext4":
run(f"mkfs.ext4 {img_file}")
elif fs_type == "btrfs":
run(f"mkfs.btrfs {img_file}")
else:
raise Exception(f"Unsupported fsType: {fs_type}")
@remote_fn
def expand_rawfile(volume_id, size):
import rawfile_util
from util import run
img_file = rawfile_util.img_file(volume_id)
2020-06-13 23:47:32 +00:00
if rawfile_util.metadata(volume_id)["size"] >= size:
return
rawfile_util.patch_metadata(
volume_id, {"size": size},
)
run(f"truncate -s {size} {img_file}")