diff --git a/bot-core/bot-core-simple/src/main/java/dev/struchkov/godfather/simple/core/GeneralAutoResponder.java b/bot-core/bot-core-simple/src/main/java/dev/struchkov/godfather/simple/core/GeneralAutoResponder.java index fa0174d..3fdf20e 100644 --- a/bot-core/bot-core-simple/src/main/java/dev/struchkov/godfather/simple/core/GeneralAutoResponder.java +++ b/bot-core/bot-core-simple/src/main/java/dev/struchkov/godfather/simple/core/GeneralAutoResponder.java @@ -14,6 +14,7 @@ import dev.struchkov.haiti.context.exception.NotFoundException; import org.slf4j.Logger; import org.slf4j.LoggerFactory; +import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; @@ -51,7 +52,8 @@ public class GeneralAutoResponder { } public void registrationActionUnit(ActionUnit actionUnit) { - actionUnitMap.computeIfAbsent(actionUnit.getUnitType(), k -> new HashMap<>()).putIfAbsent(actionUnit.getMessageType(), actionUnit); + actionUnitMap.computeIfAbsent(actionUnit.getUnitType(), k -> new ArrayList<>()); + actionUnitMap.get(actionUnit.getUnitType()).add(actionUnit); } /** @@ -132,7 +134,7 @@ public class GeneralAutoResponder { return accessibility.isEmpty() || accessibility.get().check(message); } - public void answer(UnitRequest, M> unitRequest) { + public void answer(UnitRequest unitRequest) { if (checkNotNull(errorHandler)) { try { unitRequest = getAction(unitRequest); @@ -146,7 +148,7 @@ public class GeneralAutoResponder { } } - private UnitRequest, M> activeUnitAfter(UnitRequest, M> unitRequest) { + private UnitRequest activeUnitAfter(UnitRequest unitRequest) { final Set> nextUnits = unitRequest.getUnit().getNextUnits(); if (checkNotNull(nextUnits)) { Optional> first = nextUnits.stream()