码迷,mamicode.com
首页 > 其他好文 > 详细

openstack中运行定时任务的两种方法及源代码分析

时间:2014-11-09 15:13:36      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:style   http   io   ar   使用   sp   for   数据   div   

启动一个进程,如要想要这个进程的某个方法定时得进行执行的话,在openstack有两种方式: 一种是通过继承 periodic_task.PeriodicTasks,另一种是使用loopingcall.py,针对两种方式分别说一下实现原理。
 
(1) 继承periodic_task.PeriodicTasks
         这种方式比较复杂,用到了python中的一些比较高级的特性,装饰器和元类;首先看一下periodic_task.py,在nova/openstack/common中,其他组件也有。
         看一下PeriodicTasks 这个类。
144 @six.add_metaclass(_PeriodicTasksMeta)
145 class PeriodicTasks(object):
146     def __init__(self):
147         super(PeriodicTasks, self).__init__()
148         self._periodic_last_run = {}
149         for name, task in self._periodic_tasks:
150             self._periodic_last_run[name] = task._periodic_last_run
151
152     def run_periodic_tasks(self, context, raise_on_error=False):
153         """Tasks to be run at a periodic interval."""
154         idle_for = DEFAULT_INTERVAL
155         for task_name, task in self._periodic_tasks:
156             full_task_name = ‘.‘.join([self.__class__.__name__, task_name])
157
158             spacing = self._periodic_spacing[task_name]
159             last_run = self._periodic_last_run[task_name]
160
161             # If a periodic task is _nearly_ due, then we‘ll run it early
162             if spacing is not None:
163                 idle_for = min(idle_for, spacing)
164                 if last_run is not None:
165                     delta = last_run + spacing - time.time()
166                     if delta > 0.2:
167                         idle_for = min(idle_for, delta)
168                         continue
169
170             LOG.debug("Running periodic task %(full_task_name)s",
171                       {"full_task_name": full_task_name})
172             self._periodic_last_run[task_name] = time.time()
173
174             try:
175                 task(self, context)
176             except Exception as e:
177                 if raise_on_error:
178                     raise
179                 LOG.exception(_LE("Error during %(full_task_name)s: %(e)s"),
180                               {"full_task_name": full_task_name, "e": e})
181             time.sleep(0)
182
183         return idle_for
 
run_periodic_tasks 函数是用户启动各个定时任务的,其中里面有几个数据结构比较重要,self._periodic_tasks:记录来每个task和每个task的函数句柄;self._periodic_spacing: 记录每一个task的运行间隔时间。在__init__函数中,还有构造一个self._periodic_last_run 结构用来记录每一个task上一次运行的时间;具体运行的时候会根据上次运行时间和间隔时间来确定是否运行,函数第162~168行;那具体的self._periodic_tasks和self._periodic_spacing是怎么得来的,是通过元类的方式来实现的;
 
元类可以干预一个类的实现形式,比方说在为一个类添加一个方法,或者为了一个类统一添加某种行为;上文的@six.add_metaclass(_PeriodicTasksMeta)语法就是添加了一个元类,我们看一下_PeriodicTasksMeta的实现;
 
100 class _PeriodicTasksMeta(type):
101     def __init__(cls, names, bases, dict_):
102         """Metaclass that allows us to collect decorated periodic tasks."""
103         super(_PeriodicTasksMeta, cls).__init__(names, bases, dict_)
104
105         # NOTE(sirp): if the attribute is not present then we must be the base
106         # class, so, go ahead an initialize it. If the attribute is present,
107         # then we‘re a subclass so make a copy of it so we don‘t step on our
108         # parent‘s toes.
109         try:
110             cls._periodic_tasks = cls._periodic_tasks[:]
111         except AttributeError:
112             cls._periodic_tasks = []
113
114         try:
115             cls._periodic_spacing = cls._periodic_spacing.copy()
116         except AttributeError:
117             cls._periodic_spacing = {}
118
119         for value in cls.__dict__.values():
120             if getattr(value, ‘_periodic_task‘, False):
121                 task = value
122                 name = task.__name__
123
124                 if task._periodic_spacing < 0:
125                     LOG.info(_LI(‘Skipping periodic task %(task)s because ‘
126                                  ‘its interval is negative‘),
127                              {‘task‘: name})
128                     continue
129                 if not task._periodic_enabled:
130                     LOG.info(_LI(‘Skipping periodic task %(task)s because ‘
131                                  ‘it is disabled‘),
132                              {‘task‘: name})
133                     continue
134
135                 # A periodic spacing of zero indicates that this task should
136                 # be run every pass
137                 if task._periodic_spacing == 0:
138                     task._periodic_spacing = None
139
140                 cls._periodic_tasks.append((name, task))
141                 cls._periodic_spacing[name] = task._periodic_spacing
其中109~117为类添加_periodic_tasks与_periodic_spacing两个类变量, for value in cls.__dict__.values() 语句访问类的各个成员,主要是函数成员;如果发现成员中有_periodic_task属性,并且等于True,则构造_periodic_tasks与_periodic_spacing两个数据结构;那么剩下就要弄清楚task的结构了,task就是类中的一个函数,它为什么具有_periodic_task属性和_periodic_spacing的呢?这个活就是装饰器做的事情了。
 
当你给一个函数设置定时任务的装饰器时,一般会这样写:
@periodic_task.periodic_task(spacing=…, run_immediately=...)
def f(args,kwargs):
…….
 
奥妙就在这个装饰器里面了。
 42 def periodic_task(*args, **kwargs):
 43     """Decorator to indicate that a method is a periodic task.
 44
 45     This decorator can be used in two ways:
 46
 47         1. Without arguments ‘@periodic_task‘, this will be run on every cycle
 48            of the periodic scheduler.
 49
 50         2. With arguments:
 51            @periodic_task(spacing=N [, run_immediately=[True|False]])
 52            this will be run on approximately every N seconds. If this number is
 53            negative the periodic task will be disabled. If the run_immediately
 54            argument is provided and has a value of ‘True‘, the first run of the
 55            task will be shortly after task scheduler starts.  If
 56            run_immediately is omitted or set to ‘False‘, the first time the
 57            task runs will be approximately N seconds after the task scheduler
 58            starts.
 59     """
 60     def decorator(f):
 61         # Test for old style invocation
 62         if ‘ticks_between_runs‘ in kwargs:
 63             raise InvalidPeriodicTaskArg(arg=‘ticks_between_runs‘)
 64
 65         # Control if run at all
 66         f._periodic_task = True
 67         f._periodic_external_ok = kwargs.pop(‘external_process_ok‘, False)
 68         if f._periodic_external_ok and not CONF.run_external_periodic_tasks:
 69             f._periodic_enabled = False
 70         else:
 71             f._periodic_enabled = kwargs.pop(‘enabled‘, True)
 72
 73         # Control frequency
 74         f._periodic_spacing = kwargs.pop(‘spacing‘, 0)
 75         f._periodic_immediate = kwargs.pop(‘run_immediately‘, False)
 76         if f._periodic_immediate:
 77             f._periodic_last_run = None
 78         else:
 79             f._periodic_last_run = time.time()
 80         return f
 81
 82     # NOTE(sirp): The `if` is necessary to allow the decorator to be used with
 83     # and without parents.
 84     #
 85     # In the ‘with-parents‘ case (with kwargs present), this function needs to
 86     # return a decorator function since the interpreter will invoke it like:
 87     #
 88     #   periodic_task(*args, **kwargs)(f)
 89     #
 90     # In the ‘without-parents‘ case, the original function will be passed
 91     # in as the first argument, like:
 92     #
 93     #   periodic_task(f)
 94     if kwargs:
 95         return decorator
 96     else:
 97         return decorator(args[0])
 
在装饰器中,66~80行就是为一个函数设置_periodic_task属性,并从装饰器的kwargs中取得spacing参数,如果有run_immediately(启动之后先立刻执行一次,然后再定时执行)会设置_periodic_last_run属性。这样定时函数运行需要的信息就都齐全了;
 
(2)另一个种方法是使用loopingcall.py ,也在nova/openstack/common中。
使用方法是:obj = loopingcall.FixedIntervalLoopingCall(f, args,kwargs),
                      obj.start(interval,initial_delay)
  代码:
 62 class FixedIntervalLoopingCall(LoopingCallBase):
 63     """A fixed interval looping call."""
 64
 65     def start(self, interval, initial_delay=None):
 66         self._running = True
 67         done = event.Event()
 68
 69         def _inner():
 70             if initial_delay:
 71                 greenthread.sleep(initial_delay)
 72
 73             try:
 74                 while self._running:
 75                     start = timeutils.utcnow()
 76                     self.f(*self.args, **self.kw)
 77                     end = timeutils.utcnow()
 78                     if not self._running:
 79                         break
 80                     delay = interval - timeutils.delta_seconds(start, end)
 81                     if delay <= 0:
 82                         LOG.warn(_LW(‘task run outlasted interval by %s sec‘) %
 83                                  -delay)
 84                     greenthread.sleep(delay if delay > 0 else 0)
 85             except LoopingCallDone as e:
 86                 self.stop()
 87                 done.send(e.retvalue)
 88             except Exception:
 89                 LOG.exception(_LE(‘in fixed duration looping call‘))
 90                 done.send_exception(*sys.exc_info())
 91                 return
 92             else:
 93                 done.send(True)
 94
 95         self.done = done
 96
 97         greenthread.spawn_n(_inner)
 98         return self.done
 
start()方法运行定时任务,initial_delay表示是否有延时,75~84每次运行任务要记录开始时间和结束时间,如果开始时间减去结束时间比interval还大的话,那么就不等待了,立刻运行:greenthread.sleep(delay if delay > 0 else 0)

 

openstack中运行定时任务的两种方法及源代码分析

标签:style   http   io   ar   使用   sp   for   数据   div   

原文地址:http://www.cnblogs.com/yuhan-TB/p/4085074.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!