From 3a53039b3ee82e3453f319e22189d9564ad57469 Mon Sep 17 00:00:00 2001 From: Stuart Boston Date: Fri, 22 Nov 2013 14:38:50 +0000 Subject: [PATCH] Updated Results to get the size of the results when not populated by the wrapper --- .../com/omertron/themoviedbapi/results/TmdbResults.java | 2 +- .../omertron/themoviedbapi/results/TmdbResultsList.java | 9 +++++++++ .../omertron/themoviedbapi/results/TmdbResultsMap.java | 9 +++++++++ 3 files changed, 19 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/omertron/themoviedbapi/results/TmdbResults.java b/src/main/java/com/omertron/themoviedbapi/results/TmdbResults.java index ac24988f6..ca218add0 100644 --- a/src/main/java/com/omertron/themoviedbapi/results/TmdbResults.java +++ b/src/main/java/com/omertron/themoviedbapi/results/TmdbResults.java @@ -100,6 +100,6 @@ public abstract class TmdbResults { @Override public String toString() { - return ToStringBuilder.reflectionToString(this, ToStringStyle.DEFAULT_STYLE); + return ToStringBuilder.reflectionToString(this, ToStringStyle.SHORT_PREFIX_STYLE); } } diff --git a/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsList.java b/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsList.java index 8242095ba..221d05a32 100644 --- a/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsList.java +++ b/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsList.java @@ -47,4 +47,13 @@ public final class TmdbResultsList extends TmdbResults { public void setResults(List results) { this.results = results; } + + @Override + public int getTotalResults() { + if (super.getTotalResults() == 0) { + return results.size(); + } else { + return super.getTotalResults(); + } + } } diff --git a/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsMap.java b/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsMap.java index 1e27cbd05..0203b158f 100644 --- a/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsMap.java +++ b/src/main/java/com/omertron/themoviedbapi/results/TmdbResultsMap.java @@ -44,4 +44,13 @@ public final class TmdbResultsMap extends TmdbResults { public void setResults(Map results) { this.results = results; } + + @Override + public int getTotalResults() { + if (super.getTotalResults() == 0) { + return results.size(); + } else { + return super.getTotalResults(); + } + } }