github/ocf
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pyocf: rio: Introduce copy()

Browse Source 
This allows to create a Rio prototype object and the copy it for every
invocations so that some parameters can be overwritten without modifying
the prototype object itself.

Signed-off-by: Robert Baldyga <robert.baldyga@huawei.com>
Signed-off-by: Michal Mielewczyk <michal.mielewczyk@huawei.com>
This commit is contained in:
Robert Baldyga 2024-02-02 17:55:25 +01:00 committed by Michal Mielewczyk
parent d892e3b0d5
commit 8d4661bdc9
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/functional/pyocf/rio.py
View File

@ -11,7 +11,7 @@ from dataclasses import dataclass, field
from datetime import timedelta, datetime from datetime import timedelta, datetime
from itertools import cycle from itertools import cycle
from threading import Thread, Condition, Event from threading import Thread, Condition, Event
from copy import deepcopy import copy
from pyocf.utils import Size from pyocf.utils import Size
from pyocf.types.volume import Volume from pyocf.types.volume import Volume
@ -173,6 +173,11 @@ class Rio:
self.errors = {} self.errors = {}
self.error_count = 0 self.error_count = 0
def copy(self):
r = copy.copy(self)
r.global_jobspec = copy.copy(self.global_jobspec)
return r
def readwrite(self, rw: ReadWrite): def readwrite(self, rw: ReadWrite):
self.global_jobspec.readwrite = rw self.global_jobspec.readwrite = rw
return self return self
@ -269,7 +274,7 @@ class Rio:
def run_async(self, queues): def run_async(self, queues):
self.clear() self.clear()
jobs = deepcopy(self.jobs) jobs = copy.deepcopy(self.jobs)
if not jobs: if not jobs:
jobs = [self.global_jobspec for _ in range(self.global_jobspec.njobs)] jobs = [self.global_jobspec for _ in range(self.global_jobspec.njobs)]