Skip to content

Commit a8e95d1

Browse files
committed
feat: Add MultiAgentScheduling new algorithm with Junit tests
1 parent bcf4034 commit a8e95d1

File tree

2 files changed

+113
-0
lines changed

2 files changed

+113
-0
lines changed
Lines changed: 61 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
package com.thealgorithms.scheduling;
2+
3+
import java.util.ArrayList;
4+
import java.util.HashMap;
5+
import java.util.List;
6+
import java.util.Map;
7+
8+
/**
9+
* MultiAgentScheduling assigns tasks to different autonomous agents
10+
* who independently decide the execution order of their assigned tasks.
11+
* The focus is on collaboration between agents to optimize the overall schedule.
12+
*
13+
* Use Case: Distributed scheduling in decentralized systems like IoT networks.
14+
*
15+
* @author Hardvan
16+
*/
17+
public class MultiAgentScheduling {
18+
19+
static class Agent {
20+
String name;
21+
List<String> tasks;
22+
23+
Agent(String name) {
24+
this.name = name;
25+
this.tasks = new ArrayList<>();
26+
}
27+
28+
void addTask(String task) {
29+
tasks.add(task);
30+
}
31+
32+
List<String> getTasks() {
33+
return tasks;
34+
}
35+
}
36+
37+
private final Map<String, Agent> agents;
38+
39+
public MultiAgentScheduling() {
40+
agents = new HashMap<>();
41+
}
42+
43+
public void addAgent(String agentName) {
44+
agents.putIfAbsent(agentName, new Agent(agentName));
45+
}
46+
47+
public void assignTask(String agentName, String task) {
48+
Agent agent = agents.get(agentName);
49+
if (agent != null) {
50+
agent.addTask(task);
51+
}
52+
}
53+
54+
public Map<String, List<String>> getScheduledTasks() {
55+
Map<String, List<String>> schedule = new HashMap<>();
56+
for (Agent agent : agents.values()) {
57+
schedule.put(agent.name, agent.getTasks());
58+
}
59+
return schedule;
60+
}
61+
}
Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
package com.thealgorithms.scheduling;
2+
3+
import static org.junit.jupiter.api.Assertions.assertEquals;
4+
5+
import java.util.List;
6+
import java.util.Map;
7+
import org.junit.jupiter.api.BeforeEach;
8+
import org.junit.jupiter.api.Test;
9+
10+
public class MultiAgentSchedulingTest {
11+
12+
private MultiAgentScheduling scheduler;
13+
14+
@BeforeEach
15+
public void setup() {
16+
scheduler = new MultiAgentScheduling();
17+
}
18+
19+
@Test
20+
public void testAddAgentAndAssignTask() {
21+
scheduler.addAgent("Agent1");
22+
scheduler.assignTask("Agent1", "Task1");
23+
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1"));
24+
assertEquals(expected, scheduler.getScheduledTasks());
25+
}
26+
27+
@Test
28+
public void testMultipleAgentsWithTasks() {
29+
scheduler.addAgent("Agent1");
30+
scheduler.addAgent("Agent2");
31+
scheduler.assignTask("Agent1", "Task1");
32+
scheduler.assignTask("Agent2", "Task2");
33+
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1"), "Agent2", List.of("Task2"));
34+
assertEquals(expected, scheduler.getScheduledTasks());
35+
}
36+
37+
@Test
38+
public void testAgentWithMultipleTasks() {
39+
scheduler.addAgent("Agent1");
40+
scheduler.assignTask("Agent1", "Task1");
41+
scheduler.assignTask("Agent1", "Task2");
42+
Map<String, List<String>> expected = Map.of("Agent1", List.of("Task1", "Task2"));
43+
assertEquals(expected, scheduler.getScheduledTasks());
44+
}
45+
46+
@Test
47+
public void testEmptyAgentSchedule() {
48+
scheduler.addAgent("Agent1");
49+
Map<String, List<String>> expected = Map.of("Agent1", List.of());
50+
assertEquals(expected, scheduler.getScheduledTasks());
51+
}
52+
}

0 commit comments

Comments
 (0)