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-01-21 10:06:53 +03:00
parent b93caa67e2
commit f391f37f13
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pom.xml
View File

@ -6,7 +6,7 @@
<groupId>org.sadtech.autoresponder</groupId>
<artifactId>core</artifactId>
<version>1.1.0-RELEASE</version>
<version>1.1.1-SNAPSHOT</version>
<build>
<plugins>
<plugin>

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

@ -1,5 +1,6 @@
package org.sadtech.autoresponder;
import lombok.extern.log4j.Log4j;
import org.sadtech.autoresponder.entity.Person;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.entity.compare.UnitPriorityComparator;
@ -7,9 +8,11 @@ import org.sadtech.autoresponder.service.PersonService;
import org.sadtech.autoresponder.service.UnitService;
import org.sadtech.autoresponder.submodule.parser.Parser;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
@Log4j
public class Autoresponder {
private UnitService unitService;
@ -48,10 +51,17 @@ public class Autoresponder {
Parser parser = new Parser();
parser.setText(message);
parser.parse();
return nextUnits.stream().filter(nextUnit -> !Collections.disjoint(nextUnit.getKeyWords(), parser.getWords())).max(new UnitPriorityComparator()).get();
return nextUnits.stream().filter(nextUnit -> textPercentageMatch(nextUnit, parser.getWords()) >= nextUnit.getMatchThreshold()).max(new UnitPriorityComparator()).get();
} else {
return null;
}
}
private Double textPercentageMatch(Unit unit, Set<String> words) {
Set<String> temp = new HashSet<>(unit.getKeyWords());
temp.retainAll(words);
log.info((temp.size() / unit.getKeyWords().size()) * 100);
return (double) (temp.size() / unit.getKeyWords().size()) * 100;
}
}