Groovy Goodness: Use Sortable Annotation to Make Classes Comparable
Since Groovy 2.3 we can use the @Sortable
annotation to make a class implement the Comparable
interface. Also new comparator
methods are added. All properties of a class are used to implement the compareTo
method. The order of the properties determines the priority used when sorting. With the annotation parameters includes
and excludes
we can define which properties of the class need to be used to implement the compareTo
method.
In the following class with the name Course
we define three properties title
, beginDate
and maxAttendees
. We also apply the @Sortable
annotation. Notice we cannot use int
as a type, because it doesn't implement the Comparable
interface. The class Integer
does.
import groovy.transform.Sortable
import groovy.transform.ToString
@Sortable
@ToString
class Course {
// Order of properties determines priority when sorting
String title
Date beginDate
Integer maxAttendees // int doesn't implement Comparable, so use Integer
}
final Course groovy = new Course(
title: 'Groovy', beginDate: new Date() + 7, maxAttendees: 40)
final Course groovy2 = new Course(
title: 'Groovy', beginDate: new Date() + 2, maxAttendees: 50)
final Course grails = new Course(
title: 'Grails', beginDate: new Date() + 1, maxAttendees: 20)
final List courses = [groovy, groovy2, grails]
assert courses.last().title == 'Grails'
// Use sort() method to sort
final List sorted = courses.sort(false /* do not mutate original collection */)
assert sorted.first().title == 'Grails'
assert sorted.last().title == 'Groovy'
assert sorted.maxAttendees == [20, 50, 40]
If we only want the properties title
and maxAttendees
to be used in the compareTo
method we define those properties with the includes
parameter:
// Order of fields for includes determines priority when sorting
@Sortable(includes = ['title', 'maxAttendees'])
// Or @Sortable(excludes = ['beginDate'])
@ToString
class CourseSort {
String title
Date beginDate
Integer maxAttendees
}
final Course groovy = new Course(
title: 'Groovy', beginDate: new Date() + 7, maxAttendees: 40)
final Course groovy2 = new Course(
title: 'Groovy', beginDate: new Date() + 2, maxAttendees: 50)
final Course grails = new Course(
title: 'Grails', beginDate: new Date() + 1, maxAttendees: 20)
final List courses = [groovy, groovy2, grails]
// Use sort() method to sort
final List sorted = courses.sort(false)
assert sorted.first().title == 'Grails'
assert sorted.last().title == 'Groovy'
assert sorted.maxAttendees == [20, 40, 50]
Besides implementing the Comparable
interface with a compareTo
method, the AST transformation also adds public static methods with the pattern comparatorBy_Property_
. These methods return a Comparator
object for the given property.
final Comparator byMaxAttendees = Course.comparatorByMaxAttendees()
final List sortedByMaxAttendees = courses.sort(false, byMaxAttendees)
assert sortedByMaxAttendees.maxAttendees == [20, 40, 50]
// beginDate is not used for sorting
assert sortedByMaxAttendees[2].beginDate < sortedByMaxAttendees[1].beginDate
assert Course.declaredMethods.name.findAll { it.startsWith('comparatorBy') } == ['comparatorByTitle', 'comparatorByMaxAttendees']
Code written with Groovy 2.3.