com.intuit.fuzzymatcher.domain.Match Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fuzzy-matcher Show documentation

A java library to determine probability of objects being similar

There is a newer version: 1.2.1

package com.intuit.fuzzymatcher.domain;

import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * 
 * A generic class to hold the match between 2 objects and the score of the match result.
 * A match between similar Token, Element or Document is represented by this class.
 * 
 * The "data" and "matchedWith" object holds the 2 records that matched. And "score" represents the match for these 2 objects.
 * "childScore" is used by ScoringFunction to aggregate and calculate the "score" value
 */
public class Match {

    public Match(T t, T matchedWith, List childScores) {
        this.data = t;
        this.matchedWith = matchedWith;
        this.childScores = childScores;
    }

    public Match(T t, T matchedWith, double result) {
        this.data = t;
        this.matchedWith = matchedWith;
        this.score = new Score(result, this);
    }

    private T data;

    private T matchedWith;

    private Score score;

    private List childScores;

    public T getData() {
        return this.data;
    }

    public T getMatchedWith() {
        return matchedWith;
    }

    public void setMatchedWith(T matchedWith) {
        this.matchedWith = matchedWith;
    }

    public double getResult() {
        return this.getScore().getResult();
    }

    public Score getScore() {
        if (this.score == null) {
            this.score = this.data.getScoringFunction().apply(this);
        }
        return this.score;
    }

    public List getChildScores() {
        return getMaxDistinctScores(this.childScores);
    }

    private List getMaxDistinctScores(List scoreList) {
        Map> map = scoreList.stream()
                .collect(Collectors.groupingBy(s -> s.getMatch().getData(),
                        Collectors.maxBy(Comparator.comparingDouble(Score::getResult))));

        return map.entrySet().stream().map(entry -> entry.getValue().get()).collect(Collectors.toList());
    }

    public double getWeight() {
        return getData().getWeight();
    }

    @Override
    public String toString() {
        return "{" +
                data +
                ", " + score.getResult() +
                '}';
    }
}