CWMSVUE-600 Updating effective data retrieval to not include aliases

Author: rma-bryson

cwms-data-api/src/main/java/cwms/cda/data/dao/RatingSpecDao.java

@@ -40,7 +40,6 @@
 import java.time.Instant;
 import java.time.ZoneId;
 import java.time.ZonedDateTime;
-import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Calendar;
@@ -66,9 +65,16 @@
 import org.jetbrains.annotations.NotNull;
 import org.jooq.Condition;
 import org.jooq.DSLContext;
+import org.jooq.Field;
 import org.jooq.Record;
+import org.jooq.Record1;
+import org.jooq.Record2;
 import org.jooq.ResultQuery;
+import org.jooq.SelectConditionStep;
+import org.jooq.SelectForUpdateStep;
 import org.jooq.conf.ParamType;
+import org.jooq.impl.DSL;
+import static org.jooq.impl.DSL.field;
 import usace.cwms.db.dao.util.OracleTypeMap;
 import usace.cwms.db.jooq.codegen.packages.CWMS_RATING_PACKAGE;
 import usace.cwms.db.jooq.codegen.tables.AV_RATING;
@@ -422,13 +428,17 @@ public void create(String xml, boolean failIfExists) {
 
     public RatingEffectiveDatesMap retrieveSpecEffectiveDates(String officeIdMask, String specIdMask, Instant begin, Instant end) {
         return connectionResult(dsl, conn -> {
+            Set<String> ratingIdsNoAliases = getRatingIds(conn, officeIdMask, "*", false);
             //office->spec->dates
             NavigableMap<String, NavigableMap<String, NavigableSet<Instant>>> specDateMap = new TreeMap<>();
             ResultSet rs = catRatings(conn, officeIdMask, specIdMask, begin, end);
             OracleTypeMap.checkMetaData(rs.getMetaData(), RATINGS_COLUMN_LIST, "Ratings");
             while(rs.next()) {
                 String officeId = rs.getString(OFFICE_ID);
                 String specId = rs.getString(SPECIFICATION_ID);
+                if(!ratingIdsNoAliases.contains(specId)) { // skip aliased specs
+                    continue;
+                }
                 Timestamp timestamp = rs.getTimestamp(EFFECTIVE_DATE, GMT_CALENDAR);
                 Instant date = timestamp.toInstant();
                 NavigableSet<Instant> dateList = specDateMap.computeIfAbsent(officeId, k -> new TreeMap<>())
@@ -490,4 +500,36 @@ private ResultSet catRatings(Connection conn, String officeIdMask, String specId
 
         return output;
     }
+
+    private Set<String> getRatingIds(Connection conn, String office, String templateIdMask, boolean includeAliases) {
+        AV_RATING_SPEC specView = AV_RATING_SPEC.AV_RATING_SPEC;
+        Condition condition = DSL.noCondition();
+
+        if (office != null) {
+            condition = condition.and(specView.OFFICE_ID.eq(office));
+        }
+
+        if (templateIdMask != null) {
+            Condition ratingIdLike = JooqDao.caseInsensitiveLikeRegex(specView.RATING_ID,
+                    templateIdMask);
+            condition = condition.and(ratingIdLike);
+        }
+
+        if(!includeAliases) {
+            condition = condition.and(specView.ALIASED_ITEM.isNull());
+        }
+
+        Field<String> idField = field("RATING_ID", String.class);
+
+        SelectConditionStep<Record2<String, String>> ratingStep = DSL.using(conn).select(
+                        specView.OFFICE_ID,
+                        specView.RATING_ID.as(idField))
+                .from(specView)
+                .where(condition);
+
+        SelectForUpdateStep<Record1<String>> query = DSL.using(conn).selectDistinct(idField)
+                .from(ratingStep)
+                .orderBy(idField.asc());
+        return new LinkedHashSet<>(query.fetch(idField));
+    }
 }