Изменил бизнес-логику

Теперь автоответчик возвращает не строку, а любой объект, который является наследником unit. Для этого:

* Перенес поиск следующего юнита в основной класс
* Добавил компоратор для сортировки юнитов по приоритету
* Юнит теперь абстрактный класс
* Удален репозиторий юнита
* У юнита удалено поле id
* ЮнитСервис теперь хранит в себе все ЮнитРепозитории
This commit is contained in:
Mark Struchkov 2019-01-20 22:59:33 +03:00
parent 2f05b55706
commit 65ecc3d38c
11 changed files with 312 additions and 237 deletions

View File

@ -1,42 +1,38 @@
package org.sadtech.autoresponder;
import lombok.AllArgsConstructor;
import lombok.NonNull;
import org.sadtech.autoresponder.entity.Person;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.entity.compare.UnitPriorityComparator;
import org.sadtech.autoresponder.service.PersonService;
import org.sadtech.autoresponder.service.UnitService;
import org.sadtech.autoresponder.submodule.insertwords.InsertWords;
import org.sadtech.autoresponder.submodule.parser.Parser;
import java.util.Collections;
import java.util.List;
@AllArgsConstructor
public class Autoresponder {
private UnitService unitService;
private PersonService personService;
public String answer(@NonNull Integer idPerson, @NonNull String message) {
Person person = CheckAndAddPerson(idPerson);
public Autoresponder(UnitService unitService, PersonService personService) {
this.unitService = unitService;
this.personService = personService;
}
public Unit answer(Integer idPerson, String message) {
Person person = checkAndAddPerson(idPerson);
Unit unit;
if (person.getUnit() == null) {
unit = unitService.nextUnit(unitService.menuUnit(), message);
unit = nextUnit(unitService.menuUnit(), message);
} else {
unit = unitService.nextUnit(person.getUnit(), message);
unit = nextUnit(person.getUnit().getNextUnits(), message);
}
person.setUnit(unit);
return unit.getAnswer();
return unit;
}
public String answer(@NonNull Integer idPerson, @NonNull String message, @NonNull List<String> words) {
String answer = answer(idPerson, message);
InsertWords insertWords = new InsertWords();
insertWords.setInText(answer);
insertWords.insert(words);
return insertWords.getOutText();
}
private Person CheckAndAddPerson(Integer idPerson) {
private Person checkAndAddPerson(Integer idPerson) {
Person person;
if (personService.checkPerson(idPerson)) {
person = personService.getPersonById(idPerson);
@ -48,4 +44,15 @@ public class Autoresponder {
return person;
}
private Unit nextUnit(List<Unit> nextUnits, String message) {
if (nextUnits.size() > 0) {
Parser parser = new Parser();
parser.setText(message);
parser.parse();
return nextUnits.stream().filter(nextUnit -> !Collections.disjoint(nextUnit.getKeyWords(), parser.getWords())).max(new UnitPriorityComparator()).get();
} else {
return null;
}
}
}

View File

@ -1,11 +1,48 @@
package org.sadtech.autoresponder.entity;
import lombok.Data;
import java.util.Objects;
@Data
public class Person {
private Integer id;
private Unit unit;
public Person(Integer id, Unit unit) {
this.id = id;
this.unit = unit;
}
public Person(Integer id) {
this.id = id;
}
public Integer getId() {
return id;
}
public void setId(Integer id) {
this.id = id;
}
public Unit getUnit() {
return unit;
}
public void setUnit(Unit unit) {
this.unit = unit;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Person person = (Person) o;
return Objects.equals(id, person.id) &&
Objects.equals(unit, person.unit);
}
@Override
public int hashCode() {
return Objects.hash(id, unit);
}
}

View File

@ -1,19 +1,83 @@
package org.sadtech.autoresponder.entity;
import lombok.Data;
import java.util.List;
import java.util.Objects;
import java.util.Set;
@Data
public class Unit {
public abstract class Unit {
private Integer idUnit;
private List<Unit> nextUnits;
private String answer;
private Integer priority;
private Set<String> keyWords;
private Integer matchThreshold;
private Boolean level = false;
private Integer priority;
private Boolean level;
private List<Unit> nextUnits;
public Unit() {
level=false;
}
public Unit(Set<String> keyWords, Integer matchThreshold, Integer priority, Boolean level, List<Unit> nextUnits) {
this.keyWords = keyWords;
this.matchThreshold = matchThreshold;
this.priority = priority;
this.level = level;
this.nextUnits = nextUnits;
}
public Set<String> getKeyWords() {
return keyWords;
}
public void setKeyWords(Set<String> keyWords) {
this.keyWords = keyWords;
}
public Integer getMatchThreshold() {
return matchThreshold;
}
public void setMatchThreshold(Integer matchThreshold) {
this.matchThreshold = matchThreshold;
}
public Integer getPriority() {
return priority;
}
public void setPriority(Integer priority) {
this.priority = priority;
}
public Boolean getLevel() {
return level;
}
public void setLevel(Boolean level) {
this.level = level;
}
public List<Unit> getNextUnits() {
return nextUnits;
}
public void setNextUnits(List<Unit> nextUnits) {
this.nextUnits = nextUnits;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Unit unit = (Unit) o;
return Objects.equals(keyWords, unit.keyWords) &&
Objects.equals(matchThreshold, unit.matchThreshold) &&
Objects.equals(priority, unit.priority) &&
Objects.equals(level, unit.level) &&
Objects.equals(nextUnits, unit.nextUnits);
}
@Override
public int hashCode() {
return Objects.hash(keyWords, matchThreshold, priority, level, nextUnits);
}
}

View File

@ -0,0 +1,19 @@
package org.sadtech.autoresponder.entity.compare;
import org.sadtech.autoresponder.entity.Unit;
import java.util.Comparator;
public class UnitPriorityComparator implements Comparator<Unit> {
@Override
public int compare(Unit o1, Unit o2) {
if (o1.getPriority() > o2.getPriority()) {
return -1;
} else if (o1.getPriority().equals(o2.getPriority())) {
return 0;
}
return 1;
}
}

View File

@ -8,8 +8,11 @@ import java.util.Map;
public interface PersonRepository {
void addPerson(Person person);
void removePerson(Person person);
void addPersonAll(Map<Integer, Person> personCollection);
Person getPersonById(Integer idPerson);
}

View File

@ -7,8 +7,6 @@ import java.util.List;
public interface UnitRepository {
Unit getUnitById(Integer idUnit);
void addUnit(Unit unit);
void addUnits(Collection<Unit> units);
@ -16,4 +14,5 @@ public interface UnitRepository {
void removeUnit(Unit idUnit);
List<Unit> menuUnits();
}

View File

@ -1,42 +0,0 @@
package org.sadtech.autoresponder.repository.impl;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.repository.UnitRepository;
import java.util.*;
public class UnitRepositoryMap implements UnitRepository {
private Map<Integer, Unit> units = new HashMap<>();
@Override
public Unit getUnitById(Integer idUnit) {
return units.get(idUnit);
}
@Override
public void addUnit(Unit unit) {
units.put(unit.getIdUnit(), unit);
}
@Override
public void addUnits(Collection<Unit> units) {
units.addAll(units);
}
@Override
public void removeUnit(Unit idUnit) {
units.remove(idUnit.getIdUnit());
}
@Override
public List<Unit> menuUnits() {
ArrayList<Unit> unitsMenu = new ArrayList<>();
for (Integer integer : units.keySet()) {
if (units.get(integer).getLevel()) {
unitsMenu.add(units.get(integer));
}
}
return unitsMenu;
}
}

View File

@ -1,13 +1,14 @@
package org.sadtech.autoresponder.service;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.repository.UnitRepository;
import java.util.List;
public interface UnitService {
Unit nextUnit(Unit unit, String message);
List<Unit> menuUnit();
Unit getUnitById(Integer idUnit);
Unit menuUnit();
void addUnitRepository(UnitRepository unitRepository);
}

View File

@ -1,15 +1,17 @@
package org.sadtech.autoresponder.service.impl;
import lombok.AllArgsConstructor;
import org.sadtech.autoresponder.entity.Person;
import org.sadtech.autoresponder.repository.PersonRepository;
import org.sadtech.autoresponder.service.PersonService;
@AllArgsConstructor
public class PersonServiceImpl implements PersonService {
private PersonRepository personRepository;
public PersonServiceImpl(PersonRepository personRepository) {
this.personRepository = personRepository;
}
@Override
public Person getPersonById(Integer integer) {
return personRepository.getPersonById(integer);

View File

@ -1,52 +1,38 @@
package org.sadtech.autoresponder.service.impl;
import lombok.AllArgsConstructor;
import lombok.NonNull;
import lombok.extern.log4j.Log4j;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.repository.UnitRepository;
import org.sadtech.autoresponder.service.UnitService;
import org.sadtech.autoresponder.submodule.parser.Parser;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
@Log4j
@AllArgsConstructor
public class UnitServiceImpl implements UnitService {
private UnitRepository unitRepository;
private List<UnitRepository> unitRepositories;
public Unit nextUnit(Unit unit, @NonNull String message) {
ArrayList<Unit> nextUnits = (ArrayList<Unit>) unit.getNextUnits();
if (nextUnits.size() > 0) {
Parser parser = new Parser();
parser.setText(message);
parser.parse();
Unit unitReturn = new Unit();
unitReturn.setPriority(0);
for (Unit nextUnit : nextUnits) {
if (!Collections.disjoint(nextUnit.getKeyWords(), parser.getWords()) && (nextUnit.getPriority() > unitReturn.getPriority())) {
unitReturn = nextUnit;
}
}
return unitReturn;
} else {
return null;
public UnitServiceImpl() {
unitRepositories = new ArrayList<>();
}
public UnitServiceImpl(List<UnitRepository> unitRepositories) {
this.unitRepositories = unitRepositories;
}
@Override
public List<Unit> menuUnit() {
List<Unit> units = new ArrayList<>();
for (UnitRepository unitRepository : unitRepositories) {
units.addAll(unitRepository.menuUnits());
}
return units;
}
@Override
public Unit getUnitById(@NonNull Integer idUnit) {
return unitRepository.getUnitById(idUnit);
}
@Override
public Unit menuUnit() {
Unit menuUnit = new Unit();
menuUnit.setNextUnits(unitRepository.menuUnits());
return menuUnit;
public void addUnitRepository(UnitRepository unitRepository) {
unitRepositories.add(unitRepository);
}

View File

@ -1,127 +1,126 @@
package org.sadtech.autoresponder;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.sadtech.autoresponder.entity.Person;
import org.sadtech.autoresponder.entity.Unit;
import org.sadtech.autoresponder.repository.impl.PersonRepositoryMap;
import org.sadtech.autoresponder.repository.impl.UnitRepositoryMap;
import org.sadtech.autoresponder.service.impl.PersonServiceImpl;
import org.sadtech.autoresponder.service.impl.UnitServiceImpl;
import java.util.ArrayList;
import java.util.HashSet;
public class AutoresponderTest {
private Person person = new Person();
private Unit unit = new Unit();
private Unit unit2 = new Unit();
private Unit unit3 = new Unit();
private ArrayList<Unit> units = new ArrayList<>();
private UnitRepositoryMap unitRepository = new UnitRepositoryMap();
private PersonRepositoryMap personRepository = new PersonRepositoryMap();
private UnitServiceImpl unitService = new UnitServiceImpl(unitRepository);
private PersonServiceImpl personService = new PersonServiceImpl(personRepository);
private Autoresponder autoresponder = new Autoresponder(unitService, personService);
@Before
public void before() {
HashSet<String> words = new HashSet<>();
HashSet<String> words2 = new HashSet<>();
words.add("тест");
words2.add("тест");
words2.add("привет");
unit.setIdUnit(1);
unit.setLevel(true);
unit.setPriority(50);
unit.setKeyWords(words);
unit.setAnswer("Здравствуйте, вы написали в нашу компанию!");
unit.setMatchThreshold(100);
units.add(unit2);
units.add(unit3);
unit.setNextUnits(units);
unit2.setIdUnit(2);
unit2.setAnswer("Ответ с {0} параметрами!");
unit2.setPriority(60);
unit2.setKeyWords(words);
unit2.setMatchThreshold(100);
unit3.setIdUnit(3);
unit3.setAnswer("Второй Ответ с {0} параметрами!");
unit3.setPriority(50);
unit3.setKeyWords(words2);
unit3.setMatchThreshold(100);
person.setUnit(unit);
person.setId(1);
unitRepository.addUnit(unit);
unitRepository.addUnit(unit2);
personRepository.addPerson(person);
}
@Test
public void usualAnswer() {
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
}
@Test
public void answerOneParameter() {
ArrayList<String> words = new ArrayList<>();
words.add("одним");
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с одним параметрами!");
}
@Test
public void NoAnswer() {
person.setUnit(null);
String test = autoresponder.answer(person.getId(), "Привет это срабатывания");
Assert.assertNull(test);
}
@Test
public void answerTwoParameter() {
ArrayList<String> words = new ArrayList<>();
words.add("одним");
words.add("двумя");
unit2.setAnswer("Ответ с {0} и {1}");
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с одним и двумя");
}
@Test
public void incorrectSettingsWords() {
ArrayList<String> words = new ArrayList<>();
words.add("одним");
words.add("двумя");
unit2.setAnswer("Ответ с {1} и {3}");
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с двумя и {3}");
}
@Test
public void answerRepeatSingleParameter() {
ArrayList<String> words = new ArrayList<>();
words.add("одним");
words.add("двумя");
unit2.setAnswer("Ответ с {1} и {1}");
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с двумя и двумя");
}
@Test
public void answerByPriority() {
Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
}
@Test
public void answerNoPerson() {
Assert.assertEquals(autoresponder.answer(100, "Привет это тест срабатывания"), "Здравствуйте, вы написали в нашу компанию!");
Assert.assertEquals(autoresponder.answer(100, "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
}
//package org.sadtech.autoresponder;
//
//import org.junit.Assert;
//import org.junit.Before;
//import org.junit.Test;
//import org.sadtech.autoresponder.entity.Person;
//import org.sadtech.autoresponder.entity.Unit;
//import org.sadtech.autoresponder.repository.impl.PersonRepositoryMap;
//import org.sadtech.autoresponder.service.impl.PersonServiceImpl;
//import org.sadtech.autoresponder.service.impl.UnitServiceImpl;
//
//import java.util.ArrayList;
//import java.util.HashSet;
//
//public class AutoresponderTest {
//
// private Person person = new Person();
// private Unit unit = new Unit();
// private Unit unit2 = new Unit();
// private Unit unit3 = new Unit();
// private ArrayList<Unit> units = new ArrayList<>();
// private UnitRepositoryMap unitRepository = new UnitRepositoryMap();
// private PersonRepositoryMap personRepository = new PersonRepositoryMap();
//
// private UnitServiceImpl unitService = new UnitServiceImpl(unitRepository);
// private PersonServiceImpl personService = new PersonServiceImpl(personRepository);
// private Autoresponder autoresponder = new Autoresponder(unitService, personService);
//
//
// @Before
// public void before() {
// HashSet<String> words = new HashSet<>();
// HashSet<String> words2 = new HashSet<>();
// words.add("тест");
// words2.add("тест");
// words2.add("привет");
//
// unit.setIdUnit(1);
// unit.setLevel(true);
// unit.setPriority(50);
// unit.setKeyWords(words);
// unit.setAnswer("Здравствуйте, вы написали в нашу компанию!");
// unit.setMatchThreshold(100);
//
// units.add(unit2);
// units.add(unit3);
//
// unit.setNextUnits(units);
//
// unit2.setIdUnit(2);
// unit2.setAnswer("Ответ с {0} параметрами!");
// unit2.setPriority(60);
// unit2.setKeyWords(words);
// unit2.setMatchThreshold(100);
//
// unit3.setIdUnit(3);
// unit3.setAnswer("Второй Ответ с {0} параметрами!");
// unit3.setPriority(50);
// unit3.setKeyWords(words2);
// unit3.setMatchThreshold(100);
//
// person.setUnit(unit);
// person.setId(1);
//
// unitRepository.addUnit(unit);
// unitRepository.addUnit(unit2);
// personRepository.addPerson(person);
// }
//
// @Test
// public void usualAnswer() {
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
// }
//
// @Test
// public void answerOneParameter() {
// ArrayList<String> words = new ArrayList<>();
// words.add("одним");
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с одним параметрами!");
// }
//
// @Test
// public void NoAnswer() {
// person.setUnit(null);
// String test = autoresponder.answer(person.getId(), "Привет это срабатывания");
// Assert.assertNull(test);
// }
//
// @Test
// public void answerTwoParameter() {
// ArrayList<String> words = new ArrayList<>();
// words.add("одним");
// words.add("двумя");
// unit2.setAnswer("Ответ с {0} и {1}");
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с одним и двумя");
// }
//
// @Test
// public void incorrectSettingsWords() {
// ArrayList<String> words = new ArrayList<>();
// words.add("одним");
// words.add("двумя");
// unit2.setAnswer("Ответ с {1} и {3}");
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с двумя и {3}");
// }
//
// @Test
// public void answerRepeatSingleParameter() {
// ArrayList<String> words = new ArrayList<>();
// words.add("одним");
// words.add("двумя");
// unit2.setAnswer("Ответ с {1} и {1}");
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания", words), "Ответ с двумя и двумя");
// }
//
// @Test
// public void answerByPriority() {
// Assert.assertEquals(autoresponder.answer(person.getId(), "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
// }
//
// @Test
// public void answerNoPerson() {
// Assert.assertEquals(autoresponder.answer(100, "Привет это тест срабатывания"), "Здравствуйте, вы написали в нашу компанию!");
// Assert.assertEquals(autoresponder.answer(100, "Привет это тест срабатывания"), "Ответ с {0} параметрами!");
// }
}