Добавил в интерфейсы service и repository редактирование UnitPointer

This commit is contained in:
Mark Struchkov 2019-05-15 22:27:11 +03:00
parent 19f7e4bb71
commit f8f8a51d79
8 changed files with 77 additions and 47 deletions

View File

@ -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() + ")");

View File

@ -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;
} }

View File

@ -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;

View File

@ -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);

View File

@ -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);

View File

@ -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);
} }

View File

@ -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);

View File

@ -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() {