com.snowplowanalytics.snowplow.sources.pubsub.PubsubSourceConfig.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pubsub_2.12 Show documentation

pubsub

The newest version!

/*
 * Copyright (c) 2023-present Snowplow Analytics Ltd. All rights reserved.
 *
 * This program is licensed to you under the Snowplow Community License Version 1.0,
 * and you may not use this file except in compliance with the Snowplow Community License Version 1.0.
 * You may obtain a copy of the Snowplow Community License Version 1.0 at https://docs.snowplow.io/community-license-1.0
 */
package com.snowplowanalytics.snowplow.sources.pubsub

import io.circe.Decoder
import io.circe.generic.semiauto._
import io.circe.config.syntax._

import scala.concurrent.duration.FiniteDuration

import com.snowplowanalytics.snowplow.pubsub.GcpUserAgent

case class PubsubSourceConfig(
  subscription: PubsubSourceConfig.Subscription,
  parallelPullFactor: BigDecimal,
  bufferMaxBytes: Int,
  maxAckExtensionPeriod: FiniteDuration,
  minDurationPerAckExtension: FiniteDuration,
  maxDurationPerAckExtension: FiniteDuration,
  gcpUserAgent: GcpUserAgent,
  shutdownTimeout: FiniteDuration,
  maxPullsPerTransportChannel: Int
)

object PubsubSourceConfig {

  case class Subscription(projectId: String, subscriptionId: String)

  private implicit def subscriptionDecoder: Decoder[Subscription] =
    Decoder.decodeString
      .map(_.split("/"))
      .emap {
        case Array("projects", projectId, "subscriptions", subscriptionId) =>
          Right(Subscription(projectId, subscriptionId))
        case _ =>
          Left("Expected format: projects//subscriptions/")
      }

  implicit def decoder: Decoder[PubsubSourceConfig] = deriveDecoder[PubsubSourceConfig]
}