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
|
--!A cross-platform build utility based on Lua
--
-- Licensed 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-present, TBOOX Open Source Group.
--
-- @author ruki
-- @file server.lua
--
-- imports
import("private.service.config")
import("private.service.message")
import("private.service.server")
import("private.service.stream", {alias = "socket_stream"})
import("private.service.distributed_build.session", {alias = "server_session"})
import("lib.detect.find_tool")
-- define module
local distributed_build_server = distributed_build_server or server()
local super = distributed_build_server:class()
-- init server
function distributed_build_server:init(daemon)
super.init(self, daemon)
if self:daemon() then
config.load()
end
-- init address
local address = assert(config.get("distributed_build.server.listen"), "config(distributed_build.server.listen): not found!")
super.address_set(self, address)
-- init handler
super.handler_set(self, self._on_handle)
-- init sessions
self._SESSIONS = {}
end
-- get class
function distributed_build_server:class()
return distributed_build_server
end
-- on handle message
function distributed_build_server:_on_handle(stream, msg)
local session_id = msg:session_id()
local session = self:_session(session_id)
vprint("%s: %s: <session %s>: on handle message(%d)", self, stream:sock(), session_id, msg:code())
vprint(msg:body())
session:stream_set(stream)
local respmsg = msg:clone()
local session_errs
local session_ok = try
{
function()
if self:need_verfiy() then
local ok, errors = self:verify_user(msg:token(), stream:sock():peeraddr())
if not ok then
session_errs = errors
return false
end
end
if msg:is_connect() then
session:open()
elseif msg:is_disconnect() then
session:close()
self._SESSIONS[session_id] = nil
else
assert(session:is_connected(), "session has not been connected!")
-- TODO
end
return true
end,
catch
{
function (errors)
if errors then
session_errs = tostring(errors)
vprint(session_errs)
end
end
}
}
respmsg:status_set(session_ok)
if not session_ok and session_errs then
respmsg:errors_set(session_errs)
end
local ok = stream:send_msg(respmsg) and stream:flush()
vprint("%s: %s: <session %s>: send %s", self, stream:sock(), session_id, ok and "ok" or "failed")
end
-- get session
function distributed_build_server:_session(session_id)
local session = self._SESSIONS[session_id]
if not session then
session = server_session(session_id)
self._SESSIONS[session_id] = session
end
return session
end
-- close session
function distributed_build_server:_session_close(session_id)
self._SESSIONS[session_id] = nil
end
function distributed_build_server:__tostring()
return "<distributed_build_server>"
end
function main(daemon)
local instance = distributed_build_server()
instance:init(daemon ~= nil)
return instance
end
|
