|
| 1 | +/*********************************************************************** |
| 2 | + * Copyright (c) 2013-2025 General Atomics Integrated Intelligence, Inc. |
| 3 | + * All rights reserved. This program and the accompanying materials |
| 4 | + * are made available under the terms of the Apache License, Version 2.0 |
| 5 | + * which accompanies this distribution and is available at |
| 6 | + * https://www.apache.org/licenses/LICENSE-2.0 |
| 7 | + ***********************************************************************/ |
| 8 | + |
| 9 | +package org.locationtech.geomesa.fs.data.stats |
| 10 | + |
| 11 | +import org.geotools.api.feature.simple.SimpleFeatureType |
| 12 | +import org.geotools.api.filter.Filter |
| 13 | +import org.geotools.util.factory.Hints |
| 14 | +import org.locationtech.geomesa.features.ScalaSimpleFeature |
| 15 | +import org.locationtech.geomesa.fs.data.FileSystemDataStore |
| 16 | +import org.locationtech.geomesa.index.index.attribute.AttributeIndexKey |
| 17 | +import org.locationtech.geomesa.index.stats.RunnableStats.UnoptimizedRunnableStats |
| 18 | +import org.locationtech.geomesa.index.stats.impl.MinMax |
| 19 | +import org.locationtech.geomesa.index.stats.impl.MinMax.MinMaxDefaults |
| 20 | +import org.locationtech.jts.geom.Point |
| 21 | + |
| 22 | +/** |
| 23 | + * Optimized stats using per-file bounds for non-exact cases |
| 24 | + * |
| 25 | + * @param ds datastore |
| 26 | + */ |
| 27 | +class FileSystemStats(ds: FileSystemDataStore) extends UnoptimizedRunnableStats(ds) { |
| 28 | + |
| 29 | + import org.locationtech.geomesa.fs.storage.core.RichSimpleFeatureType |
| 30 | + |
| 31 | + override def getCount( |
| 32 | + sft: SimpleFeatureType, |
| 33 | + filter: Filter, |
| 34 | + exact: Boolean, |
| 35 | + queryHints: Hints): Option[Long] = { |
| 36 | + if (!exact || filter == Filter.INCLUDE) { |
| 37 | + Some(ds.storage(sft.getTypeName).metadata.getFiles(filter).map(_.count).sum) |
| 38 | + } else { |
| 39 | + super.getCount(sft, filter, exact, queryHints) |
| 40 | + } |
| 41 | + } |
| 42 | + |
| 43 | + override def getMinMax[T]( |
| 44 | + sft: SimpleFeatureType, |
| 45 | + attribute: String, |
| 46 | + filter: Filter, |
| 47 | + exact: Boolean): Option[MinMax[T]] = { |
| 48 | + if (!exact || filter == Filter.INCLUDE) { |
| 49 | + val i = sft.indexOf(attribute) |
| 50 | + if (sft.nonSpatialBounds().contains(i)) { |
| 51 | + val binding = sft.getDescriptor(i).getType.getBinding |
| 52 | + val minMax = new MinMax[T](sft, sft.getDescriptor(i).getLocalName)(MinMaxDefaults(binding)) |
| 53 | + |
| 54 | + var min: String = null |
| 55 | + var max: String = null |
| 56 | + ds.storage(sft.getTypeName).metadata.getFiles(filter).foreach { file => |
| 57 | + file.attributeBounds.filter(_.attribute == i).foreach { bounds => |
| 58 | + if (max == null || max < bounds.upper) { |
| 59 | + max = bounds.upper |
| 60 | + } |
| 61 | + if (min == null || min > bounds.lower) { |
| 62 | + min = bounds.lower |
| 63 | + } |
| 64 | + } |
| 65 | + } |
| 66 | + if (min != null) { |
| 67 | + val alias = AttributeIndexKey.alias(binding) |
| 68 | + val sf = new ScalaSimpleFeature(sft, "") |
| 69 | + Seq(min, max).foreach { value => |
| 70 | + sf.setAttribute(i, AttributeIndexKey.decode(alias, value)) |
| 71 | + minMax.observe(sf) |
| 72 | + } |
| 73 | + } |
| 74 | + Some(minMax) |
| 75 | + } else if (sft.spatialBounds().contains(i) && sft.getDescriptor(i).getType.getBinding == classOf[Point]) { |
| 76 | + val minMax = new MinMax[T](sft, sft.getDescriptor(i).getLocalName)(MinMaxDefaults(classOf[Point])) |
| 77 | + |
| 78 | + var xmin, ymin, xmax, ymax: java.lang.Double = null |
| 79 | + ds.storage(sft.getTypeName).metadata.getFiles(filter).foreach { file => |
| 80 | + file.spatialBounds.filter(_.attribute == i).foreach { bounds => |
| 81 | + xmin = if (xmin == null) { bounds.xmin } else { math.min(xmin, bounds.xmin) } |
| 82 | + xmax = if (xmax == null) { bounds.xmax } else { math.max(xmax, bounds.xmax) } |
| 83 | + ymin = if (ymin == null) { bounds.ymin } else { math.min(ymin, bounds.ymin) } |
| 84 | + ymax = if (ymax == null) { bounds.ymax } else { math.max(ymax, bounds.ymax) } |
| 85 | + } |
| 86 | + } |
| 87 | + if (xmin != null) { |
| 88 | + val sf = new ScalaSimpleFeature(sft, "") |
| 89 | + Seq(xmin -> ymin, xmax -> ymax).foreach { case (x, y) => |
| 90 | + sf.setAttribute(i, s"POINT ($x $y)") |
| 91 | + minMax.observe(sf) |
| 92 | + } |
| 93 | + } |
| 94 | + Some(minMax) |
| 95 | + } else { |
| 96 | + super.getMinMax(sft, attribute, filter, exact) |
| 97 | + } |
| 98 | + } else { |
| 99 | + super.getMinMax(sft, attribute, filter, exact) |
| 100 | + } |
| 101 | + } |
| 102 | +} |
0 commit comments