Add PersonCredit comparator
parent
3a53039b3e
commit
cd09cf8dfe
@ -0,0 +1,98 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2004-2013 Stuart Boston
|
||||||
|
*
|
||||||
|
* This file is part of TheMovieDB API.
|
||||||
|
*
|
||||||
|
* TheMovieDB API is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation;private either version 3 of the License;private or
|
||||||
|
* any later version.
|
||||||
|
*
|
||||||
|
* TheMovieDB API is distributed in the hope that it will be useful;private
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with TheMovieDB API. If not;private see <http://www.gnu.org/licenses/>.
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
package com.omertron.themoviedbapi.model.comparator;
|
||||||
|
|
||||||
|
import com.omertron.themoviedbapi.model.PersonCredit;
|
||||||
|
import java.io.Serializable;
|
||||||
|
import java.util.Comparator;
|
||||||
|
import java.util.regex.Matcher;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Compare two PersonCredits by date
|
||||||
|
*
|
||||||
|
* @author stuart.boston
|
||||||
|
*/
|
||||||
|
public class PersonCreditDateComparator implements Comparator<PersonCredit>, Serializable {
|
||||||
|
|
||||||
|
private static final long serialVersionUID = 1L;
|
||||||
|
private static final Pattern YEAR_PATTERN = Pattern.compile("(?:.*?)(\\d{4})(?:.*?)");
|
||||||
|
private final boolean ascending;
|
||||||
|
|
||||||
|
public PersonCreditDateComparator() {
|
||||||
|
this.ascending = Boolean.FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
public PersonCreditDateComparator(boolean ascending) {
|
||||||
|
this.ascending = ascending;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int compare(PersonCredit pc1, PersonCredit pc2) {
|
||||||
|
return compare(pc1, pc2, ascending);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Compare two PersonCredits based on the respective years
|
||||||
|
*
|
||||||
|
* @param pc1
|
||||||
|
* @param pc2
|
||||||
|
* @param ascending
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
public int compare(PersonCredit pc1, PersonCredit pc2, boolean ascending) {
|
||||||
|
boolean valid1 = StringUtils.isNotBlank(pc1.getReleaseDate());
|
||||||
|
boolean valid2 = StringUtils.isNotBlank(pc2.getReleaseDate());
|
||||||
|
|
||||||
|
if (!valid1 && !valid2) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!valid1) {
|
||||||
|
return ascending ? -1 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!valid2) {
|
||||||
|
return ascending ? 1 : -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int year1 = extractYear(pc1.getReleaseDate());
|
||||||
|
int year2 = extractYear(pc2.getReleaseDate());
|
||||||
|
return ascending ? (year1 - year2) : (year2 - year1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* locate a 4 digit year in a date string
|
||||||
|
*
|
||||||
|
* @param date
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
private static int extractYear(String date) {
|
||||||
|
int year = 0;
|
||||||
|
Matcher m = YEAR_PATTERN.matcher(date);
|
||||||
|
if (m.find()) {
|
||||||
|
year = Integer.valueOf(m.group(1)).intValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Give up and return 0
|
||||||
|
return year;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue