All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.openmetadata.service.dataInsight.MostViewedEntitiesAggregator Maven / Gradle / Ivy

Go to download

There is a newer version: 1.5.11
Show newest version
package org.openmetadata.service.dataInsight;

import java.util.ArrayList;
import java.util.List;
import org.openmetadata.schema.dataInsight.type.MostViewedEntities;

public abstract class MostViewedEntitiesAggregator
    implements DataInsightAggregatorInterface {
  protected final A aggregations;

  protected MostViewedEntitiesAggregator(A aggregations) {
    this.aggregations = aggregations;
  }

  @Override
  public List aggregate() {
    M entityFqnBuckets = getEntityFqnBuckets(this.aggregations);
    List data = new ArrayList<>();
    for (B entityFqnBucket : getBuckets(entityFqnBuckets)) {
      String tableFqn = getKeyAsString(entityFqnBucket);
      S sumPageViews = getAggregations(entityFqnBucket, "pageViews");
      M ownerBucket = getBucketAggregation(entityFqnBucket, "owners");
      M entityTypeBucket = getBucketAggregation(entityFqnBucket, "entityType");
      M entityHrefBucket = getBucketAggregation(entityFqnBucket, "entityHref");
      String owner = getFirstValueFromBucketOrNull(ownerBucket);
      String entityType = getFirstValueFromBucketOrNull(entityTypeBucket);
      String entityHref = getFirstValueFromBucketOrNull(entityHrefBucket);

      data.add(
          new MostViewedEntities()
              .withEntityFqn(tableFqn)
              .withOwner(owner)
              .withEntityType(entityType)
              .withEntityHref(entityHref)
              .withPageViews(getValue(sumPageViews)));
    }

    return data;
  }

  protected abstract Double getValue(S sumPageViews);

  protected abstract M getBucketAggregation(B bucket, String key);

  protected abstract S getAggregations(B bucket, String key);

  protected abstract String getKeyAsString(B bucket);

  protected abstract List getBuckets(M bucket);

  protected abstract M getEntityFqnBuckets(A aggregations);

  protected String getFirstValueFromBucketOrNull(M bucket) {
    // for list values we'll pick the first value
    if (!getBuckets(bucket).isEmpty()) {
      return getKeyAsString(getBuckets(bucket).get(0));
    }
    return null;
  }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy