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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
|
local json = require 'json'
local parser = require 'parser'
local ErrorCodes = {
-- Defined by JSON RPC
ParseError = -32700,
InvalidRequest = -32600,
MethodNotFound = -32601,
InvalidParams = -32602,
InternalError = -32603,
serverErrorStart = -32099,
serverErrorEnd = -32000,
ServerNotInitialized = -32002,
UnknownErrorCode = -32001,
-- Defined by the protocol.
RequestCancelled = -32800,
}
local mt = {}
mt.__index = mt
mt._input = nil
mt._output = nil
mt._method = nil
mt._file = nil
function mt:_callback(method, params)
local f = self._method
if f then
return f(method, params)
end
return nil, '没有注册method'
end
function mt:_send(data)
local f = self._output
if not f then
return
end
data.jsonrpc = '2.0'
local content = json.encode(data)
local buf = ('Content-Length: %d\r\n\r\n%s'):format(#content, content)
f(buf)
end
function mt:_readProtoHead()
local header = self:read 'l'
if not header then
return
end
if header:sub(1, #'Content-Length') == 'Content-Length' then
return header
elseif header:sub(1, #'Content-Type') == 'Content-Type' then
else
log.error('错误的协议头：', header)
end
end
function mt:_readProtoContent(header)
local len = tonumber(header:match('%d+'))
if not len then
log.error('错误的协议头：', header)
return
end
local buf = self:read(len+2)
if not buf then
return
end
local suc, res = pcall(json.decode, buf)
if not suc then
log.error('错误的协议：', buf)
return
end
local id = res.id
local method = res.method
local params = res.params
local response, err = self:_callback(method, params)
if id then
if response then
self:_send {
id = id,
result = response,
}
else
self:_send {
id = id,
error = {
code = ErrorCodes.UnknownErrorCode,
message = err or ('没有回应：' .. method),
},
}
end
end
if not response then
log.error(err or ('没有回应：' .. method))
end
return true
end
function mt:_buildTextCache()
if not next(self._need_compile) then
return
end
local list = {}
for uri in pairs(self._need_compile) do
list[#list+1] = uri
end
local size = 0
local clock = os.clock()
for _, uri in ipairs(list) do
local obj = self:compileText(uri)
if obj then
size = size + #obj.text
end
end
local passed = os.clock() - clock
log.debug(('\n\z
语法树缓存完成\n\z
耗时：[%.3f]秒\n\z
数量：[%d]\n\z
总大小：[%.3f]kb\n\z
速度：[%.3f]kb/s\n\z
内存：[%.3f]kb'):format(
passed,
#list,
size / 1000,
size / passed / 1000,
collectgarbage 'count'
))
end
function mt:setInput(input)
self._input = input
end
function mt:setOutput(output)
self._output = output
end
function mt:read(mode)
if not self._input then
return nil
end
return self._input(mode)
end
function mt:saveText(uri, version, text)
local obj = self._file[uri]
if obj then
if obj.version >= version then
return
end
obj.version = version
obj.text = text
self._need_compile[uri] = true
else
self._file[uri] = {
version = version,
text = text,
}
self._need_compile[uri] = true
end
end
function mt:loadText(uri)
local obj = self._file[uri]
if not obj then
return nil
end
self:compileText(uri)
return obj.ast, obj.lines
end
function mt:compileText(uri)
local obj = self._file[uri]
if not obj then
return nil
end
if not self._need_compile[uri] then
return nil
end
self._need_compile[uri] = nil
local ast, err = parser:ast(obj.text)
if not ast then
return nil
end
obj.ast = ast
obj.lines = parser:lines(obj.text)
return obj
end
function mt:removeText(uri)
self._file[uri] = nil
self._need_compile[uri] = nil
end
function mt:setMethod(method)
self._method = method
end
function mt:runStep()
if not self._header then
-- 如果没有协议头，则尝试读一条协议头
self._header = self:_readProtoHead()
end
if self._header then
-- 如果有协议头，则读取协议内容
local suc = self:_readProtoContent(self._header)
if suc then
-- 协议内容读取成功后重置
self._header = nil
self._idle_clock = 0
end
return
end
self._idle_clock = self._idle_clock + 1
if self._idle_clock == 1000 then
self:_buildTextCache()
end
end
function mt:stop()
self._input = nil
self._output = nil
end
return function ()
return setmetatable({
_file = {},
_need_compile = {},
_header = nil,
_idle_clock = 0,
}, mt)
end
|
