Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
Expand All @@ -39,6 +40,11 @@ public static List<ColumnGridItem> groupColumns(

for (Map.Entry<String, List<ColumnWithContext>> entry : columnsByName.entrySet()) {
String columnName = entry.getKey();
if (columnName == null) {
// A column with no name isn't a valid grid row (columnName is required in the response);
// skip the malformed bucket so we return the remaining valid columns.
continue;
}
List<ColumnWithContext> occurrences = entry.getValue();

Map<String, ColumnMetadataGroup> groups = new HashMap<>();
Expand Down Expand Up @@ -101,6 +107,17 @@ public static List<ColumnGridItem> groupColumns(
gridItems.add(gridItem);
}

// groupColumns is the single sink for every query path. The input map's iteration order is not
// guaranteed (current callers build it as a HashMap), so sort here for a stable alphabetical
// order. Case-insensitive for natural listing, with a case-sensitive tie-breaker so names that
// differ only in case (e.g. "Name"/"name") stay stable instead of relying on map order.
// nullsLast guards against a null column name (e.g. a null map key) rather than throwing.
gridItems.sort(
Comparator.comparing(
ColumnGridItem::getColumnName, Comparator.nullsLast(String.CASE_INSENSITIVE_ORDER))
.thenComparing(
ColumnGridItem::getColumnName, Comparator.nullsLast(Comparator.naturalOrder())));
Comment thread
sonika-shah marked this conversation as resolved.

return gridItems;
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.Test;
Expand Down Expand Up @@ -337,4 +338,103 @@ void testGroupColumns_emptyMap() {
assertNotNull(result);
assertEquals(0, result.size());
}

@Test
void testGroupColumns_resultsAreSortedAscendingByColumnName() {
// Feed columns in a non-alphabetical order (LinkedHashMap keeps the input order deterministic)
// and verify the grid is returned sorted ascending by column name.
Map<String, List<ColumnMetadataGrouper.ColumnWithContext>> columnsByName =
new LinkedHashMap<>();
List<String> scrambled =
List.of(
"c_admpkg",
"c_addin",
"c_active_agent_id",
"c_admincare",
"c_address",
"c_admission_date",
"c_acasclop",
"c_abnorm",
"c_abbrevation");
for (String name : scrambled) {
columnsByName.put(name, singleOccurrence(name));
}

List<ColumnGridItem> result = ColumnMetadataGrouper.groupColumns(columnsByName);

List<String> actual = result.stream().map(ColumnGridItem::getColumnName).toList();
List<String> expected = new ArrayList<>(scrambled);
expected.sort(String.CASE_INSENSITIVE_ORDER);
assertEquals(expected, actual);
assertEquals("c_abbrevation", actual.get(0));
assertEquals("c_abnorm", actual.get(1));
}

@Test
void testGroupColumns_sortIsCaseInsensitive() {
// Insertion order (Zebra, apple, Mango) is non-alphabetical and preserved by LinkedHashMap.
Map<String, List<ColumnMetadataGrouper.ColumnWithContext>> columnsByName =
new LinkedHashMap<>();
for (String name : List.of("Zebra", "apple", "Mango")) {
columnsByName.put(name, singleOccurrence(name));
}

List<ColumnGridItem> result = ColumnMetadataGrouper.groupColumns(columnsByName);

List<String> actual = result.stream().map(ColumnGridItem::getColumnName).toList();
assertEquals(List.of("apple", "Mango", "Zebra"), actual);
}

@Test
void testGroupColumns_caseOnlyCollisionsAreDeterministic() {
// Names differing only in case are equal under CASE_INSENSITIVE_ORDER, so the case-sensitive
// tie-breaker decides their order. Insert in reverse of the expected result to prove the sort,
// not the map, determines it.
Map<String, List<ColumnMetadataGrouper.ColumnWithContext>> columnsByName =
new LinkedHashMap<>();
for (String name : List.of("name", "Name")) {
columnsByName.put(name, singleOccurrence(name));
}

List<ColumnGridItem> result = ColumnMetadataGrouper.groupColumns(columnsByName);

List<String> actual = result.stream().map(ColumnGridItem::getColumnName).toList();
// Natural String order: 'N' (0x4E) precedes 'n' (0x6E).
assertEquals(List.of("Name", "name"), actual);
}

@Test
void testGroupColumns_nullColumnNameIsSkipped() {
// A null column name (e.g. a null map key) is dropped, not returned with a null required field,
// and must not throw. The remaining valid columns still come back sorted.
Map<String, List<ColumnMetadataGrouper.ColumnWithContext>> columnsByName =
new LinkedHashMap<>();
columnsByName.put("beta", singleOccurrence("beta"));
columnsByName.put(null, singleOccurrence(null));
columnsByName.put("alpha", singleOccurrence("alpha"));

List<ColumnGridItem> result = ColumnMetadataGrouper.groupColumns(columnsByName);

List<String> actual = result.stream().map(ColumnGridItem::getColumnName).toList();
assertEquals(List.of("alpha", "beta"), actual);
}

private static List<ColumnMetadataGrouper.ColumnWithContext> singleOccurrence(String columnName) {
Column column = new Column();
column.setName(columnName);
column.setDataType(ColumnDataType.STRING);
column.setFullyQualifiedName("sample_data.db.schema.table1." + columnName);

List<ColumnMetadataGrouper.ColumnWithContext> occurrences = new ArrayList<>();
occurrences.add(
new ColumnMetadataGrouper.ColumnWithContext(
column,
"table",
"sample_data.db.schema.table1",
"Table 1",
"sample_data",
"db",
"schema"));
return occurrences;
}
}
Loading