File tree Expand file tree Collapse file tree
src/main/java/com/afrozaar/wordpress/wpapi/v2/response Expand file tree Collapse file tree Original file line number Diff line number Diff line change 88import org .slf4j .Logger ;
99import org .slf4j .LoggerFactory ;
1010
11+ import java .util .function .Function ;
12+
1113import java .net .URI ;
1214import java .util .List ;
1315import java .util .Objects ;
1416import java .util .Optional ;
15- import java .util .function .Function ;
1617
1718public class PagedResponse <T > {
1819
@@ -125,10 +126,15 @@ public Builder<BT> withPages(int pages) {
125126 }
126127
127128 public Builder <BT > withPages (HttpHeaders headers ) {
128- headers .get (Strings .HEADER_TOTAL_PAGES ).stream ()
129+
130+ getHeaders (headers , Strings .HEADER_TOTAL_PAGES ).stream ()
129131 .findFirst ()
130132 .ifPresent (pages -> Builder .this .withPages (Integer .valueOf (pages )));
131133 return this ;
132134 }
135+
136+ private List <String > getHeaders (HttpHeaders headers , String headerTotalPages ) {
137+ return Optional .ofNullable (headers .get (headerTotalPages )).orElse (headers .get (headerTotalPages .toLowerCase ()));
138+ }
133139 }
134140}
You can’t perform that action at this time.
0 commit comments