144 lines
4.7 KiB
Python
144 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright (C) 2013 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 logging
|
|
import os
|
|
import sys
|
|
|
|
logging.basicConfig(level=logging.ERROR)
|
|
|
|
self_dir = os.path.abspath(os.path.dirname(__file__))
|
|
top_dir = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
|
os.pardir,
|
|
os.pardir))
|
|
sys.path.insert(0, top_dir)
|
|
sys.path.insert(0, self_dir)
|
|
|
|
import taskflow.engines
|
|
|
|
from taskflow.patterns import linear_flow as lf
|
|
from taskflow import task
|
|
from taskflow.utils import persistence_utils as p_utils
|
|
|
|
import example_utils # noqa
|
|
|
|
# INTRO: In this example linear_flow is used to group three tasks, one which
|
|
# will suspend the future work the engine may do. This suspend engine is then
|
|
# discarded and the workflow is reloaded from the persisted data and then the
|
|
# workflow is resumed from where it was suspended. This allows you to see how
|
|
# to start an engine, have a task stop the engine from doing future work (if
|
|
# a multi-threaded engine is being used, then the currently active work is not
|
|
# preempted) and then resume the work later.
|
|
#
|
|
# Usage:
|
|
#
|
|
# With a filesytem directory as backend
|
|
#
|
|
# python taskflow/examples/resume_from_backend.py
|
|
#
|
|
# With ZooKeeper as backend
|
|
#
|
|
# python taskflow/examples/resume_from_backend.py \
|
|
# zookeeper://127.0.0.1:2181/taskflow/resume_from_backend/
|
|
#
|
|
|
|
|
|
### UTILITY FUNCTIONS #########################################
|
|
|
|
|
|
def print_wrapped(text):
|
|
print("-" * (len(text)))
|
|
print(text)
|
|
print("-" * (len(text)))
|
|
|
|
|
|
def print_task_states(flowdetail, msg):
|
|
print_wrapped(msg)
|
|
print("Flow '%s' state: %s" % (flowdetail.name, flowdetail.state))
|
|
# Sort by these so that our test validation doesn't get confused by the
|
|
# order in which the items in the flow detail can be in.
|
|
items = sorted((td.name, td.version, td.state, td.results)
|
|
for td in flowdetail)
|
|
for item in items:
|
|
print(" %s==%s: %s, result=%s" % item)
|
|
|
|
|
|
def find_flow_detail(backend, lb_id, fd_id):
|
|
conn = backend.get_connection()
|
|
lb = conn.get_logbook(lb_id)
|
|
return lb.find(fd_id)
|
|
|
|
|
|
### CREATE FLOW ###############################################
|
|
|
|
|
|
class InterruptTask(task.Task):
|
|
def execute(self):
|
|
# DO NOT TRY THIS AT HOME
|
|
engine.suspend()
|
|
|
|
|
|
class TestTask(task.Task):
|
|
def execute(self):
|
|
print('executing %s' % self)
|
|
return 'ok'
|
|
|
|
|
|
def flow_factory():
|
|
return lf.Flow('resume from backend example').add(
|
|
TestTask(name='first'),
|
|
InterruptTask(name='boom'),
|
|
TestTask(name='second'))
|
|
|
|
|
|
### INITIALIZE PERSISTENCE ####################################
|
|
|
|
with example_utils.get_backend() as backend:
|
|
logbook = p_utils.temporary_log_book(backend)
|
|
|
|
### CREATE AND RUN THE FLOW: FIRST ATTEMPT ####################
|
|
|
|
flow = flow_factory()
|
|
flowdetail = p_utils.create_flow_detail(flow, logbook, backend)
|
|
engine = taskflow.engines.load(flow, flow_detail=flowdetail,
|
|
backend=backend)
|
|
|
|
print_task_states(flowdetail, "At the beginning, there is no state")
|
|
print_wrapped("Running")
|
|
engine.run()
|
|
print_task_states(flowdetail, "After running")
|
|
|
|
### RE-CREATE, RESUME, RUN ####################################
|
|
|
|
print_wrapped("Resuming and running again")
|
|
|
|
# NOTE(harlowja): reload the flow detail from backend, this will allow us
|
|
# to resume the flow from its suspended state, but first we need to search
|
|
# for the right flow details in the correct logbook where things are
|
|
# stored.
|
|
#
|
|
# We could avoid re-loading the engine and just do engine.run() again, but
|
|
# this example shows how another process may unsuspend a given flow and
|
|
# start it again for situations where this is useful to-do (say the process
|
|
# running the above flow crashes).
|
|
flow2 = flow_factory()
|
|
flowdetail2 = find_flow_detail(backend, logbook.uuid, flowdetail.uuid)
|
|
engine2 = taskflow.engines.load(flow2,
|
|
flow_detail=flowdetail2,
|
|
backend=backend)
|
|
engine2.run()
|
|
print_task_states(flowdetail2, "At the end")
|