private void updateExistingProfilesForScoring(Scoring scoring) {
long t = System.currentTimeMillis();
Condition scoringCondition = new Condition();
scoringCondition.setConditionType(
definitionsService.getConditionType("profilePropertyCondition"));
scoringCondition.setParameter("propertyName", "scores." + scoring.getItemId());
scoringCondition.setParameter("comparisonOperator", "exists");
List<Profile> previousProfiles =
persistenceService.query(scoringCondition, null, Profile.class);
HashMap<String, Object> scriptParams = new HashMap<>();
scriptParams.put("scoringId", scoring.getItemId());
for (Profile profileToRemove : previousProfiles) {
persistenceService.updateWithScript(
profileToRemove.getItemId(),
null,
Profile.class,
"if (ctx._source.systemProperties.scoreModifiers == null) { ctx._source.systemProperties.scoreModifiers=[:] } ; if (ctx._source.systemProperties.scoreModifiers.containsKey(scoringId)) { ctx._source.scores[scoringId] = ctx._source.systemProperties.scoreModifiers[scoringId] } else { ctx._source.scores.remove(scoringId) }",
scriptParams);
}
if (scoring.getMetadata().isEnabled()) {
String script =
"if (ctx._source.scores == null) { ctx._source.scores=[:] } ; if (ctx._source.scores.containsKey(scoringId)) { ctx._source.scores[scoringId] += scoringValue } else { ctx._source.scores[scoringId] = scoringValue }";
Map<String, Event> updatedProfiles = new HashMap<>();
for (ScoringElement element : scoring.getElements()) {
scriptParams.put("scoringValue", element.getValue());
for (Profile p : persistenceService.query(element.getCondition(), null, Profile.class)) {
persistenceService.updateWithScript(
p.getItemId(), null, Profile.class, script, scriptParams);
Event profileUpdated = new Event("profileUpdated", null, p, null, null, p, new Date());
profileUpdated.setPersistent(false);
updatedProfiles.put(p.getItemId(), profileUpdated);
}
}
Iterator<Map.Entry<String, Event>> entries = updatedProfiles.entrySet().iterator();
while (entries.hasNext()) {
eventService.send(entries.next().getValue());
}
}
logger.info("Profiles updated in {}", System.currentTimeMillis() - t);
}
private void updateExistingProfilesForRemovedScoring(String scoringId) {
long t = System.currentTimeMillis();
Condition scoringCondition = new Condition();
scoringCondition.setConditionType(
definitionsService.getConditionType("profilePropertyCondition"));
scoringCondition.setParameter("propertyName", "scores." + scoringId);
scoringCondition.setParameter("comparisonOperator", "exists");
List<Profile> previousProfiles =
persistenceService.query(scoringCondition, null, Profile.class);
HashMap<String, Object> scriptParams = new HashMap<>();
scriptParams.put("scoringId", scoringId);
for (Profile profileToRemove : previousProfiles) {
persistenceService.updateWithScript(
profileToRemove.getItemId(),
null,
Profile.class,
"ctx._source.scores.remove(scoringId)",
scriptParams);
}
logger.info("Profiles updated in {}", System.currentTimeMillis() - t);
}
@Before
public void setUp() {
assertNotNull("Definition service should be available", definitionsService);
assertNotNull("Persistence service should be available", persistenceService);
lastVisit = new GregorianCalendar(2015, 1, 1, 20, 30, 0).getTime();
Profile profile = new Profile("profile-" + UUID.randomUUID().toString());
profile.setProperty("firstVisit", lastVisit);
profile.setProperty("age", Integer.valueOf(30));
profile.setProperty("gender", "female");
profile.setProperty("lastVisit", lastVisit);
profile.setSegments(new HashSet<String>(Arrays.asList("s1", "s2", "s3")));
this.item = profile;
builder = new ConditionBuilder(definitionsService);
}
public SegmentsAndScores getSegmentsAndScoresForProfile(Profile profile) {
Set<String> segments = new HashSet<String>();
Map<String, Integer> scores = new HashMap<String, Integer>();
List<Segment> allSegments = this.allSegments;
for (Segment segment : allSegments) {
if (persistenceService.testMatch(segment.getCondition(), profile)) {
segments.add(segment.getMetadata().getId());
}
}
List<Scoring> allScoring = this.allScoring;
Map<String, Integer> scoreModifiers =
(Map<String, Integer>) profile.getSystemProperties().get("scoreModifiers");
for (Scoring scoring : allScoring) {
if (scoring.getMetadata().isEnabled()) {
int score = 0;
for (ScoringElement scoringElement : scoring.getElements()) {
if (persistenceService.testMatch(scoringElement.getCondition(), profile)) {
score += scoringElement.getValue();
}
}
String scoringId = scoring.getMetadata().getId();
if (scoreModifiers != null
&& scoreModifiers.containsKey(scoringId)
&& scoreModifiers.get(scoringId) != null) {
score += scoreModifiers.get(scoringId);
}
if (score > 0) {
scores.put(scoringId, score);
}
}
}
return new SegmentsAndScores(segments, scores);
}
private void updateExistingProfilesForSegment(Segment segment) {
long t = System.currentTimeMillis();
Condition segmentCondition = new Condition();
segmentCondition.setConditionType(
definitionsService.getConditionType("profilePropertyCondition"));
segmentCondition.setParameter("propertyName", "segments");
segmentCondition.setParameter("comparisonOperator", "equals");
segmentCondition.setParameter("propertyValue", segment.getItemId());
if (segment.getMetadata().isEnabled()) {
List<Profile> previousProfiles =
persistenceService.query(segmentCondition, null, Profile.class);
List<Profile> newProfiles =
persistenceService.query(segment.getCondition(), null, Profile.class);
List<Profile> add = new ArrayList<>(newProfiles);
add.removeAll(previousProfiles);
previousProfiles.removeAll(newProfiles);
for (Profile profileToAdd : add) {
profileToAdd.getSegments().add(segment.getItemId());
persistenceService.update(
profileToAdd.getItemId(), null, Profile.class, "segments", profileToAdd.getSegments());
Event profileUpdated =
new Event("profileUpdated", null, profileToAdd, null, null, profileToAdd, new Date());
profileUpdated.setPersistent(false);
eventService.send(profileUpdated);
}
for (Profile profileToRemove : previousProfiles) {
profileToRemove.getSegments().remove(segment.getItemId());
persistenceService.update(
profileToRemove.getItemId(),
null,
Profile.class,
"segments",
profileToRemove.getSegments());
Event profileUpdated =
new Event(
"profileUpdated", null, profileToRemove, null, null, profileToRemove, new Date());
profileUpdated.setPersistent(false);
eventService.send(profileUpdated);
}
} else {
List<Profile> previousProfiles =
persistenceService.query(segmentCondition, null, Profile.class);
for (Profile profileToRemove : previousProfiles) {
profileToRemove.getSegments().remove(segment.getItemId());
persistenceService.update(
profileToRemove.getItemId(),
null,
Profile.class,
"segments",
profileToRemove.getSegments());
}
}
logger.info("Profiles updated in {}", System.currentTimeMillis() - t);
}
public DependentMetadata removeSegmentDefinition(String segmentId, boolean validate) {
Set<Segment> impactedSegments = getSegmentDependentSegments(segmentId);
Set<Scoring> impactedScorings = getSegmentDependentScorings(segmentId);
if (!validate || (impactedSegments.isEmpty() && impactedScorings.isEmpty())) {
// update profiles
Condition segmentCondition = new Condition();
segmentCondition.setConditionType(
definitionsService.getConditionType("profilePropertyCondition"));
segmentCondition.setParameter("propertyName", "segments");
segmentCondition.setParameter("comparisonOperator", "equals");
segmentCondition.setParameter("propertyValue", segmentId);
List<Profile> previousProfiles =
persistenceService.query(segmentCondition, null, Profile.class);
for (Profile profileToRemove : previousProfiles) {
profileToRemove.getSegments().remove(segmentId);
persistenceService.update(
profileToRemove.getItemId(),
null,
Profile.class,
"segments",
profileToRemove.getSegments());
}
// update impacted segments
for (Segment segment : impactedSegments) {
Condition updatedCondition =
updateSegmentDependentCondition(segment.getCondition(), segmentId);
segment.setCondition(updatedCondition);
if (updatedCondition == null) {
clearAutoGeneratedRules(
persistenceService.query(
"linkedItems", segment.getMetadata().getId(), null, Rule.class),
segment.getMetadata().getId());
segment.getMetadata().setEnabled(false);
}
setSegmentDefinition(segment);
}
// update impacted scorings
for (Scoring scoring : impactedScorings) {
List<ScoringElement> updatedScoringElements = new ArrayList<>();
for (ScoringElement scoringElement : scoring.getElements()) {
Condition updatedCondition =
updateSegmentDependentCondition(scoringElement.getCondition(), segmentId);
if (updatedCondition != null) {
scoringElement.setCondition(updatedCondition);
updatedScoringElements.add(scoringElement);
}
}
scoring.setElements(updatedScoringElements);
if (updatedScoringElements.isEmpty()) {
clearAutoGeneratedRules(
persistenceService.query(
"linkedItems", scoring.getMetadata().getId(), null, Rule.class),
scoring.getMetadata().getId());
scoring.getMetadata().setEnabled(false);
}
setScoringDefinition(scoring);
}
persistenceService.remove(segmentId, Segment.class);
List<Rule> previousRules =
persistenceService.query("linkedItems", segmentId, null, Rule.class);
clearAutoGeneratedRules(previousRules, segmentId);
}
List<Metadata> segments = new LinkedList<>();
List<Metadata> scorings = new LinkedList<>();
for (Segment definition : impactedSegments) {
segments.add(definition.getMetadata());
}
for (Scoring definition : impactedScorings) {
scorings.add(definition.getMetadata());
}
return new DependentMetadata(segments, scorings);
}