mesos-python-framework/executor.py at master · tarnfeld/mesos-python-framework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#!/usr/bin/env python

# _ _
# _____ ____ _ _ __ ___ _ __ | | ___ _____ _____ ___ _ _| |_ ___ _ __
# / _ \ \/ / _` | '_ ` _ \| '_ \| |/ _ \_____ / _ \ \/ / _ \/ __| | | | __/ _ \| '__|
# | __/> < (_| | | | | | | |_) | | __/_____| __/> < __/ (__| |_| | || (_) | |
# \___/_/\_\__,_|_| |_| |_| .__/|_|\___| \___/_/\_\___|\___|\__,_|\__\___/|_|
# |_|
#

importlogging
importthreading
importtime

importpesos.api
importpesos.executor
frompesos.vendor.mesosimportmesos_pb2

logger=logging.getLogger(__name__)


classExampleExecutor(pesos.api.Executor):

deflaunch_task(self, driver, task):

logger.info("HELLO I AM TASK")

defrun_task():
logger.info("Launching task %s", task.task_id.value)

update=mesos_pb2.TaskStatus()
update.task_id.value=task.task_id.value
update.state=mesos_pb2.TASK_RUNNING
driver.send_status_update(update)

time.sleep(15)

update=mesos_pb2.TaskStatus()
update.task_id.value=task.task_id.value
update.state=mesos_pb2.TASK_FINISHED

# Send the terminal update
driver.send_status_update(update)

thread=threading.Thread(target=run_task)
thread.daemon=True
thread.start()


if__name__=="__main__":

forlin ('pesos', 'compactor', 'tornado', '__main__'):
l=logging.getLogger(l)
l.setLevel(logging.DEBUG)

# Launch the executor driver
driver=pesos.executor.MesosExecutorDriver(ExampleExecutor())

status=0
ifdriver.run() ==mesos_pb2.DRIVER_STOPPED:
status=1

driver.stop()
exit(status)