Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Return the host failures in the same process sleep requests.
[simgrid.git] / src / simix / smx_context_ruby.c
1 /* context_Ruby - implementation of context switching with/for ruby         */
2
3 /* Copyright (c) 2010. The SimGrid Team.
4  * All rights reserved.                                                     */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9 #include "xbt/function_types.h"
10 #include "xbt/sysdep.h"
11 #include "xbt/log.h"
12 #include "xbt/asserts.h"
13
14 #include "bindings/ruby_bindings.h"
15
16 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(ruby);
17
18 static smx_context_t
19 smx_ctx_ruby_create_context(xbt_main_func_t code, int argc, char **argv,
20     void_pfn_smxprocess_t cleanup_func,
21     void *data);
22
23 static void smx_ctx_ruby_stop(smx_context_t context);
24 static void smx_ctx_ruby_suspend(smx_context_t context);
25 static void smx_ctx_ruby_resume(smx_context_t new_context);
26 static void smx_ctx_ruby_runall(xbt_dynar_t processes);
27
28 void SIMIX_ctx_ruby_factory_init(smx_context_factory_t * factory)
29 {
30   smx_ctx_base_factory_init(factory);
31
32   (*factory)->create_context = smx_ctx_ruby_create_context;
33   /* Do not overload that method (*factory)->finalize */
34   /* Do not overload that method (*factory)->free */
35   (*factory)->stop = smx_ctx_ruby_stop;
36   (*factory)->suspend = smx_ctx_ruby_suspend;
37   (*factory)->name = "smx_ruby_context_factory";
38   (*factory)->runall = smx_ctx_ruby_runall;
39   ruby_init();
40   ruby_init_loadpath();
41 }
42
43 static smx_context_t
44 smx_ctx_ruby_create_context(xbt_main_func_t code, int argc, char **argv,
45     void_pfn_smxprocess_t cleanup_func, void *data)
46 {
47
48   smx_ctx_ruby_t context = (smx_ctx_ruby_t)
49       smx_ctx_base_factory_create_context_sized(sizeof(s_smx_ctx_ruby_t),
50                                                 code, argc, argv,
51                                                 cleanup_func, data);
52
53   /* if the user provided a function for the process , then use it
54      Otherwise it's the context for maestro */
55   if (code) {
56     context->process = (VALUE) code;
57
58     DEBUG1("smx_ctx_ruby_create_context(%s)...Done", argv[0]);
59   }
60   
61   return (smx_context_t) context;
62 }
63
64 static void smx_ctx_ruby_stop(smx_context_t context)
65 {
66   DEBUG0("smx_ctx_ruby_stop()");
67   VALUE process = Qnil;
68   smx_ctx_ruby_t ctx_ruby, current;
69
70   smx_ctx_base_stop(context);
71
72   ctx_ruby = (smx_ctx_ruby_t) context;
73
74   if (smx_current_context->iwannadie) {
75     if (ctx_ruby->process) {
76
77       //if the Ruby Process still Alive ,let's Schedule it
78       if (rb_process_isAlive(ctx_ruby->process)) {
79
80         current = (smx_ctx_ruby_t) smx_current_context;
81         rb_process_schedule(current->process);
82         process = ctx_ruby->process;
83         // interupt/kill The Ruby Process
84         rb_process_kill_up(process);
85       }
86     }
87   } else {
88
89     if (ctx_ruby->process)
90       ctx_ruby->process = Qnil;
91
92   }
93 }
94
95 static void smx_ctx_ruby_suspend(smx_context_t context)
96 {
97
98   DEBUG1("smx_ctx_ruby_suspend(%s)", context->argv[0]);
99   smx_ctx_ruby_t ctx_ruby = (smx_ctx_ruby_t) context;
100   if (ctx_ruby->process)
101     rb_process_unschedule(ctx_ruby->process);
102 }
103
104 static void smx_ctx_ruby_resume(smx_context_t new_context)
105 {
106   DEBUG1("smx_ctx_ruby_resume(%s)",
107          (new_context->argc ? new_context->argv[0] : "maestro"));
108
109   smx_ctx_ruby_t ctx_ruby = (smx_ctx_ruby_t) new_context;
110   rb_process_schedule(ctx_ruby->process);
111 }
112
113 static void smx_ctx_ruby_runall(xbt_dynar_t processes)
114 {
115   smx_process_t process;
116   smx_context_t old_context;
117   unsigned int cursor;
118
119   xbt_dynar_foreach(processes, cursor, process) {
120     old_context = smx_current_context;
121     smx_current_context = process->context;
122     smx_ctx_ruby_resume(smx_current_context);
123     smx_current_context = old_context;
124   }
125   xbt_dynar_reset(processes);
126 }