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-08-04 09:07:41 +03:00
parent 7bc6923ed4
commit 32ec215d81
8 changed files with 47 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main/java/org/sadtech/autoresponder/AutoResponder.java
View File

@ -22,23 +22,23 @@ import java.util.Set;
* @author upagge [07/07/2019] * @author upagge [07/07/2019]
*/ */
@Slf4j @Slf4j
public class AutoResponder { public class AutoResponder<U extends Unit> {
@Description("Компоратор для сортировки Unit-ов") @Description("Компоратор для сортировки Unit-ов")
private static final UnitPriorityComparator UNIT_PRIORITY_COMPARATOR = new UnitPriorityComparator(); private final UnitPriorityComparator<U> unitPriorityComparator = new UnitPriorityComparator<>();
@Description("Начальные юниты, первый запрос приходит на них") @Description("Начальные юниты, первый запрос приходит на них")
private final Set<Unit> startUnits; private final Set<U> startUnits;
@Getter @Getter
@Setter @Setter
@Description("Дефолтный юнит, отправляется если ни один Unit не подошел") @Description("Дефолтный юнит, отправляется если ни один Unit не подошел")
private Unit defaultUnit; private U defaultUnit;
@Getter @Getter
private final UnitPointerService unitPointerService; private final UnitPointerService<U> unitPointerService;
public AutoResponder(UnitPointerService unitPointerService, Set<Unit> startUnits) { public AutoResponder(UnitPointerService<U> unitPointerService, Set<U> startUnits) {
this.startUnits = startUnits; this.startUnits = startUnits;
this.unitPointerService = unitPointerService; this.unitPointerService = unitPointerService;
} }
@ -50,14 +50,14 @@ public class AutoResponder {
* @param message Запрос пользователя - текстовое сообщение * @param message Запрос пользователя - текстовое сообщение
* @return {@link Unit}, который отвечает за данные для обработки данного запроса * @return {@link Unit}, который отвечает за данные для обработки данного запроса
*/ */
public Unit answer(@NonNull Integer personId, String message) { public U answer(@NonNull Integer personId, String message) {
UnitPointer unitPointer = checkAndAddPerson(personId); UnitPointer<U> unitPointer = checkAndAddPerson(personId);
Unit unit; U unit;
try { try {
if (unitPointer.getUnit() == null || unitPointer.getUnit().getNextUnits() == null || unitPointer.getUnit().getNextUnits().isEmpty()) { if (unitPointer.getUnit() == null || unitPointer.getUnit().getNextUnits() == null || unitPointer.getUnit().getNextUnits().isEmpty()) {
unit = nextUnit(startUnits, message); unit = nextUnit(startUnits, message);
} else { } else {
unit = nextUnit(unitPointer.getUnit().getNextUnits(), message); unit = (U) nextUnit(unitPointer.getUnit().getNextUnits(), message);
} }
unitPointerService.edit(personId, unit); unitPointerService.edit(personId, unit);
} catch (NotFoundUnitException e) { } catch (NotFoundUnitException e) {
@ -73,8 +73,8 @@ public class AutoResponder {
* @param message Запрос пользователя - текстовое сообщение * @param message Запрос пользователя - текстовое сообщение
* @return Юнит, который нуждается в обработке в соответствии с запросом пользователя * @return Юнит, который нуждается в обработке в соответствии с запросом пользователя
*/ */
private Unit nextUnit(@NonNull Set<Unit> nextUnits, String message) { private U nextUnit(@NonNull Set<U> nextUnits, String message) {
Set<Unit> searchUnit = new HashSet<>(); Set<U> searchUnit = new HashSet<>();
nextUnits.stream() nextUnits.stream()
.filter(nextUnit -> nextUnit.getPhrase() != null .filter(nextUnit -> nextUnit.getPhrase() != null
@ -97,7 +97,7 @@ public class AutoResponder {
.forEach(searchUnit::add); .forEach(searchUnit::add);
} }
return searchUnit.stream().max(UNIT_PRIORITY_COMPARATOR).orElseThrow(NotFoundUnitException::new); return searchUnit.stream().max(unitPriorityComparator).orElseThrow(NotFoundUnitException::new);
} }
/** /**
@ -106,12 +106,12 @@ public class AutoResponder {
* @param personId Идентификатор пользователя * @param personId Идентификатор пользователя
* @return {@link UnitPointer}, который сохраняет {@link Unit}, который был обработан последним * @return {@link UnitPointer}, который сохраняет {@link Unit}, который был обработан последним
*/ */
private UnitPointer checkAndAddPerson(@NonNull Integer personId) { private UnitPointer<U> checkAndAddPerson(@NonNull Integer personId) {
UnitPointer unitPointer; UnitPointer<U> unitPointer;
if (unitPointerService.check(personId)) { if (unitPointerService.check(personId)) {
unitPointer = unitPointerService.getByEntityId(personId); unitPointer = unitPointerService.getByEntityId(personId);
} else { } else {
unitPointer = new UnitPointer(personId); unitPointer = new UnitPointer<>(personId);
unitPointerService.add(unitPointer); unitPointerService.add(unitPointer);
} }
return unitPointer; return unitPointer;

2
src/main/java/org/sadtech/autoresponder/compare/UnitPriorityComparator.java
View File

@ -9,7 +9,7 @@ import java.util.Comparator;
* *
* @author upagge [07/07/2019] * @author upagge [07/07/2019]
*/ */
public class UnitPriorityComparator implements Comparator<Unit> { public class UnitPriorityComparator<U extends Unit> implements Comparator<U> {
@Override @Override
public int compare(Unit o1, Unit o2) { public int compare(Unit o1, Unit o2) {

6
src/main/java/org/sadtech/autoresponder/entity/Unit.java
View File

@ -19,7 +19,7 @@ import java.util.regex.Pattern;
@EqualsAndHashCode @EqualsAndHashCode
@ToString @ToString
@Setter @Setter
public abstract class Unit { public abstract class Unit<U extends Unit> {
@Description("Ключевые слова") @Description("Ключевые слова")
protected Set<String> keyWords; protected Set<String> keyWords;
@ -37,9 +37,9 @@ public abstract class Unit {
protected Integer priority; protected Integer priority;
@Description("Множество следующих Unit в сценарии") @Description("Множество следующих Unit в сценарии")
protected Set<Unit> nextUnits; protected Set<U> nextUnits;
protected Unit(Set<String> keyWords, String phrase, Pattern pattern, Integer matchThreshold, Integer priority, Set<Unit> nextUnits) { protected Unit(Set<String> keyWords, String phrase, Pattern pattern, Integer matchThreshold, Integer priority, Set<U> nextUnits) {
this.keyWords = keyWords; this.keyWords = keyWords;
this.phrase = phrase; this.phrase = phrase;
this.pattern = pattern; this.pattern = pattern;

4
src/main/java/org/sadtech/autoresponder/entity/UnitPointer.java
View File

@ -11,13 +11,13 @@ import org.sadtech.autoresponder.util.Description;
*/ */
@Data @Data
@AllArgsConstructor @AllArgsConstructor
public class UnitPointer { public class UnitPointer<U extends Unit> {
@Description("Идентификатор пользователя") @Description("Идентификатор пользователя")
private Integer entityId; private Integer entityId;
@Description("Юнит, который был обработан") @Description("Юнит, который был обработан")
private Unit unit; private U unit;
public UnitPointer(Integer entityId) { public UnitPointer(Integer entityId) {
this.entityId = entityId; this.entityId = entityId;

11
src/main/java/org/sadtech/autoresponder/repository/UnitPointerRepository.java
View File

@ -1,5 +1,6 @@
package org.sadtech.autoresponder.repository; package org.sadtech.autoresponder.repository;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.entity.UnitPointer; import org.sadtech.autoresponder.entity.UnitPointer;
import java.util.Collection; import java.util.Collection;
@ -9,20 +10,20 @@ import java.util.Collection;
* *
* @author upagge [07/07/2019] * @author upagge [07/07/2019]
*/ */
public interface UnitPointerRepository { public interface UnitPointerRepository<U extends Unit> {
void add(UnitPointer unitPointer); void add(UnitPointer<U> unitPointer);
void edit(UnitPointer unitPointer); void edit(UnitPointer<U> unitPointer);
void remove(Integer entityId); void remove(Integer entityId);
void addAll(Collection<UnitPointer> unitPointers); void addAll(Collection<UnitPointer<U>> unitPointers);
/** /**
* @param entityId Идентификатор пользователя * @param entityId Идентификатор пользователя
* @return Объект с последним обработанным {@link org.sadtech.autoresponder.entity.Unit} для пользователя * @return Объект с последним обработанным {@link org.sadtech.autoresponder.entity.Unit} для пользователя
*/ */
UnitPointer findByEntityId(Integer entityId); UnitPointer<U> findByEntityId(Integer entityId);
} }

13
src/main/java/org/sadtech/autoresponder/repository/UnitPointerRepositoryMap.java
View File

@ -1,6 +1,7 @@
package org.sadtech.autoresponder.repository; package org.sadtech.autoresponder.repository;
import lombok.NonNull; import lombok.NonNull;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.entity.UnitPointer; import org.sadtech.autoresponder.entity.UnitPointer;
import java.util.Collection; import java.util.Collection;
@ -12,17 +13,17 @@ import java.util.Map;
* *
* @author upagge [07/07/2019] * @author upagge [07/07/2019]
*/ */
public class UnitPointerRepositoryMap implements UnitPointerRepository { public class UnitPointerRepositoryMap<U extends Unit> implements UnitPointerRepository<U> {
private Map<Integer, UnitPointer> unitPointerMap = new HashMap<>(); private Map<Integer, UnitPointer<U>> unitPointerMap = new HashMap<>();
@Override @Override
public void add(@NonNull UnitPointer unitPointer) { public void add(@NonNull UnitPointer<U> unitPointer) {
unitPointerMap.put(unitPointer.getEntityId(), unitPointer); unitPointerMap.put(unitPointer.getEntityId(), unitPointer);
} }
@Override @Override
public void edit(@NonNull UnitPointer unitPointer) { public void edit(@NonNull UnitPointer<U> unitPointer) {
unitPointerMap.get(unitPointer.getEntityId()).setUnit(unitPointer.getUnit()); unitPointerMap.get(unitPointer.getEntityId()).setUnit(unitPointer.getUnit());
} }
@ -32,12 +33,12 @@ public class UnitPointerRepositoryMap implements UnitPointerRepository {
} }
@Override @Override
public void addAll(@NonNull Collection<UnitPointer> unitPointers) { public void addAll(@NonNull Collection<UnitPointer<U>> unitPointers) {
unitPointers.parallelStream().forEach(unitPointer -> unitPointerMap.put(unitPointer.getEntityId(), unitPointer)); unitPointers.parallelStream().forEach(unitPointer -> unitPointerMap.put(unitPointer.getEntityId(), unitPointer));
} }
@Override @Override
public UnitPointer findByEntityId(@NonNull Integer entityId) { public UnitPointer<U> findByEntityId(@NonNull Integer entityId) {
return unitPointerMap.get(entityId); return unitPointerMap.get(entityId);
} }
} }

9
src/main/java/org/sadtech/autoresponder/service/UnitPointerService.java
View File

@ -8,19 +8,20 @@ import org.sadtech.autoresponder.entity.UnitPointer;
* *
* @author upagge [07/07/2019] * @author upagge [07/07/2019]
*/ */
public interface UnitPointerService { public interface UnitPointerService<U extends Unit> {
void add(UnitPointer unitPointer); void add(UnitPointer<U> unitPointer);
/** /**
* Проверка наличия {@link UnitPointer} для пользователя * Проверка наличия {@link UnitPointer} для пользователя
*
* @param entityId Идентификатор пользователя * @param entityId Идентификатор пользователя
* @return true - если найдено * @return true - если найдено
*/ */
boolean check(Integer entityId); boolean check(Integer entityId);
UnitPointer getByEntityId(Integer entityId); UnitPointer<U> getByEntityId(Integer entityId);
void edit(Integer personId, Unit unit); void edit(Integer personId, U unit);
} }

14
src/main/java/org/sadtech/autoresponder/service/UnitPointerServiceImpl.java
View File

@ -7,28 +7,28 @@ import org.sadtech.autoresponder.entity.UnitPointer;
import org.sadtech.autoresponder.repository.UnitPointerRepository; import org.sadtech.autoresponder.repository.UnitPointerRepository;
@Slf4j @Slf4j
public class UnitPointerServiceImpl implements UnitPointerService { public class UnitPointerServiceImpl<U extends Unit> implements UnitPointerService<U> {
private final UnitPointerRepository unitPointerRepository; private final UnitPointerRepository<U> unitPointerRepository;
public UnitPointerServiceImpl(UnitPointerRepository unitPointerRepository) { public UnitPointerServiceImpl(UnitPointerRepository<U> unitPointerRepository) {
this.unitPointerRepository = unitPointerRepository; this.unitPointerRepository = unitPointerRepository;
} }
@Override @Override
public UnitPointer getByEntityId(@NonNull Integer entityId) { public UnitPointer<U> getByEntityId(@NonNull Integer entityId) {
return unitPointerRepository.findByEntityId(entityId); return unitPointerRepository.findByEntityId(entityId);
} }
@Override @Override
public void edit(@NonNull Integer personId, Unit unit) { public void edit(@NonNull Integer personId, U unit) {
if (check(personId)) { if (check(personId)) {
unitPointerRepository.edit(new UnitPointer(personId, unit)); unitPointerRepository.edit(new UnitPointer<>(personId, unit));
} }
} }
@Override @Override
public void add(@NonNull UnitPointer unitPointer) { public void add(@NonNull UnitPointer<U> unitPointer) {
unitPointerRepository.add(unitPointer); unitPointerRepository.add(unitPointer);
log.info("Пользователь отправлен в репозиторий"); log.info("Пользователь отправлен в репозиторий");
} }