Skip to content
This repository has been archived by the owner on Jan 27, 2025. It is now read-only.

Commit

Permalink
Add sort_field() Fix function. (#100)
Browse files Browse the repository at this point in the history
  • Loading branch information
@blackwinter
blackwinter committed Jan 11, 2022
1 parent 3d5b4e4 commit 0155cee
Show file tree
Hide file tree
Showing 2 changed files with 175 additions and 0 deletions.
26 changes: 26 additions & 0 deletions metafix/src/main/java/org/metafacture/metafix/FixMethod.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,13 @@

import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.Set;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
Expand Down Expand Up @@ -344,6 +348,23 @@ else if (v.isArray()) {
});
}
},
sort_field {
public void apply(final Metafix metafix, final Record record, final List<String> params, final Map<String, String> options) {
record.transformField(params.get(0), v -> {
final boolean numeric = getBoolean(options, "numeric");
final boolean reverse = getBoolean(options, "reverse");
final boolean uniq = getBoolean(options, "uniq");

final Stream<Value> stream = v.asArray().stream();
final Function<Value, String> function = Value::asString;
final Comparator<Value> comparator = numeric ?
Comparator.comparing(function.andThen(Integer::parseInt)) : Comparator.comparing(function);

return v.isArray() ? new Value((uniq ? unique(stream) : stream)
.sorted(reverse ? comparator.reversed() : comparator).collect(Collectors.toList())) : null;
});
}
},
substring {
public void apply(final Metafix metafix, final Record record, final List<String> params, final Map<String, String> options) {
record.transformFields(params, s -> s.substring(getInteger(params, 1), getInteger(params, 2) - 1));
Expand Down Expand Up @@ -382,6 +403,11 @@ private static Value newArray(final Stream<Value> stream) {
return Value.newArray(a -> stream.forEach(a::add));
}

private static Stream<Value> unique(final Stream<Value> stream) {
final Set<Value> set = new HashSet<>();
return stream.filter(set::add);
}

abstract void apply(Metafix metafix, Record record, List<String> params, Map<String, String> options);

}
149 changes: 149 additions & 0 deletions metafix/src/test/java/org/metafacture/metafix/MetafixMethodTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -701,6 +701,155 @@ public void shouldReverseArray() {
);
}

@Test
public void shouldSortField() {
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers)"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "41");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
o.get().startRecord("1");
o.get().startEntity("numbers");
o.get().literal("1", "41");
o.get().literal("2", "42");
o.get().literal("3", "6");
o.get().literal("4", "6");
o.get().endEntity();
o.get().endRecord();
}
);
}

@Test
public void shouldSortFieldNumerically() {
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers, numeric: 'true')"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "41");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
o.get().startRecord("1");
o.get().startEntity("numbers");
o.get().literal("1", "6");
o.get().literal("2", "6");
o.get().literal("3", "41");
o.get().literal("4", "42");
o.get().endEntity();
o.get().endRecord();
}
);
}

@Test
public void shouldFailToSortNumericallyWithInvalidNumber() {
assertThrows(NumberFormatException.class, "For input string: \"x\"", () ->
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers, numeric: 'true')"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "x");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
}
)
);
}

@Test
public void shouldSortFieldInReverse() {
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers, reverse: 'true')"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "41");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
o.get().startRecord("1");
o.get().startEntity("numbers");
o.get().literal("1", "6");
o.get().literal("2", "6");
o.get().literal("3", "42");
o.get().literal("4", "41");
o.get().endEntity();
o.get().endRecord();
}
);
}

@Test
public void shouldSortFieldNumericallyInReverse() {
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers, numeric: 'true', reverse: 'true')"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "41");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
o.get().startRecord("1");
o.get().startEntity("numbers");
o.get().literal("1", "42");
o.get().literal("2", "41");
o.get().literal("3", "6");
o.get().literal("4", "6");
o.get().endEntity();
o.get().endRecord();
}
);
}

@Test
public void shouldSortFieldAndRemoveDuplicates() {
MetafixTestHelpers.assertFix(streamReceiver, Arrays.asList(
"sort_field(numbers, uniq: 'true')"
),
i -> {
i.startRecord("1");
i.literal("numbers", "6");
i.literal("numbers", "42");
i.literal("numbers", "41");
i.literal("numbers", "6");
i.endRecord();
},
o -> {
o.get().startRecord("1");
o.get().startEntity("numbers");
o.get().literal("1", "41");
o.get().literal("2", "42");
o.get().literal("3", "6");
o.get().endEntity();
o.get().endRecord();
}
);
}

private void assertThrows(final Class<?> expectedClass, final String expectedMessage, final Executable executable) {
final Throwable exception = Assertions.assertThrows(MetafactureException.class, executable).getCause();
Assertions.assertSame(expectedClass, exception.getClass());
Expand Down

0 comments on commit 0155cee

Please sign in to comment.