// Add jitpack
repositories {
    maven { url = uri("https://jitpack.io") }
}

// Include packages
dependencies {
    implementation("com.github.cdekok:kueue-core:0.1.0")
    implementation("com.github.cdekok:kueue-pg-vertx:0.1.0")
    implementation("com.github.cdekok:kueue-retry:0.1.0")
    implementation("com.github.cdekok:kueue-serializer-kotlinx:0.1.0")
}

// postgres client
val db = PgPool.pool(
    PgConnectOptions.fromEnv(),
    PoolOptions().setMaxSize(10),
)

// kotlinx json serializer
val serializer = KotlinxMessageSerializer(
    Json {
        classDiscriminator = "type"
        serializersModule = SerializersModule {
            polymorphic(Message::class) {
                subclass(RecordCreated::class)
            }
        }
    }
)

// postgres producer
val producer: Producer = PgProducer(
    client = db,
    serializer = serializer,
)

// message to send
@Serializable
data class RecordCreated(
    val id: Int,
    val title: String,
) : Message

val message = RecordCreated(
    id = 123,
    title = "test title $it",
)

// send the message
producer.send(
    topic = DEFAULT_TOPIC,
    message = message,
)

/**
 * Process messages one by one
 */
class TestListener : EventListener {
    @EventHandler
    fun on(event: RecordCreated) {
        logger.info { "handle record created $event" }
    }

    @EventHandler
    fun on(event: RecordCreated) {
        logger.info { "handle record created $event" }
    }
}

/**
 * Batch process messages
 */
class BatchListener : EventListener {
    @EventHandler
    fun on(event: List<RecordCreated>) {
        logger.info { "handle record created ${event.size}" }
    }
}

val listeners = listOf(
    TestListener(),
    BatchListener(),
)

val consumer = PgConsumer(
    client = pgPool(),
    serializer = kotlinXSerializer(),
)

runBlocking {
    consumer.subscribe(
        topic = "records",
        amount = 10,
        listeners = listeners,
    )
}