Miroirs/pinnacle
1
0
Fork 0
mirror of https://github.com/pinnacle-comp/pinnacle.git synced 2024-12-26 21:58:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Port Lua client fix

Browse source
This commit is contained in:
Ottatop 2024-02-23 18:10:28 -06:00
parent 556294d4ef
commit 67ee5021bd
1 changed files with 36 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
api/lua/pinnacle/grpc/client.lua
View file

@ -136,19 +136,27 @@ function client.server_streaming_request(grpc_request_params, callback)
client.loop:wrap(function() client.loop:wrap(function()
for response_body in stream:each_chunk() do for response_body in stream:each_chunk() do
-- Skip the 1-byte compressed flag and the 4-byte message length
---@diagnostic disable-next-line: redefined-local ---@diagnostic disable-next-line: redefined-local
local response_body = response_body:sub(6) local response_body = response_body
---@diagnostic disable-next-line: redefined-local while response_body:len() > 0 do
local success, obj = pcall(pb.decode, response_type, response_body) local msg_len = string.unpack(">I4", response_body:sub(2, 5))
if not success then
print(obj) -- Skip the 1-byte compressed flag and the 4-byte message length
os.exit(1) response_body = response_body:sub(6, 6 + msg_len - 1)
---@diagnostic disable-next-line: redefined-local
local success, obj = pcall(pb.decode, response_type, response_body)
if not success then
print(obj)
os.exit(1)
end
local response = obj
callback(response)
response_body = response_body:sub(msg_len + 1)
end end
local response = obj
callback(response)
end end
local trailers = stream:get_headers() local trailers = stream:get_headers()
@ -184,19 +192,27 @@ function client.bidirectional_streaming_request(grpc_request_params, callback)
client.loop:wrap(function() client.loop:wrap(function()
for response_body in stream:each_chunk() do for response_body in stream:each_chunk() do
-- Skip the 1-byte compressed flag and the 4-byte message length
---@diagnostic disable-next-line: redefined-local ---@diagnostic disable-next-line: redefined-local
local response_body = response_body:sub(6) local response_body = response_body
---@diagnostic disable-next-line: redefined-local while response_body:len() > 0 do
local success, obj = pcall(pb.decode, response_type, response_body) local msg_len = string.unpack(">I4", response_body:sub(2, 5))
if not success then
print(obj) -- Skip the 1-byte compressed flag and the 4-byte message length
os.exit(1) response_body = response_body:sub(6, 6 + msg_len - 1)
---@diagnostic disable-next-line: redefined-local
local success, obj = pcall(pb.decode, response_type, response_body)
if not success then
print(obj)
os.exit(1)
end
local response = obj
callback(response)
response_body = response_body:sub(msg_len + 1)
end end
local response = obj
callback(response)
end end
local trailers = stream:get_headers() local trailers = stream:get_headers()