Добавил в интерфейсы service и repository редактирование UnitPointer
This commit is contained in:
parent
19f7e4bb71
commit
f8f8a51d79
@ -14,43 +14,46 @@ import java.util.Set;
|
|||||||
import java.util.regex.Matcher;
|
import java.util.regex.Matcher;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
/*
|
||||||
|
Отвечает за функционирование автоответчика
|
||||||
|
*/
|
||||||
public class Autoresponder {
|
public class Autoresponder {
|
||||||
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(Autoresponder.class);
|
private static final Logger log = LoggerFactory.getLogger(Autoresponder.class);
|
||||||
|
private static final UnitPriorityComparator UNIT_PRIORITY_COMPARATOR = new UnitPriorityComparator();
|
||||||
|
|
||||||
private Set<Unit> menuUnits;
|
/*
|
||||||
private UnitPointerService unitPointerService;
|
Начальные unit, по которым происходит поиск, если пользователь только обратился или закончил сценарий.
|
||||||
|
*/
|
||||||
|
private final Set<Unit> menuUnits;
|
||||||
|
private final UnitPointerService unitPointerService;
|
||||||
|
|
||||||
public Autoresponder(UnitPointerService unitPointerService) {
|
public Autoresponder(UnitPointerService unitPointerService, Set<Unit> menuUnits) {
|
||||||
this.unitPointerService = unitPointerService;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setMenuUnits(Set<Unit> menuUnits) {
|
|
||||||
this.menuUnits = menuUnits;
|
this.menuUnits = menuUnits;
|
||||||
|
this.unitPointerService = unitPointerService;
|
||||||
}
|
}
|
||||||
|
|
||||||
public UnitPointerService getUnitPointerService() {
|
public UnitPointerService getUnitPointerService() {
|
||||||
return unitPointerService;
|
return unitPointerService;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setUnitPointerService(UnitPointerService unitPointerService) {
|
/*
|
||||||
this.unitPointerService = unitPointerService;
|
Возвращает unit на основании сообщения пользователя
|
||||||
}
|
*/
|
||||||
|
public Unit answer(Integer personId, String message) {
|
||||||
public Unit answer(Integer idPerson, String message) {
|
UnitPointer unitPointer = checkAndAddPerson(personId);
|
||||||
UnitPointer unitPointer = checkAndAddPerson(idPerson);
|
|
||||||
Unit unit;
|
Unit unit;
|
||||||
if (unitPointer.getUnit() == null) {
|
if (unitPointer.getUnit() == null) {
|
||||||
unit = nextUnit(menuUnits, message);
|
unit = nextUnit(menuUnits, message); // выбирает unit из menuUnits, если пользователь обращается впервые
|
||||||
} else {
|
} else {
|
||||||
if (unitPointer.getUnit().getNextUnits() == null) {
|
if (unitPointer.getUnit().getNextUnits() == null) {
|
||||||
unit = nextUnit(menuUnits, message);
|
unit = nextUnit(menuUnits, message); // если пользователь закончил сценарий, то выбирает следующий юнит из menuUnits
|
||||||
} else {
|
} else {
|
||||||
unit = nextUnit(unitPointer.getUnit().getNextUnits(), message);
|
unit = nextUnit(unitPointer.getUnit().getNextUnits(), message);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (unit != null) {
|
if (unit != null) {
|
||||||
unitPointer.setUnit(unit);
|
unitPointerService.edit(personId, unit);
|
||||||
}
|
}
|
||||||
return unit;
|
return unit;
|
||||||
}
|
}
|
||||||
@ -68,19 +71,21 @@ public class Autoresponder {
|
|||||||
|
|
||||||
private Unit nextUnit(Set<Unit> nextUnits, String message) {
|
private Unit nextUnit(Set<Unit> nextUnits, String message) {
|
||||||
if (nextUnits.size() > 0) {
|
if (nextUnits.size() > 0) {
|
||||||
UnitPriorityComparator unitPriorityComparator = new UnitPriorityComparator();
|
Optional<Unit> patternUnits = nextUnits.stream()
|
||||||
Optional<Unit> patternUnits = nextUnits.stream().filter(nextUnit -> nextUnit.getPattern() != null).filter(nextUnit -> patternReg(nextUnit, message)).max(unitPriorityComparator);
|
.filter(nextUnit -> nextUnit.getPattern() != null)
|
||||||
|
.filter(nextUnit -> patternReg(nextUnit, message))
|
||||||
|
.max(UNIT_PRIORITY_COMPARATOR);
|
||||||
|
|
||||||
if (!patternUnits.isPresent()) {
|
if (!patternUnits.isPresent()) {
|
||||||
patternUnits = nextUnits.stream()
|
patternUnits = nextUnits.stream()
|
||||||
.filter(nextUnit -> textPercentageMatch(nextUnit, Parser.parse(message)) >= nextUnit.getMatchThreshold())
|
.filter(nextUnit -> textPercentageMatch(nextUnit, Parser.parse(message)) >= nextUnit.getMatchThreshold())
|
||||||
.max(unitPriorityComparator);
|
.max(UNIT_PRIORITY_COMPARATOR);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!patternUnits.isPresent()) {
|
if (!patternUnits.isPresent()) {
|
||||||
patternUnits = nextUnits.stream()
|
patternUnits = nextUnits.stream()
|
||||||
.filter(nextUnit -> (nextUnit.getPattern() == null && nextUnit.getKeyWords() == null))
|
.filter(nextUnit -> (nextUnit.getPattern() == null && nextUnit.getKeyWords() == null))
|
||||||
.max(unitPriorityComparator);
|
.max(UNIT_PRIORITY_COMPARATOR);
|
||||||
}
|
}
|
||||||
|
|
||||||
return patternUnits.orElse(null);
|
return patternUnits.orElse(null);
|
||||||
@ -99,7 +104,6 @@ public class Autoresponder {
|
|||||||
if (unit.getKeyWords() != null) {
|
if (unit.getKeyWords() != null) {
|
||||||
Set<String> temp = new HashSet<>(unit.getKeyWords());
|
Set<String> temp = new HashSet<>(unit.getKeyWords());
|
||||||
temp.retainAll(words);
|
temp.retainAll(words);
|
||||||
log.info("Юнит: " + unit.getClass().getSimpleName());
|
|
||||||
log.info("Ключевые слова юнита: " + unit.getKeyWords() + " (" + unit.getKeyWords().size() + ")");
|
log.info("Ключевые слова юнита: " + unit.getKeyWords() + " (" + unit.getKeyWords().size() + ")");
|
||||||
log.info("Ключевые слова от пользователя: " + words);
|
log.info("Ключевые слова от пользователя: " + words);
|
||||||
log.info("Пересечение: " + temp + " (" + temp.size() + ")");
|
log.info("Пересечение: " + temp + " (" + temp.size() + ")");
|
||||||
|
@ -1,10 +1,14 @@
|
|||||||
package org.sadtech.autoresponder.entity;
|
package org.sadtech.autoresponder.entity;
|
||||||
|
|
||||||
|
import java.util.Arrays;
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
import java.util.Objects;
|
import java.util.Objects;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
/*
|
||||||
|
Абстрактный класс, предпологающий какие-то действия
|
||||||
|
*/
|
||||||
public abstract class Unit {
|
public abstract class Unit {
|
||||||
|
|
||||||
private Set<String> keyWords;
|
private Set<String> keyWords;
|
||||||
@ -13,34 +17,40 @@ public abstract class Unit {
|
|||||||
private Integer priority = 10;
|
private Integer priority = 10;
|
||||||
private Set<Unit> nextUnits;
|
private Set<Unit> nextUnits;
|
||||||
|
|
||||||
public Unit() {
|
public void setKeyWord(String... keyWords) {
|
||||||
|
if (this.keyWords == null) {
|
||||||
}
|
this.keyWords = new HashSet<>();
|
||||||
|
|
||||||
public Unit(Unit... nextUnit) {
|
|
||||||
nextUnits = new HashSet(nextUnits);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setKeyWord(String keyWord) {
|
|
||||||
if (keyWords == null) {
|
|
||||||
keyWords = new HashSet<>();
|
|
||||||
}
|
}
|
||||||
keyWords.add(keyWord);
|
this.keyWords.addAll(Arrays.asList(keyWords));
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setNextUnit(Unit unit) {
|
public void setKeyWords(Set<String> keyWords) {
|
||||||
if (nextUnits == null) {
|
if (this.keyWords == null) {
|
||||||
nextUnits = new HashSet<>();
|
this.keyWords = new HashSet<>();
|
||||||
}
|
}
|
||||||
nextUnits.add(unit);
|
this.keyWords.addAll(keyWords);
|
||||||
}
|
}
|
||||||
|
|
||||||
public Set<String> getKeyWords() {
|
public Set<String> getKeyWords() {
|
||||||
return keyWords;
|
return keyWords;
|
||||||
}
|
}
|
||||||
|
|
||||||
public void setKeyWords(Set<String> keyWords) {
|
public void setNextUnit(Unit... units) {
|
||||||
this.keyWords = keyWords;
|
if (nextUnits == null) {
|
||||||
|
nextUnits = new HashSet<>();
|
||||||
|
}
|
||||||
|
nextUnits.addAll(Arrays.asList(units));
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setNextUnits(Set<Unit> nextUnits) {
|
||||||
|
if (nextUnits == null) {
|
||||||
|
nextUnits = new HashSet<>();
|
||||||
|
}
|
||||||
|
this.nextUnits.addAll(nextUnits);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Set<Unit> getNextUnits() {
|
||||||
|
return nextUnits;
|
||||||
}
|
}
|
||||||
|
|
||||||
public Integer getMatchThreshold() {
|
public Integer getMatchThreshold() {
|
||||||
@ -59,14 +69,6 @@ public abstract class Unit {
|
|||||||
this.priority = priority;
|
this.priority = priority;
|
||||||
}
|
}
|
||||||
|
|
||||||
public Set<Unit> getNextUnits() {
|
|
||||||
return nextUnits;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void setNextUnits(Set<Unit> nextUnits) {
|
|
||||||
this.nextUnits = nextUnits;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Pattern getPattern() {
|
public Pattern getPattern() {
|
||||||
return pattern;
|
return pattern;
|
||||||
}
|
}
|
||||||
|
@ -2,6 +2,9 @@ package org.sadtech.autoresponder.entity;
|
|||||||
|
|
||||||
import java.util.Objects;
|
import java.util.Objects;
|
||||||
|
|
||||||
|
/*
|
||||||
|
Сохраняет юнит, на котором остановился пользователь.
|
||||||
|
*/
|
||||||
public class UnitPointer {
|
public class UnitPointer {
|
||||||
|
|
||||||
private Integer entityId;
|
private Integer entityId;
|
||||||
|
@ -8,6 +8,8 @@ public interface UnitPointerRepository {
|
|||||||
|
|
||||||
Integer add(UnitPointer unitPointer);
|
Integer add(UnitPointer unitPointer);
|
||||||
|
|
||||||
|
void edit(UnitPointer unitPointer);
|
||||||
|
|
||||||
void remove(Integer entityId);
|
void remove(Integer entityId);
|
||||||
|
|
||||||
void addAll(Map<Integer, UnitPointer> unitPointerMap);
|
void addAll(Map<Integer, UnitPointer> unitPointerMap);
|
||||||
|
@ -15,6 +15,11 @@ public class UnitPointerRepositoryMap implements UnitPointerRepository {
|
|||||||
return unitPointer.getEntityId();
|
return unitPointer.getEntityId();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void edit(UnitPointer unitPointer) {
|
||||||
|
unitPointerMap.get(unitPointer.getEntityId()).setUnit(unitPointer.getUnit());
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void remove(Integer entityId) {
|
public void remove(Integer entityId) {
|
||||||
unitPointerMap.remove(entityId);
|
unitPointerMap.remove(entityId);
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
package org.sadtech.autoresponder.service;
|
package org.sadtech.autoresponder.service;
|
||||||
|
|
||||||
|
import org.sadtech.autoresponder.entity.Unit;
|
||||||
import org.sadtech.autoresponder.entity.UnitPointer;
|
import org.sadtech.autoresponder.entity.UnitPointer;
|
||||||
|
|
||||||
public interface UnitPointerService {
|
public interface UnitPointerService {
|
||||||
@ -10,4 +11,6 @@ public interface UnitPointerService {
|
|||||||
|
|
||||||
UnitPointer getByEntityId(Integer entityId);
|
UnitPointer getByEntityId(Integer entityId);
|
||||||
|
|
||||||
|
void edit(Integer personId, Unit unit);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
package org.sadtech.autoresponder.service;
|
package org.sadtech.autoresponder.service;
|
||||||
|
|
||||||
|
import org.sadtech.autoresponder.entity.Unit;
|
||||||
import org.sadtech.autoresponder.entity.UnitPointer;
|
import org.sadtech.autoresponder.entity.UnitPointer;
|
||||||
import org.sadtech.autoresponder.repository.UnitPointerRepository;
|
import org.sadtech.autoresponder.repository.UnitPointerRepository;
|
||||||
import org.sadtech.autoresponder.repository.UnitPointerRepositoryMap;
|
import org.sadtech.autoresponder.repository.UnitPointerRepositoryMap;
|
||||||
@ -25,6 +26,13 @@ public class UnitPointerServiceImpl implements UnitPointerService {
|
|||||||
return unitPointerRepository.findByEntityId(entityId);
|
return unitPointerRepository.findByEntityId(entityId);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void edit(Integer personId, Unit unit) {
|
||||||
|
if (check(personId)) {
|
||||||
|
unitPointerRepository.edit(new UnitPointer(personId, unit));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void add(UnitPointer unitPointer) {
|
public void add(UnitPointer unitPointer) {
|
||||||
unitPointerRepository.add(unitPointer);
|
unitPointerRepository.add(unitPointer);
|
||||||
|
@ -5,6 +5,9 @@ import java.util.Set;
|
|||||||
import java.util.regex.Matcher;
|
import java.util.regex.Matcher;
|
||||||
import java.util.regex.Pattern;
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
/*
|
||||||
|
Возвращает Set слов из текста
|
||||||
|
*/
|
||||||
public class Parser {
|
public class Parser {
|
||||||
|
|
||||||
private Parser() {
|
private Parser() {
|
||||||
|
Loading…
Reference in New Issue
Block a user