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
144
145
|
/*!The Make-like Build Utility based on Lua
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
*
* Copyright (C) 2015 - 2018, TBOOX Open Source Group.
*
* @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(%d) for process.waitlist", count);
lua_error(lua);
return 0;
}
// get the processes
tb_int_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[%d] is null for process.waitlist", i);
lua_error(lua);
}
}
else
{
// error
lua_pushfstring(lua, "invalid process[%d] 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;
}
|
