Godfather-Bots/autoresponder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Теперь есть дефолтный ответ, если ни один из юнитов не подошел

Browse Source
This commit is contained in:
Mark Struchkov 2019-05-16 00:34:47 +03:00
parent 9c821ee1fb
commit 666e9223ea
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/main/java/org/sadtech/autoresponder/Autoresponder.java
View File

@ -24,6 +24,7 @@ public class Autoresponder {
private static final UnitPriorityComparator UNIT_PRIORITY_COMPARATOR = new UnitPriorityComparator(); private static final UnitPriorityComparator UNIT_PRIORITY_COMPARATOR = new UnitPriorityComparator();
private final Set<Unit> startUnits; private final Set<Unit> startUnits;
private Unit defaultUnit;
private final UnitPointerService unitPointerService; private final UnitPointerService unitPointerService;
public Autoresponder(UnitPointerService unitPointerService, Set<Unit> startUnits) { public Autoresponder(UnitPointerService unitPointerService, Set<Unit> startUnits) {
@ -35,18 +36,26 @@ public class Autoresponder {
return unitPointerService; return unitPointerService;
} }
public void setDefaultUnit(Unit defaultUnit) {
this.defaultUnit = defaultUnit;
}
/* /*
Возвращает unit на основании сообщения пользователя Возвращает unit на основании сообщения пользователя
*/ */
public Unit answer(Integer personId, String message) { public Unit answer(Integer personId, String message) {
UnitPointer unitPointer = checkAndAddPerson(personId); UnitPointer unitPointer = checkAndAddPerson(personId);
Unit unit; Unit unit;
if (unitPointer.getUnit() == null || unitPointer.getUnit().getNextUnits() == null) { try {
unit = nextUnit(startUnits, message); // выбирает unit из startUnits, если пользователь обращается впервые if (unitPointer.getUnit() == null || unitPointer.getUnit().getNextUnits() == null) {
} else { unit = nextUnit(startUnits, message); // выбирает unit из startUnits, если пользователь обращается впервые
unit = nextUnit(unitPointer.getUnit().getNextUnits(), message); } else {
unit = nextUnit(unitPointer.getUnit().getNextUnits(), message);
}
unitPointerService.edit(personId, unit);
} catch (NotFoundUnitException e) {
unit = defaultUnit;
} }
unitPointerService.edit(personId, unit);
return unit; return unit;
} }