AdvertiserDashboardUrlTopN.scala

object TopURLHelper {
  // expects a list of aggregates of type 'only' URL or RR
  // def combine(l: List[Aggregate]): Aggregate = {
  //   val builder = Aggregate.newBuilder()
  // }

  def combine(l: List[Event]): List[Aggregate] = {
    val map = new HashMap[ByteString, Aggregate.Builder]()

    l.map { x =>

      // add either rr, or url or both
      // I think this will throw an exception when not set
      if (!x.getRr.isEmpty) {
        // get it from map
        val b = map.get(x.getRrHash()).getOrElse{
          Aggregate
            .newBuilder
            .setType(Aggregate.Type.RR)
            .setUrl(x.getRr)
        }
        bookKeeping b

        map.set(x.getRrHash, b)
      }

      if (!x.getUrl.isEmpty) {
        val b = map.get(x.getUrlHash()).getOrElse{
          Aggregate
            .newBuilder
            .setType(Aggregate.Type.URL)
            .setUrl(x.getUrl)
        }
        bookKeeping b

        map.set(x.getUrlHash, b)
      }

      def bookKeeping(builder: Aggregate.Builder) = {
        x.getType match {
          case Event.Type.IMPRESSION =>
            builder.setImps(builder.getImps + 1)
          case Event.Type.CLICK =>
            builder.setClks(builder.getClks + 1)
        }
        builder.setSpend(x.getCost)
      }
    }

    map.values.map(_.build).toList
  }
}