acd545f9c2
In order to rework some of the persistence layer I would like to move around some of the files first, keeping job specifics in a jobs folder. Having some of the items which are root level taskflow items (flow, task) be at the root of the hiearchy. Also for now until the celery work is commited move that, since it doesn't make sense in backends anyway. Change-Id: If6c149710b40f70d4ec69ee8e523defe8f5e766d
129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import contextlib
|
|
|
|
from taskflow import flow
|
|
from taskflow.jobs import job
|
|
from taskflow.jobs import jobboard
|
|
from taskflow.persistence import flowdetail
|
|
from taskflow.persistence import logbook
|
|
from taskflow.persistence import taskdetail
|
|
from taskflow import task
|
|
from taskflow import utils
|
|
|
|
|
|
class MemoryJobBoard(jobboard.JobBoard):
|
|
def __init__(self, name, jb_id=None):
|
|
super(MemoryJobBoard, self).__init__(name, jb_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryJob(job.Job):
|
|
def __init__(self, name, job_id=None):
|
|
super(MemoryJob, self).__init__(name, job_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryLogBook(logbook.LogBook):
|
|
def __init__(self, name, lb_id=None):
|
|
super(MemoryLogBook, self).__init__(name, lb_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryFlow(flow.Flow):
|
|
def __init__(self, name, wf_id=None):
|
|
super(MemoryFlow, self).__init__(name, wf_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
self.flowdetails = {}
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryFlowDetail(flowdetail.FlowDetail):
|
|
def __init__(self, name, wf, fd_id=None):
|
|
super(MemoryFlowDetail, self).__init__(name, wf, fd_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryTask(task.Task):
|
|
def __init__(self, name, task_id=None):
|
|
super(MemoryTask, self).__init__(name, task_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
self.taskdetails = {}
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|
|
|
|
|
|
class MemoryTaskDetail(taskdetail.TaskDetail):
|
|
def __init__(self, name, task, td_id=None):
|
|
super(MemoryTaskDetail, self).__init__(name, task, td_id)
|
|
self._lock = utils.ReaderWriterLock()
|
|
|
|
@contextlib.contextmanager
|
|
def acquire_lock(self, read=True):
|
|
try:
|
|
self._lock.acquire(read)
|
|
yield self._lock
|
|
finally:
|
|
self._lock.release()
|