frontend/admin/app/dfp/Reader.scala at team-values · devhttps/frontend · GitHub

28 lines (22 loc) · 849 Bytes

package dfp

import com.google.api.ads.dfp.axis.utils.v201802.StatementBuilder

import com.google.api.ads.dfp.axis.utils.v201802.StatementBuilder.SUGGESTED_PAGE_LIMIT

import com.google.api.ads.dfp.axis.v201802._

import scala.annotation.tailrec

object Reader {

def read[T](statementBuilder: StatementBuilder)(readPage: Statement => (Array[T], Int)): Seq[T] = {

@tailrec

def read(soFar: Seq[T]): Seq[T] = {

val (pageOfResults, totalResultSetSize) = readPage(statementBuilder.toStatement)

val resultsSoFar = Option(pageOfResults).map(soFar ++ _).getOrElse(soFar)

if (resultsSoFar.size >= totalResultSetSize) {

resultsSoFar

} else {

statementBuilder.increaseOffsetBy(SUGGESTED_PAGE_LIMIT)

read(resultsSoFar)

}

statementBuilder.limit(SUGGESTED_PAGE_LIMIT)

read(Nil)

}

Expand file tree