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-07-14 14:12:45 +03:00
parent 74ef11e253
commit 7bc6923ed4
3 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -76,6 +76,13 @@ public class AutoResponder {
private Unit nextUnit(@NonNull Set<Unit> nextUnits, String message) {
Set<Unit> searchUnit = new HashSet<>();
nextUnits.stream()
.filter(nextUnit -> nextUnit.getPhrase() != null
&& !nextUnit.getPhrase().isEmpty()
&& nextUnit.getPhrase().equalsIgnoreCase(message)
)
.forEach(searchUnit::add);
nextUnits.stream()
.filter(nextUnit -> nextUnit.getPattern() != null && patternReg(nextUnit, message))
.forEach(searchUnit::add);
@ -84,9 +91,11 @@ public class AutoResponder {
.filter(nextUnit -> percentageMatch(nextUnit, Parser.parse(message)) >= nextUnit.getMatchThreshold())
.forEach(searchUnit::add);
nextUnits.stream()
.filter(nextUnit -> (nextUnit.getPattern() == null && (nextUnit.getKeyWords() == null || nextUnit.getKeyWords().isEmpty())))
.forEach(searchUnit::add);
if (searchUnit.isEmpty()) {
nextUnits.stream()
.filter(nextUnit -> (nextUnit.getPattern() == null && (nextUnit.getKeyWords() == null || nextUnit.getKeyWords().isEmpty())))
.forEach(searchUnit::add);
}
return searchUnit.stream().max(UNIT_PRIORITY_COMPARATOR).orElseThrow(NotFoundUnitException::new);
}

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

@ -24,6 +24,9 @@ public abstract class Unit {
@Description("Ключевые слова")
protected Set<String> keyWords;
@Description("Точная фраза")
protected String phrase;
@Description("Регулярное выражение")
protected Pattern pattern;
@ -36,8 +39,9 @@ public abstract class Unit {
@Description("Множество следующих Unit в сценарии")
protected Set<Unit> nextUnits;
protected Unit(Set<String> keyWords, Pattern pattern, Integer matchThreshold, Integer priority, Set<Unit> nextUnits) {
protected Unit(Set<String> keyWords, String phrase, Pattern pattern, Integer matchThreshold, Integer priority, Set<Unit> nextUnits) {
this.keyWords = keyWords;
this.phrase = phrase;
this.pattern = pattern;
this.matchThreshold = Optional.ofNullable(matchThreshold).orElse(10);
this.priority = Optional.ofNullable(priority).orElse(10);

8
src/test/java/org/sadtech/autoresponder/test/TestUnit.java
View File

@ -18,13 +18,15 @@ public class TestUnit extends Unit {
private String message;
@Builder
public TestUnit(@Singular(value = "keyWord") Set<String> keyWords,
public TestUnit(@Singular Set<String> keyWords,
String phrase,
Pattern pattern,
Integer matchThreshold,
Integer priority,
@Singular(value = "nextUnit") Set<Unit> nextUnits,
@Singular Set<Unit> nextUnits,
String message) {
super(keyWords, pattern, matchThreshold, priority, nextUnits);
super(keyWords, phrase, pattern, matchThreshold, priority, nextUnits);
this.message = message;
}
}