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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
/*!The Make-like Build Utility based on Lua
*
* XMake is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* XMake is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with XMake;
* If not, see <a href="http://www.gnu.org/licenses/"> http://www.gnu.org/licenses/</a>
*
* Copyright (C) 2015 - 2016, ruki All rights reserved.
*
* @author ruki
* @file waitlist.c
*
*/
/* //////////////////////////////////////////////////////////////////////////////////////
* trace
*/
#define TB_TRACE_MODULE_NAME "process.waitlist"
#define TB_TRACE_MODULE_DEBUG (0)
/* //////////////////////////////////////////////////////////////////////////////////////
* includes
*/
#include "prefix.h"
/* //////////////////////////////////////////////////////////////////////////////////////
* implementation
*/
/* count, list = process.waitlist(proclist, timeout)
*
* count:
*
* the finished count: > 0
* timeout: 0
* failed: -1
*
* for _, procinfo in ipairs(list) do
* print("proc: ", procinfo[1])
* print("index: ", procinfo[2])
* print("status: ", procinfo[3])
* end
*/
tb_int_t xm_process_waitlist(lua_State* lua)
{
// check
tb_assert_and_check_return_val(lua, 0);
// is table?
if (!lua_istable(lua, 1))
{
// error
lua_pushfstring(lua, "invalid argument type(%s) for process.waitlist", luaL_typename(lua, 1));
lua_error(lua);
return 0;
}
// get the processes count
tb_long_t count = lua_objlen(lua, 1);
if (count <= 0 || count > 64)
{
// error
lua_pushfstring(lua, "invalid process count(%ld) for process.waitlist", count);
lua_error(lua);
return 0;
}
// get the processes
tb_size_t i = 0;
tb_process_ref_t processes[64 + 1];
for (i = 0; i < count; i++)
{
// get proclist[i]
lua_pushinteger(lua, i + 1);
lua_gettable(lua, 1);
// is userdata?
if (lua_isuserdata(lua, -1))
{
// save this process
processes[i] = (tb_process_ref_t)lua_touserdata(lua, -1);
if (!processes[i])
{
// error
lua_pushfstring(lua, "process[%ld] is null for process.waitlist", i);
lua_error(lua);
}
}
else
{
// error
lua_pushfstring(lua, "invalid process[%ld] type(%s) for process.waitlist", i, luaL_typename(lua, -1));
lua_error(lua);
}
// pop it
lua_pop(lua, 1);
}
// end
processes[count] = tb_null;
// get the timeout
tb_long_t timeout = (tb_long_t)luaL_checkinteger(lua, 2);
// wait it
tb_process_waitinfo_t infolist[64];
tb_long_t infosize = tb_process_waitlist(processes, infolist, tb_arrayn(infolist), timeout);
// save process info count
lua_pushinteger(lua, infosize);
if (infosize >= 0)
{
// save process info list
lua_newtable(lua);
for (i = 0; i < infosize; i++)
{
// save one process info
lua_newtable(lua);
lua_pushlightuserdata(lua, infolist[i].process);
lua_rawseti(lua, -2, 1);
lua_pushinteger(lua, infolist[i].index + 1);
lua_rawseti(lua, -2, 2);
lua_pushinteger(lua, infolist[i].status);
lua_rawseti(lua, -2, 3);
lua_rawseti(lua, -2, i + 1);
}
}
else lua_pushnil(lua);
// ok
return 2;
}
|
