Initial project structure

manager/src/wrapperManager.ts

@@ -0,0 +1,615 @@
+import { Rabbit } from "./rabbit";
+import { randomUUID } from "crypto";
+import { WrapperRegistry, WrapperInfo } from "./wrapperRegistry";
+import { ServerRegistry } from "./serverRegistry";
+
+
+export class WrapperManager {
+
+
+    constructor(
+        private rabbit: Rabbit,
+        private registry: WrapperRegistry,
+        private serverRegistry: ServerRegistry
+    ){
+
+    }
+
+
+
+    async start(){
+
+
+        const queue =
+            "wrapper.register";
+
+
+
+        await this.rabbit.channel.assertQueue(
+            queue
+        );
+
+
+
+        console.log(
+            "Warte auf Wrapper..."
+        );
+
+
+
+        this.rabbit.channel.consume(
+
+            queue,
+
+            async msg => {
+
+
+                if(!msg)
+                    return;
+
+
+
+                const data =
+                    JSON.parse(
+                        msg.content.toString()
+                    );
+
+
+
+                switch(data.type){
+
+
+
+                    case "wrapper.register":
+
+
+                        await this.register(
+                            data
+                        );
+
+
+                        break;
+
+
+
+                    case "wrapper.heartbeat":
+
+
+                        this.heartbeat(
+                            data
+                        );
+
+
+                        break;
+
+
+
+                    case "wrapper.status":
+
+
+                        this.registry.updateStatus(
+                            data
+                        );
+
+                        this.serverRegistry.syncWrapperStatus(
+                            data
+                        );
+
+
+                        break;
+
+
+
+                    default:
+
+
+                        console.log(
+                            "Unbekannte Nachricht:",
+                            data
+                        );
+
+
+                }
+
+
+
+                this.rabbit.channel.ack(
+                    msg
+                );
+
+
+            }
+
+        );
+
+
+
+
+
+        // Offline Check
+
+        setInterval(()=>{
+
+
+            this.registry.checkOffline(
+                30000
+            );
+
+
+        },5000);
+
+
+
+    }
+
+
+
+
+
+
+
+
+    async register(data:any){
+
+
+
+        let id =
+            data.wrapperId;
+
+
+
+
+        /*
+            Neue Wrapper bekommen eine ID
+        */
+
+        if(!id){
+
+
+            id =
+                "wrp-" +
+                randomUUID()
+                .slice(0,8);
+
+
+
+            console.log(
+                "Neue Wrapper ID:",
+                id
+            );
+
+
+        }
+
+
+
+
+
+        const existing =
+            this.registry.get(
+                id
+            );
+
+
+
+
+
+        /*
+            Wrapper war schon bekannt
+        */
+
+        if(existing){
+
+
+            existing.online = true;
+
+
+            existing.lastHeartbeat =
+                Date.now();
+
+
+
+            existing.hostname =
+                data.hostname;
+
+
+
+            existing.cpu =
+                data.cpu;
+
+
+
+            existing.ram =
+                data.ram;
+
+
+
+            existing.ramFree =
+                data.ramFree ?? data.ram;
+
+
+
+            existing.servers =
+                data.servers ?? existing.servers;
+
+
+
+            this.registry.update(existing);
+
+
+        }
+
+
+
+        /*
+            Neuer Wrapper
+        */
+
+        else {
+
+
+
+            const commandQueue =
+                `wrapper.${id}.commands`;
+
+
+
+
+            await this.rabbit.channel.assertQueue(
+                commandQueue
+            );
+
+
+
+
+
+            const wrapper:WrapperInfo = {
+
+
+                id,
+
+
+                hostname:
+                    data.hostname,
+
+
+
+                commandQueue,
+
+
+
+                cpu:
+                    data.cpu,
+
+
+
+                ram:
+                    data.ram,
+
+
+
+                ramFree:
+                    data.ram,
+
+
+
+                servers:
+                    [],
+
+
+
+                online:true,
+
+
+
+                lastHeartbeat:
+                    Date.now()
+
+
+
+            };
+
+
+
+
+
+            this.registry.add(
+                wrapper
+            );
+
+
+
+        }
+
+
+
+
+
+
+
+
+        /*
+            Antwort zurück an Wrapper
+        */
+
+
+        const wrapper =
+            this.registry.get(
+                id
+            );
+
+
+
+
+        if(data.replyQueue){
+
+
+            this.rabbit.channel.sendToQueue(
+
+
+                data.replyQueue,
+
+
+                Buffer.from(
+
+                    JSON.stringify({
+
+
+                        type:
+                            "wrapper.registered",
+
+
+
+                        wrapperId:
+                            id,
+
+
+
+                        commandQueue:
+                            wrapper?.commandQueue
+
+
+                    })
+
+                )
+
+            );
+
+
+        }
+
+
+
+
+    }
+
+
+
+
+
+
+
+
+
+    heartbeat(data:any){
+
+
+
+        this.registry.updateHeartbeat(
+
+            data.wrapperId
+
+        );
+
+
+
+        console.log(
+
+            "Heartbeat:",
+
+            data.wrapperId
+
+        );
+
+
+
+    }
+
+
+
+
+
+
+
+
+
+    sendCommand(
+
+        wrapperId:string,
+
+        command:any
+
+    ){
+
+
+
+        const wrapper =
+            this.registry.get(
+                wrapperId
+            );
+
+
+
+
+        if(!wrapper){
+
+
+            throw new Error(
+
+                "Wrapper nicht gefunden"
+
+            );
+
+
+        }
+
+
+
+
+
+        this.rabbit.channel.sendToQueue(
+
+
+            wrapper.commandQueue,
+
+
+
+            Buffer.from(
+
+                JSON.stringify(
+                    command
+                )
+
+            )
+
+        );
+
+
+
+    }
+
+
+
+
+
+
+
+
+
+    getWrappers(){
+
+
+        return this.registry.getAll();
+
+
+    }
+
+
+
+
+
+
+
+
+
+    listWrappers(){
+
+
+
+        const wrappers =
+            this.registry.getAll();
+
+
+
+
+        console.log(
+
+            "\n======================"
+
+        );
+
+
+        console.log(
+
+            " WRAPPER ÜBERSICHT "
+
+        );
+
+
+        console.log(
+
+            "======================\n"
+
+        );
+
+
+
+
+
+        if(wrappers.length===0){
+
+
+            console.log(
+
+                "Keine Wrapper registriert"
+
+            );
+
+
+            return;
+
+
+        }
+
+
+
+
+
+
+        for(const wrapper of wrappers){
+
+
+
+            console.log(`
+
+ID:
+${wrapper.id}
+
+
+Host:
+${wrapper.hostname}
+
+
+CPU:
+${wrapper.cpu} Cores
+
+
+RAM:
+${wrapper.ram} MB
+
+
+RAM frei:
+${wrapper.ramFree} MB
+
+
+Server:
+${wrapper.servers.length}
+
+
+Online:
+${wrapper.online}
+
+
+Heartbeat:
+${new Date(
+    wrapper.lastHeartbeat
+).toLocaleString()}
+
+
+----------------------
+
+`);
+
+
+
+
+        }
+
+
+
+    }
+
+
+
+
+}