Elijah/CollabVM-Agent-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

message queue shits

Browse source
This commit is contained in:
Elijah 2023-12-10 15:19:13 -05:00
parent 5557dc1750
commit 5d791150b0
1 changed files with 18 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
src/VM.ts
View file

@ -11,7 +11,7 @@ export default class VM {
#socketpath : string;
#socket : Socket;
#writeLock : Mutex = new Mutex();
#fileQueue : Queue<File> = new Queue<File>();
#messageQueue : Queue<protocol.ProtocolMessage> = new Queue<protocol.ProtocolMessage>();
#nopTimeout : NodeJS.Timeout | null = null;
isConnectedToVM : boolean = false;
#noNop : boolean = false;
@ -32,35 +32,30 @@ export default class VM {
this.#socket.on('data', (data) => this.#onData(data));
this.#events = new EventEmitter();
this.#events.on('ack', () => {
if (this.#fileQueue.size > 0)
this.fileQueueLoop();
if (this.#messageQueue.size > 0)
this.messageQueueLoop();
})
}
UploadFile(file : File) {
this.#fileQueue.enqueue(file);
if (this.#fileQueue.size === 1)
this.fileQueueLoop();
}
fileQueueLoop() {
return new Promise(async (res, rej) => {
var file = this.#fileQueue.dequeue();
await this.PushFile(file!);
})
}
PushFile(file : File) : Promise<void> {
return new Promise(async (res, rej) => {
const msg : protocol.ProtocolMessage = {
this.#enqueueMessage({
Operation: protocol.ProtocolOperation.UploadFile,
Filename: file.Filename,
FileData: file.FileData
};
});
}
messageQueueLoop() {
return new Promise(async (res, rej) => {
var msg = this.#messageQueue.dequeue();
var payload = msgpack.encode(msg);
await this.#sendMessage(payload);
res();
});
})
}
#enqueueMessage(msg : protocol.ProtocolMessage) {
this.#messageQueue.enqueue(msg);
if (this.#messageQueue.size === 1) this.messageQueueLoop();
}
#sendMessage(data : Buffer) : Promise<void> {
@ -102,7 +97,7 @@ export default class VM {
if (this.#nopTimeout) clearInterval(this.#nopTimeout);
this.#nopTimeout = setInterval(() => this.#nopTimeoutFunc(), 5000);
if (!this.isConnectedToVM) {
if (this.#fileQueue.size > 0) this.fileQueueLoop();
if (this.#messageQueue.size > 0) this.messageQueueLoop();
this.isConnectedToVM = true;
this.#noNop = false;
}
@ -115,11 +110,9 @@ export default class VM {
#nopTimeoutFunc() {
if (!this.#noNop) {
var payload : protocol.ProtocolMessage = {
this.#enqueueMessage({
Operation: protocol.ProtocolOperation.NOP
};
var data = msgpack.encode(payload);
this.#sendMessage(data);
});
this.#noNop = true;
} else {
this.isConnectedToVM = false;