com.caseystella.analytics.outlier.Outlier Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of logisland-outlier-detection-plugin Show documentation
There is a newer version: 0.9.7
Show newest version
/**
 * Copyright (C) 2016 Hurence ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.caseystella.analytics.outlier;

import com.caseystella.analytics.DataPoint;
import com.caseystella.analytics.distribution.TimeRange;
import com.google.common.base.Joiner;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Outlier {
    DataPoint dataPoint;
    Severity severity;
    TimeRange range;
    Double score;
    int numPts;
    List sample;

    public Outlier(DataPoint dataPoint, Severity severity, TimeRange range, Double score, int numPts) {
        this.dataPoint = dataPoint;
        this.severity = severity;
        this.range = range;
        this.score = score;
        this.numPts = numPts;
    }

    public static Map groupingFilter(DataPoint dp, List groupingKeys, List allTags) {
        Map filter = new HashMap<>();
        if(groupingKeys != null) {
            for (String gk : groupingKeys) {
                String k = dp.getMetadata().get(gk);
                if (k != null) {
                    filter.put(gk, k);
                }
            }
        }
        if(allTags != null) {
            for(String tag : allTags) {
                if(!filter.containsKey(tag)) {
                    filter.put(tag, "*");
                }
            }
        }
        return filter;
    }

    public static String groupingKey(DataPoint dp, List groupingKeys) {
        return groupingKey(dp.getSource(), dp.getMetadata(), groupingKeys);
    }

    public static String groupingKey(String source, Map metadata, List groupingKeys) {
        List keyParts = new ArrayList<>();
        keyParts.add(source);
        if(groupingKeys != null) {
            for (String gk : groupingKeys) {
                String k = metadata.get(gk);
                if (k != null) {
                    keyParts.add(k);
                }
            }
        }
        return Joiner.on('_').join(keyParts);
    }

    public List getSample() {
        return sample;
    }

    public void setSample(List sample) {
        this.sample = sample;
    }

    public int getNumPts() {
        return numPts;
    }

    public Double getScore() {
        return score;
    }

    public void setScore(Double score) {
        this.score = score;
    }

    public DataPoint getDataPoint() {
        return dataPoint;
    }

    public void setDataPoint(DataPoint dataPoint) {
        this.dataPoint = dataPoint;
    }

    public Severity getSeverity() {
        return severity;
    }

    public void setSeverity(Severity severity) {
        this.severity = severity;
    }

    public TimeRange getRange() {
        return range;
    }

    public void setRange(TimeRange range) {
        this.range = range;
    }

    @Override
    public String toString() {
        return "Outlier{" +
                "dataPoint=" + dataPoint +
                ", severity=" + severity +
                ", range=" + range +
                ", score=" + score +
                '}';
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        Outlier outlier = (Outlier) o;

        if (getNumPts() != outlier.getNumPts()) return false;
        if (getDataPoint() != null ? !getDataPoint().equals(outlier.getDataPoint()) : outlier.getDataPoint() != null)
            return false;
        if (getSeverity() != outlier.getSeverity()) return false;
        if (getRange() != null ? !getRange().equals(outlier.getRange()) : outlier.getRange() != null) return false;
        if (getScore() != null ? !getScore().equals(outlier.getScore()) : outlier.getScore() != null) return false;
        return getSample() != null ? getSample().equals(outlier.getSample()) : outlier.getSample() == null;

    }

    @Override
    public int hashCode() {
        int result = getDataPoint() != null ? getDataPoint().hashCode() : 0;
        result = 31 * result + (getSeverity() != null ? getSeverity().hashCode() : 0);
        result = 31 * result + (getRange() != null ? getRange().hashCode() : 0);
        result = 31 * result + (getScore() != null ? getScore().hashCode() : 0);
        result = 31 * result + getNumPts();
        result = 31 * result + (getSample() != null ? getSample().hashCode() : 0);
        return result;
    }
}