fix remote build

1 files changed, 16 insertions, 0 deletions

diff --git a/xmake/modules/private/service/message.lua b/xmake/modules/private/service/message.lua
index 7338b06ac..12792a129 100644
--- a/xmake/modules/private/service/message.lua
+++ b/xmake/modules/private/service/message.lua
@@ -37,6 +37,7 @@ message.CODE_PULL           = 9  -- pull the given file from server
 message.CODE_PUSH           = 10 -- push the given file to server
 message.CODE_FILEINFO       = 11 -- get the given file info in server
 message.CODE_EXISTINFO      = 12 -- get exists info in server (use bloom filter)
+message.CODE_END            = 13 -- end
 
 -- init message
 function message:init(body)
@@ -113,6 +114,11 @@ function message:is_existinfo()
     return self:code() == message.CODE_EXISTINFO
 end
 
+-- is end message?
+function message:is_end()
+    return self:code() == message.CODE_END
+end
+
 -- get user authorization
 function message:token()
     return self:body().token
@@ -300,6 +306,16 @@ function new_existinfo(session_id, name, opt)
     })
 end
 
+-- new end message
+function new_end(session_id, opt)
+    opt = opt or {}
+    return _new({
+        code = message.CODE_END,
+        session_id = session_id,
+        token = opt.token
+    })
+end
+
 function main(body)
     return _new(body)
 end