Tested controller commands.

Co-authored-by: Elshimaa Betah <79271594+ShimaaBetah@users.noreply.github.com> Co-authored-by: Ziad Othman <ziadsadek999@gmail.com>
Ahmad45123 · May 18, 2024 · 2372d42 · 2372d42
1 parent d86d13f
commit 2372d42
Show file tree

Hide file tree

Showing 19 changed files with 436 additions and 164 deletions.
diff --git a/controller/pom.xml b/controller/pom.xml
@@ -33,16 +33,41 @@
             <version>${project.version}</version>
             <exclusions>
                 <exclusion>
-                    <groupId>org.testcontainers</groupId>
-                    <artifactId>cassandra</artifactId>
+                    <groupId>*</groupId>
+                    <artifactId>*</artifactId>
                 </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>com.workup</groupId>
+            <artifactId>contracts</artifactId>
+            <version>${project.version}</version>
+            <exclusions>
                 <exclusion>
-                    <groupId>org.springframework.boot</groupId>
-                    <artifactId>spring-boot-starter-data-cassandra</artifactId>
+                    <groupId>*</groupId>
+                    <artifactId>*</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>com.workup</groupId>
+            <artifactId>payments</artifactId>
+            <version>${project.version}</version>
+            <exclusions>
+                <exclusion>
+                    <groupId>*</groupId>
+                    <artifactId>*</artifactId>
                 </exclusion>
+            </exclusions>
+        </dependency>
+        <dependency>
+            <groupId>com.workup</groupId>
+            <artifactId>users</artifactId>
+            <version>${project.version}</version>
+            <exclusions>
                 <exclusion>
-                    <groupId>org.springframework.boot</groupId>
-                    <artifactId>spring-boot-starter-web</artifactId>
+                    <groupId>*</groupId>
+                    <artifactId>*</artifactId>
                 </exclusion>
             </exclusions>
         </dependency>

diff --git a/controller/src/main/java/com/workup/controller/CLIHandler.java b/controller/src/main/java/com/workup/controller/CLIHandler.java
@@ -3,9 +3,7 @@
 import asg.cliche.CLIException;
 import asg.cliche.Command;
 import com.workup.shared.commands.controller.*;
-import com.workup.shared.commands.jobs.requests.CreateJobRequest;
 import com.workup.shared.enums.ControllerQueueNames;
-import com.workup.shared.enums.ServiceQueueNames;
 import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;
@@ -78,12 +76,7 @@ public String freeze(String app) {
     return "Command sent";
   }
 
-  @Command(description = "stops a specific app")
-  public String setmq(String app, int appNum) {
-    return "setmq";
-  }
-
-  @Command(description = "stops a specific app")
+  @Command(description = "sets a logging level")
   public String setLoggingLevel(String app, String level) {
     app = app.toLowerCase();
     if (!appQueueMap.containsKey(app)) {
@@ -96,17 +89,6 @@ public String setLoggingLevel(String app, String level) {
     return "Command sent!!";
   }
 
-  @Command(description = "test")
-  public void test() {
-    CreateJobRequest request = CreateJobRequest.builder().withTitle("Ziko").build();
-    rabbitTemplate.convertSendAndReceive(ServiceQueueNames.JOBS, request);
-  }
-
-  @Command(description = "Creates a new command")
-  public String addcommand(String app, String commandName, String className) {
-    return "Add command";
-  }
-
   @Command(description = "Updates an existing command")
   public String updateCommand(String app, String commandName, String className) throws Exception {
     app = app.toLowerCase();
@@ -142,12 +124,13 @@ private byte[] getByteCode(String commandName, String className)
   }
 
   @Command(description = "Deletes an existing command")
-  public String deletecommand(String app, String commandName, String className) {
-    return "Delete command";
-  }
-
-  @Command(description = "stops a specific app")
-  public String updateClass(String app, int appNum) {
-    return "update class";
+  public String deleteCommand(String app, String commandName) {
+    app = app.toLowerCase();
+    if (!appQueueMap.containsKey(app)) {
+      return "Error: app can only be jobs, users, contracts or payments!";
+    }
+    rabbitTemplate.convertAndSend(
+        appQueueMap.get(app), "", DeleteCommandRequest.builder().commandName(commandName).build());
+    return "Command sent";
   }
 }
diff --git a/services/contracts/src/main/java/com/workup/contracts/ContractsApplication.java b/services/contracts/src/main/java/com/workup/contracts/ContractsApplication.java
@@ -2,6 +2,7 @@
 
 import com.workup.shared.enums.ControllerQueueNames;
 import com.workup.shared.enums.ServiceQueueNames;
+import com.workup.shared.enums.ThreadPoolSize;
 import org.springframework.amqp.core.AmqpTemplate;
 import org.springframework.amqp.core.AnonymousQueue;
 import org.springframework.amqp.core.Binding;
@@ -79,8 +80,9 @@ public Binding fanoutBinding(FanoutExchange fanout, Queue controllerQueue) {
   @Bean
   public ThreadPoolTaskExecutor taskExecutor() {
     ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
-    executor.setCorePoolSize(50);
-    executor.setMaxPoolSize(50);
+    executor.setCorePoolSize(ThreadPoolSize.POOL_SIZE);
+    executor.setMaxPoolSize(ThreadPoolSize.POOL_SIZE);
+    executor.setWaitForTasksToCompleteOnShutdown(true);
     executor.setQueueCapacity(500);
     executor.setThreadNamePrefix("contracts-");
     executor.initialize();

diff --git a/services/contracts/src/main/java/com/workup/contracts/ControllerMQListener.java b/services/contracts/src/main/java/com/workup/contracts/ControllerMQListener.java
@@ -1,35 +1,138 @@
 package com.workup.contracts;
 
+import com.workup.contracts.commands.ContractCommand;
+import com.workup.contracts.commands.ContractCommandMap;
+import com.workup.shared.commands.Command;
+import com.workup.shared.commands.CommandRequest;
+import com.workup.shared.commands.CommandResponse;
+import com.workup.shared.commands.controller.ContinueRequest;
+import com.workup.shared.commands.controller.DeleteCommandRequest;
+import com.workup.shared.commands.controller.FreezeRequest;
+import com.workup.shared.commands.controller.SetLoggingLevelRequest;
 import com.workup.shared.commands.controller.SetMaxThreadsRequest;
-import java.lang.reflect.Field;
+import com.workup.shared.commands.controller.UpdateCommandRequest;
+import com.workup.shared.enums.ServiceQueueNames;
+import com.workup.shared.enums.ThreadPoolSize;
+import org.apache.logging.log4j.Level;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import org.apache.logging.log4j.core.config.Configurator;
 import org.springframework.amqp.rabbit.annotation.RabbitHandler;
 import org.springframework.amqp.rabbit.annotation.RabbitListener;
+import org.springframework.amqp.rabbit.listener.RabbitListenerEndpointRegistry;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.ApplicationContext;
 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 import org.springframework.stereotype.Service;
 
 @Service
-@RabbitListener(queues = "#{controllerQueue.name}")
+@RabbitListener(queues = "#{controllerQueue.name}", id = "#{controllerQueue.name}")
 public class ControllerMQListener {
-
+  @Autowired public ContractCommandMap commandMap;
   @Autowired public ThreadPoolTaskExecutor taskExecutor;
-
   @Autowired private ApplicationContext context;
+  @Autowired private RabbitListenerEndpointRegistry registry;
 
   @RabbitHandler
   public void receive(SetMaxThreadsRequest in) throws Exception {
     try {
-      ThreadPoolTaskExecutor myBean = context.getBean(ThreadPoolTaskExecutor.class);
-      Field maxPoolSize = ThreadPoolTaskExecutor.class.getDeclaredField("maxPoolSize");
-      maxPoolSize.setAccessible(true);
-      maxPoolSize.set(myBean, in.getMaxThreads());
-      Field corePoolSize = ThreadPoolTaskExecutor.class.getDeclaredField("corePoolSize");
-      corePoolSize.setAccessible(true);
-      corePoolSize.set(myBean, in.getMaxThreads());
+      System.out.println("Max threads is: " + taskExecutor.getMaxPoolSize());
+      setThreads(in.getMaxThreads());
+      ThreadPoolSize.POOL_SIZE = taskExecutor.getMaxPoolSize();
+      System.out.println("Max threads set to: " + taskExecutor.getMaxPoolSize());
+    } catch (Exception e) {
+      System.out.println(e.getMessage());
+      e.printStackTrace();
+    }
+  }
+
+  @RabbitHandler
+  public void receive(SetLoggingLevelRequest in) throws Exception {
+    try {
+      Logger logger = LogManager.getLogger("com.workup.contracts");
+      Configurator.setAllLevels(logger.getName(), Level.valueOf(in.getLevel()));
+      System.out.println("Logging level set to: " + in.getLevel());
+    } catch (Exception e) {
+      System.out.println(e.getMessage());
+      e.printStackTrace();
+    }
+  }
+
+  @RabbitHandler
+  public void receive(FreezeRequest in) throws Exception {
+    try {
+      registry.getListenerContainer(ServiceQueueNames.CONTRACTS).stop();
+      setThreads(1);
+      System.out.println("Stopped all threads.");
+    } catch (Exception e) {
+      System.out.println(e.getMessage());
+      e.printStackTrace();
+    }
+  }
+
+  @RabbitHandler
+  public void receive(ContinueRequest in) throws Exception {
+    try {
+      registry.getListenerContainer(ServiceQueueNames.CONTRACTS).start();
+      setThreads(ThreadPoolSize.POOL_SIZE);
+      System.out.println("Continued all threads.");
     } catch (Exception e) {
       System.out.println(e.getMessage());
       e.printStackTrace();
     }
   }
+
+  @SuppressWarnings("unchecked")
+  @RabbitHandler
+  public void receive(UpdateCommandRequest in) throws Exception {
+    try {
+      String className = commandMap.getCommand(in.getCommandName()).getClass().getName();
+      byte[] byteArray = in.getByteCode();
+      Class<?> clazz =
+          (Class<?>)
+              (new MyClassLoader(this.getClass().getClassLoader()).loadClass(byteArray, className));
+
+      commandMap.replaceCommand(
+          in.getCommandName(),
+          (Class<? extends ContractCommand<? extends CommandRequest, ? extends CommandResponse>>)
+              ((Command<?, ?>) clazz.newInstance()).getClass());
+
+      System.out.println("Updated command: " + in.getCommandName());
+      // clazz.newInstance().Run(null);
+    } catch (Exception e) {
+      System.out.println(e.getMessage());
+      e.printStackTrace();
+    }
+  }
+
+  @RabbitHandler
+  public void receive(DeleteCommandRequest in) throws Exception {
+    try {
+      commandMap.removeCommand(in.getCommandName());
+      System.out.println("Deleted command: " + in.getCommandName());
+    } catch (Exception e) {
+      System.out.println(e.getMessage());
+      e.printStackTrace();
+    }
+  }
+
+  static class MyClassLoader extends ClassLoader {
+    public MyClassLoader(ClassLoader classLoader) {
+      super(classLoader);
+    }
+
+    public Class<?> loadClass(byte[] byteCode, String className) {
+      return defineClass(className, byteCode, 0, byteCode.length);
+    }
+  }
+
+  private void setThreads(int threads) throws NoSuchFieldException, IllegalAccessException {
+    if (threads > taskExecutor.getCorePoolSize()) {
+      taskExecutor.setMaxPoolSize(threads);
+      taskExecutor.setCorePoolSize(threads);
+    } else {
+      taskExecutor.setCorePoolSize(threads);
+      taskExecutor.setMaxPoolSize(threads);
+    }
+  }
 }
diff --git a/services/contracts/src/main/java/com/workup/contracts/RabbitMQListener.java b/services/contracts/src/main/java/com/workup/contracts/RabbitMQListener.java
@@ -30,7 +30,7 @@
 import org.springframework.stereotype.Service;
 
 @Service
-@RabbitListener(queues = ServiceQueueNames.CONTRACTS)
+@RabbitListener(queues = ServiceQueueNames.CONTRACTS, id = ServiceQueueNames.CONTRACTS)
 public class RabbitMQListener {
 
   @Autowired public ContractCommandMap commandMap;

diff --git a/services/contracts/src/main/java/com/workup/contracts/commands/ContractCommand.java b/services/contracts/src/main/java/com/workup/contracts/commands/ContractCommand.java
@@ -13,13 +13,13 @@
 public abstract class ContractCommand<T extends CommandRequest, Q extends CommandResponse>
     implements Command<T, Q> {
 
-  @Setter AmqpTemplate rabbitTemplate;
+  @Setter public AmqpTemplate rabbitTemplate;
 
-  @Setter ContractRepository contractRepository;
+  @Setter public ContractRepository contractRepository;
 
-  @Setter ContractMilestoneRepository contractMilestoneRepository;
+  @Setter public ContractMilestoneRepository contractMilestoneRepository;
 
-  @Setter TerminationRequestRepository terminationRequestRepository;
+  @Setter public TerminationRequestRepository terminationRequestRepository;
 
-  @Setter RedisService redisService;
+  @Setter public RedisService redisService;
 }
diff --git a/services/contracts/src/main/java/com/workup/contracts/commands/ContractCommandMap.java b/services/contracts/src/main/java/com/workup/contracts/commands/ContractCommandMap.java
@@ -15,15 +15,15 @@
 public class ContractCommandMap
     extends CommandMap<ContractCommand<? extends CommandRequest, ? extends CommandResponse>> {
 
-  @Autowired ContractRepository contractRepository;
+  @Autowired public ContractRepository contractRepository;
 
-  @Autowired ContractMilestoneRepository contractMilestoneRepository;
+  @Autowired public ContractMilestoneRepository contractMilestoneRepository;
 
-  @Autowired TerminationRequestRepository terminationRequestRepository;
+  @Autowired public TerminationRequestRepository terminationRequestRepository;
 
-  @Autowired AmqpTemplate rabbitTemplate;
+  @Autowired public AmqpTemplate rabbitTemplate;
 
-  @Autowired RedisService redisService;
+  @Autowired public RedisService redisService;
 
   public void registerCommands() {
     commands.put("InitiateContract", InitiateContractCommand.class);